Better weighting support

This commit is contained in:
Ross McFarland
2018-11-28 09:41:25 -08:00
parent 92b025fe1a
commit c41824c3e9
2 changed files with 24 additions and 6 deletions
+17 -3
View File
@@ -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.
+7 -3
View File
@@ -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())