Add open_dir delegation to main module

tlux · May 10, 2019 · 1c49f67 · 1c49f67
1 parent 8ba50b2
commit 1c49f67
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/lib/sftp_client.ex b/lib/sftp_client.ex
@@ -263,11 +263,21 @@ defmodule SFTPClient do
     run(Operations.OpenDir, :open_dir!, [conn, path])
   end
 
+  @doc delegate_to: {Operations.OpenDir, :open_dir!, 3}
+  def open_dir!(conn, path, fun) do
+    run(Operations.OpenDir, :open_dir!, [conn, path, fun])
+  end
+
   @doc delegate_to: {Operations.OpenDir, :open_dir, 2}
   def open_dir(conn, path) do
     run(Operations.OpenDir, :open_dir, [conn, path])
   end
 
+  @doc delegate_to: {Operations.OpenDir, :open_dir, 3}
+  def open_dir(conn, path, fun) do
+    run(Operations.OpenDir, :open_dir, [conn, path, fun])
+  end
+
   @doc delegate_to: {Operations.OpenFile, :open_file!, 3}
   def open_file!(conn, path, modes) do
     run(Operations.OpenFile, :open_file!, [conn, path, modes])

diff --git a/test/sftp_client_test.exs b/test/sftp_client_test.exs
@@ -232,6 +232,16 @@ defmodule SFTPClientTest do
     end
   end
 
+  describe "open_dir!/3" do
+    test "delegate to Operations.OpenDir" do
+      assert_delegate_to_operation(Operations.OpenDir, :open_dir!, [
+        :conn_stub,
+        :path_stub,
+        :fun_stub
+      ])
+    end
+  end
+
   describe "open_dir/2" do
     test "delegate to Operations.OpenDir" do
       assert_delegate_to_operation(Operations.OpenDir, :open_dir, [
@@ -241,6 +251,16 @@ defmodule SFTPClientTest do
     end
   end
 
+  describe "open_dir/3" do
+    test "delegate to Operations.OpenDir" do
+      assert_delegate_to_operation(Operations.OpenDir, :open_dir, [
+        :conn_stub,
+        :path_stub,
+        :fun_stub
+      ])
+    end
+  end
+
   describe "open_file!/3" do
     test "delegate to Operations.OpenFile" do
       assert_delegate_to_operation(Operations.OpenFile, :open_file!, [