diff --git a/tests/config/dynamic.tests.yaml b/tests/config/dynamic.tests.yaml index d4c63e0..d4855c8 100644 --- a/tests/config/dynamic.tests.yaml +++ b/tests/config/dynamic.tests.yaml @@ -58,9 +58,23 @@ cname: - NA-US-OR - NA-US-WA pools: - 25: iad - 75: sea + 12: sea + 250: iad - default: - pool: iad + 1: sea + 4: iad type: CNAME value: target.unit.tests. +simple-weighted: + dynamic: + pools: + one: + one.unit.tests. + two: + two.unit.tests. + rules: + - default: + 100: one + 200: two + type: CNAME + value: default.unit.tests. diff --git a/tests/test_octodns_provider_yaml.py b/tests/test_octodns_provider_yaml.py index 02e85b2..a787a29 100644 --- a/tests/test_octodns_provider_yaml.py +++ b/tests/test_octodns_provider_yaml.py @@ -34,7 +34,7 @@ class TestYamlProvider(TestCase): self.assertEquals(18, len(zone.records)) source.populate(dynamic_zone) - self.assertEquals(2, len(dynamic_zone.records)) + self.assertEquals(3, len(dynamic_zone.records)) # Assumption here is that a clean round-trip means that everything # worked as expected, data that went in came back out and could be @@ -64,11 +64,11 @@ class TestYamlProvider(TestCase): # Dynamic plan plan = target.plan(dynamic_zone) - self.assertEquals(2, len(filter(lambda c: isinstance(c, Create), + self.assertEquals(3, len(filter(lambda c: isinstance(c, Create), plan.changes))) self.assertFalse(isfile(dynamic_yaml_file)) # Apply it - self.assertEquals(2, target.apply(plan)) + self.assertEquals(3, target.apply(plan)) self.assertTrue(isfile(dynamic_yaml_file)) # There should be no changes after the round trip @@ -127,6 +127,10 @@ class TestYamlProvider(TestCase): self.assertTrue('value' in dyna) # self.assertTrue('dynamic' in dyna) + dyna = data.pop('simple-weighted') + self.assertTrue('value' in dyna) + # self.assertTrue('dynamic' in dyna) + # make sure nothing is left self.assertEquals([], data.keys())