def test_run_slave(self, mock_wait, mock_get_options, mock_run_service, mock_init): rmq_slave_options = "-m SERVICE_NAME", "-c 1" mock_wait.return_value = True mock_get_options.return_value = rmq_slave_options mock_init.return_value = None init = PreInitConfig() init.run_slave() self.assertEqual(mock_wait.call_count, 1) self.assertEqual(mock_run_service.call_count, 1) self.assertEqual(mock_run_service.call_args_list[0][0][0], rmq_slave_options) self.assertEqual(mock_get_options.call_count, 1)
def test_run_slave_as_master(self, mock_wait, mock_change_master, mock_run_service, mock_init, mock_hostname): mock_wait.return_value = False mock_change_master.return_value = "container_hostname" mock_hostname.return_value = "container_hostname" mock_init.return_value = None init = PreInitConfig() init.run_slave() self.assertEqual(mock_wait.call_count, 1) self.assertEqual(mock_hostname.call_count, 1) self.assertEqual(mock_change_master.call_count, 1) self.assertEqual(mock_run_service.call_count, 1) self.assertEqual(mock_run_service.call_args_list[0][0], ())
def test_run_slave_after_switch_master(self, mock_wait, mock_get_options, mock_run_service, mock_init, mock_change_master, mock_hostname): rmq_slave_options = ['-m', 'SERVICE_NAME', '-c', '1'] mock_wait.return_value = False mock_get_options.return_value = rmq_slave_options mock_change_master.return_value = "new_master_service" mock_hostname.return_value = "container_hostname" mock_init.return_value = None init = PreInitConfig() init.run_slave() self.assertEqual(mock_wait.call_count, 1) self.assertEqual(mock_hostname.call_count, 1) self.assertEqual(mock_change_master.call_count, 1) self.assertEqual(mock_run_service.call_count, 1) self.assertEqual(mock_run_service.call_args_list[0][0][0], rmq_slave_options) self.assertEqual(mock_get_options.call_count, 1)