From 2075550f078942d642c873bd2ec62343eb1a0f16 Mon Sep 17 00:00:00 2001
From: Christian Funkhouser <christian@funkhouse.rs>
Date: Wed, 7 Apr 2021 18:21:34 -0400
Subject: [PATCH] Test that Manager passes fh to _PlanOutputs

---
 octodns/manager.py                       |  5 +++--
 tests/config/plan-output-filehandle.yaml |  6 ++++++
 tests/test_octodns_manager.py            | 23 ++++++++++++++++++++++-
 3 files changed, 31 insertions(+), 3 deletions(-)
 create mode 100644 tests/config/plan-output-filehandle.yaml

diff --git a/octodns/manager.py b/octodns/manager.py
index f4f1491..e94e41e 100644
--- a/octodns/manager.py
+++ b/octodns/manager.py
@@ -270,8 +270,9 @@ class Manager(object):
     def sync(self, eligible_zones=[], eligible_sources=[], eligible_targets=[],
              dry_run=True, force=False, plan_output_fh=stdout):
         self.log.info('sync: eligible_zones=%s, eligible_targets=%s, '
-                      'dry_run=%s, force=%s', eligible_zones, eligible_targets,
-                      dry_run, force)
+                      'dry_run=%s, force=%s, plan_output_fh=%s',
+                      eligible_zones, eligible_targets, dry_run, force,
+                      plan_output_fh)
 
         zones = self.config['zones'].items()
         if eligible_zones:
diff --git a/tests/config/plan-output-filehandle.yaml b/tests/config/plan-output-filehandle.yaml
new file mode 100644
index 0000000..9c9bb87
--- /dev/null
+++ b/tests/config/plan-output-filehandle.yaml
@@ -0,0 +1,6 @@
+manager:
+  plan_outputs:
+    "doesntexist":
+      class: octodns.provider.plan.DoesntExist
+providers: {}
+zones: {}
diff --git a/tests/test_octodns_manager.py b/tests/test_octodns_manager.py
index 442ed49..003d414 100644
--- a/tests/test_octodns_manager.py
+++ b/tests/test_octodns_manager.py
@@ -8,7 +8,6 @@ from __future__ import absolute_import, division, print_function, \
 from os import environ
 from os.path import dirname, join
 from six import text_type
-from unittest import TestCase
 
 from octodns.record import Record
 from octodns.manager import _AggregateTarget, MainThreadExecutor, Manager, \
@@ -16,6 +15,10 @@ from octodns.manager import _AggregateTarget, MainThreadExecutor, Manager, \
 from octodns.yaml import safe_load
 from octodns.zone import Zone
 
+from mock import MagicMock, patch
+from unittest import TestCase
+
+
 from helpers import DynamicProvider, GeoProvider, NoSshFpProvider, \
     SimpleProvider, TemporaryDirectory
 
@@ -371,6 +374,24 @@ class TestManager(TestCase):
             with self.assertRaises(TypeError):
                 manager._populate_and_plan('unit.tests.', [NoZone()], [])
 
+    @patch('octodns.manager.Manager._get_named_class')
+    def test_sync_passes_file_handle(self, mock):
+        plan_output_mock = MagicMock()
+        plan_output_class_mock = MagicMock()
+        plan_output_class_mock.return_value = plan_output_mock
+        mock.return_value = plan_output_class_mock
+        fh_mock = MagicMock()
+
+        Manager(get_config_filename('plan-output-filehandle.yaml')
+                ).sync(plan_output_fh=fh_mock)
+
+        # Since we only care about the fh kwarg, and different _PlanOutputs are
+        # are free to require arbitrary kwargs anyway, we concern ourselves
+        # with checking the value of fh only.
+        plan_output_mock.run.assert_called()
+        _, kwargs = plan_output_mock.run.call_args
+        self.assertEqual(fh_mock, kwargs.get('fh'))
+
 
 class TestMainThreadExecutor(TestCase):