def test_update_slave_last_heartbeat_time_calls_correspondig_slave_method(self, slave_alive, method_call_count): master = ClusterMaster() mock_slave = self.patch('app.master.cluster_master.Slave').return_value mock_slave.is_alive.return_value = slave_alive master.update_slave_last_heartbeat_time(mock_slave) self.assertEqual(mock_slave.update_last_heartbeat_time.call_count, method_call_count, 'last heartbeat time is updated for the target slave')
def test_update_slave_last_heartbeat_time_calls_update_last_heartbeat_time_on_slave( self): master = ClusterMaster() mock_slave = self.patch('app.master.cluster_master.Slave').return_value master.update_slave_last_heartbeat_time(mock_slave) self.assertEqual( mock_slave.update_last_heartbeat_time.call_count, 1, 'last heartbeat time is updated for the target slave')