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)
