def test_connect_master_succeed(self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockCheckRule.return_value = True MockBlockOutput.return_value = False player = Player(self.connection, self.exchange, self.routing_key) rtn = player.connect_master() self.assertTrue(rtn)
def test_connect_master_succeed(self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockCheckRule.return_value = True MockBlockOutput.return_value = False player = Player(self.connection, self.exchange, self.routing_key) rtn = player.connect_master() self.assertTrue(rtn)
def test_run_command_not_implemented_with_node_id(self, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockCheckRule.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command('Fake Command', [3]) self.assertFalse(rtn)
def test_disconnect_pgpool_succeed(self, MockConductor, MockCheckBlockInput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockModifyRule.return_value = None MockCheckRule.return_value = False MockCheckBlockInput.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.disconnect_pgpool() self.assertTrue(rtn)
def test_run_command_connect_pgpool_without_nodes(self, MockConductor, MockBlockInput, MockModifyRule, MockCheckRule): MockBlockInput.return_value = False MockModifyRule.return_value = None MockConductor.return_value = lambda: None MockCheckRule.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_CONNECT_PGPOOL, None) self.assertFalse(rtn)
def test_run_command_stop_replication_without_nodes(self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockOutput.return_value = True MockCheckRule.return_value = False MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_STOP_REPLICATION, None) self.assertFalse(rtn)
def test_disconnect_master_failed(self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockCheckRule.return_value = False MockModifyRule.return_value = None MockBlockOutput.return_value = False player = Player(self.connection, self.exchange, self.routing_key) rtn = player.disconnect_master() self.assertFalse(rtn)
def test_reset_players_with_pgpool_failed(self, MockConductor, MockCheckBlockInput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockCheckBlockInput.return_value = True MockCheckRule.return_value = True MockModifyRule.return_value = None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.reset_players() self.assertFalse(rtn)
def test_connect_pgpool_failed(self, MockConductor, MockCheckBlockInput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockCheckRule.return_value = True MockModifyRule.return_value = None MockCheckBlockInput.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.connect_pgpool() self.assertFalse(rtn)
def test_disconnect_master_failed(self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockCheckRule.return_value = False MockModifyRule.return_value = None MockBlockOutput.return_value = False player = Player(self.connection, self.exchange, self.routing_key) rtn = player.disconnect_master() self.assertFalse(rtn)
def test_run_command_not_implemented(self, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockCheckRule.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command('Fake Command', None) self.assertFalse(rtn)
def test_run_command_connect_pgpool_with_node_id_not_in_nodes( self, MockConductor, MockBlockInput, MockModifyRule, MockCheckRule): MockBlockInput.return_value = False MockModifyRule.return_value = None MockCheckRule.return_value = True MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_CONNECT_PGPOOL, []) self.assertFalse(rtn)
def test_run_command_stop_replication_without_nodes( self, MockConductor, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockOutput.return_value = True MockCheckRule.return_value = False MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_STOP_REPLICATION, None) self.assertFalse(rtn)
def test_run_command_reset_players_with_node_id(self, MockConductor, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockCheckRule.return_value = True MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_RESET_PLAYERS, [3]) self.assertTrue(rtn)
def test_reset_players_succeed(self, MockConductor, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockConductor.return_value = lambda: None MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockCheckRule.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.reset_players() self.assertTrue(rtn)
def test_run_command_reset_players(self, MockConductor, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockCheckRule.return_value = True MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_RESET_PLAYERS, None) self.assertTrue(rtn)
def test_run_command_register_player_with_node_id(self, MockConductor, MockBlockInput, MockBlockOutput, MockModifyRule, MockCheckRule): MockModifyRule.return_value = None MockBlockInput.return_value = False MockBlockOutput.return_value = False MockConductor.return_value = lambda: None MockCheckRule.return_value = True player = Player(self.connection, self.exchange, self.routing_key) rtn = player.run_command(Constants.COMMAND_REGISTER_PLAYER, [3]) self.assertTrue(rtn)
def test_iptables_command_exception(self, MockBlockInput, MockBlockOutput, MockCheckRule, MockSubprocess): MockBlockInput.return_value = True MockBlockOutput.return_value = False MockCheckRule.return_value = True MockSubprocess.side_effect = subprocess.CalledProcessError( -1, 'iptables') player = Player(self.connection, self.exchange, self.routing_key) self.assertRaises(IptablesCommandError, player.run_command, Constants.COMMAND_CONNECT_PGPOOL, [3])
def test_register_player_with_no_node_id(self, MockConductor): MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.register_player() self.assertTrue(rtn)
def test__hostname(self): player = Player(self.connection, self.exchange, self.routing_key) self.assertEqual("localhost", player.hostname)
def test__node_id(self): player = Player(self.connection, self.exchange, self.routing_key) self.assertEqual(3, player.node_id)
def test_register_player_with_no_node_id(self, MockConductor): MockConductor.return_value = lambda: None player = Player(self.connection, self.exchange, self.routing_key) rtn = player.register_player() self.assertTrue(rtn)