def test_set_deployment_priorities(self): nodes = [{ 'role': 'mongo' }, { 'role': 'mongo' }, { 'role': 'primary-mongo' }, { 'role': 'controller' }, { 'role': 'ceph-osd' }, { 'role': 'other' }] serializer = DeploymentMultinodeSerializer() serializer.set_deployment_priorities(nodes) expected_priorities = [{ 'role': 'mongo', 'priority': 100 }, { 'role': 'mongo', 'priority': 200 }, { 'role': 'primary-mongo', 'priority': 300 }, { 'role': 'controller', 'priority': 400 }, { 'role': 'ceph-osd', 'priority': 500 }, { 'role': 'other', 'priority': 500 }] self.assertEquals(expected_priorities, nodes)