From 876c09dcc07efb721f3ac08262d23bb2d9861a4e Mon Sep 17 00:00:00 2001
From: Ross McFarland <ross@github.com>
Date: Sat, 3 Mar 2018 10:12:34 -0800
Subject: [PATCH] Flesh out UT for new Plan.exists messaging

---
 tests/test_octodns_plan.py | 45 ++++++++++++++++++++++++++++----------
 1 file changed, 33 insertions(+), 12 deletions(-)

diff --git a/tests/test_octodns_plan.py b/tests/test_octodns_plan.py
index 91dd948..7d849be 100644
--- a/tests/test_octodns_plan.py
+++ b/tests/test_octodns_plan.py
@@ -16,15 +16,6 @@ from octodns.zone import Zone
 from helpers import SimpleProvider
 
 
-class TestPlanLogger(TestCase):
-
-    def test_invalid_level(self):
-        with self.assertRaises(Exception) as ctx:
-            PlanLogger('invalid', 'not-a-level')
-        self.assertEquals('Unsupported level: not-a-level',
-                          ctx.exception.message)
-
-
 simple = SimpleProvider()
 zone = Zone('unit.tests.', [])
 existing = Record.new(zone, 'a', {
@@ -48,15 +39,45 @@ create = Create(Record.new(zone, 'b', {
     'type': 'CNAME',
     'value': 'foo.unit.tests.'
 }, simple))
+create2 = Create(Record.new(zone, 'c', {
+    'ttl': 60,
+    'type': 'CNAME',
+    'value': 'foo.unit.tests.'
+}))
 update = Update(existing, new)
 delete = Delete(new)
-changes = [create, delete, update]
+changes = [create, create2, delete, update]
 plans = [
     (simple, Plan(zone, zone, changes, True)),
-    (simple, Plan(zone, zone, changes, True)),
+    (simple, Plan(zone, zone, changes, False)),
 ]
 
 
+class TestPlanLogger(TestCase):
+
+    def test_invalid_level(self):
+        with self.assertRaises(Exception) as ctx:
+            PlanLogger('invalid', 'not-a-level')
+        self.assertEquals('Unsupported level: not-a-level',
+                          ctx.exception.message)
+
+    def test_create(self):
+
+        class MockLogger(object):
+
+            def __init__(self):
+                self.out = StringIO()
+
+            def log(self, level, msg):
+                self.out.write(msg)
+
+        log = MockLogger()
+        PlanLogger('logger').run(log, plans)
+        out = log.out.getvalue()
+        self.assertTrue('Summary: Creates=2, Updates=1, '
+                        'Deletes=1, Existing Records=0' in out)
+
+
 class TestPlanHtml(TestCase):
     log = getLogger('TestPlanHtml')
 
@@ -69,7 +90,7 @@ class TestPlanHtml(TestCase):
         out = StringIO()
         PlanHtml('html').run(plans, fh=out)
         out = out.getvalue()
-        self.assertTrue('    <td colspan=6>Summary: Creates=1, Updates=1, '
+        self.assertTrue('    <td colspan=6>Summary: Creates=2, Updates=1, '
                         'Deletes=1, Existing Records=0</td>' in out)