def test_load_topology_not_exists(self, get_conf_mock): e = ConfigFileNotFoundError() get_conf_mock.side_effect = e main.load_topology() self.assertEqual(main.state.env.roledefs, {'coordinator': [], 'worker': [], 'all': []}) self.assertEqual(main.state.env.port, '22') self.assertNotEqual(main.state.env.user, 'user') self.assertEqual(main.state.env.topology_config_not_found, e)
def test_load_topology(self, get_conf_mock): get_conf_mock.return_value = {'username': '******', 'port': '1234', 'coordinator': 'hello', 'workers': ['a', 'b']} main.load_topology() self.assertEqual(main.state.env.roledefs, {'coordinator': ['hello'], 'worker': ['a', 'b'], 'all': ['hello', 'a', 'b']}) self.assertEqual(main.state.env.port, '1234') self.assertEqual(main.state.env.user, 'user') self.assertEqual(main.state.env.hosts, ['hello', 'a', 'b'])
def test_load_topology_not_exists(self, get_conf_mock): e = ConfigFileNotFoundError() get_conf_mock.side_effect = e main.load_topology() self.assertEqual(main.state.env.roledefs, { 'coordinator': [], 'worker': [], 'all': [] }) self.assertEqual(main.state.env.port, '22') self.assertNotEqual(main.state.env.user, 'user') self.assertEqual(main.state.env.topology_config_not_found, e)
def test_load_topology(self, get_conf_mock): get_conf_mock.return_value = { 'username': '******', 'port': '1234', 'coordinator': 'hello', 'workers': ['a', 'b'] } main.load_topology() self.assertEqual( main.state.env.roledefs, { 'coordinator': ['hello'], 'worker': ['a', 'b'], 'all': ['hello', 'a', 'b'] }) self.assertEqual(main.state.env.port, '1234') self.assertEqual(main.state.env.user, 'user') self.assertEqual(main.state.env.hosts, ['hello', 'a', 'b'])