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'])
