def test_get_stats(self, mock_query):
stats_query_mock = mock.MagicMock()
mock_query.return_value = stats_query_mock
health_daemon.get_stats('TEST')
stats_query_mock.show_stat.assert_called_once_with()
stats_query_mock.get_pool_status.assert_called_once_with()
def test_get_stats(self, mock_query):
stats_query_mock = mock.MagicMock()
mock_query.return_value = stats_query_mock
health_daemon.get_stats('TEST')
stats_query_mock.show_stat.assert_called_once_with()
stats_query_mock.get_pool_status.assert_called_once_with()