From 1c49f6789f01575569f8f73c43d9af3c03ba9db4 Mon Sep 17 00:00:00 2001
From: Tobias Casper <tobias.casper@gmail.com>
Date: Sat, 11 May 2019 00:11:51 +0200
Subject: [PATCH] Add open_dir delegation to main module

---
 lib/sftp_client.ex        | 10 ++++++++++
 test/sftp_client_test.exs | 20 ++++++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/lib/sftp_client.ex b/lib/sftp_client.ex
index 26efdfe..bd50a96 100644
--- 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
index 9d59f0f..a14121f 100644
--- 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!, [