def test_configure_collectors(self, _perfmgr_coll_cls): collector = StatsCollector(self._mock_db) mock_collector = MagicMock() _perfmgr_coll_cls.return_value = mock_collector collector.configure_collectors() _perfmgr_coll_cls.assert_called_once_with() assert_that(collector._collectors, contains(mock_collector))
def test_start_stop_collection(self, _periodic_cls): mock_thread = MagicMock() _periodic_cls.return_value = mock_thread collector = StatsCollector(self._mock_db) collector.start_collection() assert_that(mock_thread.daemon, is_(True)) mock_thread.start.assert_called_once_with() _periodic_cls.assert_called_once_with( collector.collect, StatsCollector.DEFAULT_COLLECT_INTERVAL_SECS) collector.stop_collection() mock_thread.stop.assert_called_once_with()
def test_collect(self, _datetime_cls, _perfmgr_coll_cls): mock_now = MagicMock() _datetime_cls.now.return_value = mock_now collector = StatsCollector(self._mock_db) mock_perfmgr_collector = MagicMock() _perfmgr_coll_cls.return_value = mock_perfmgr_collector mock_perfmgr_collector.collect.return_value = { "key1": [(1000000, 1), (1000020, 2)], "key2": [(1000000, 3), (1000020, 4)]} collector.configure_collectors() collector.collect() mock_perfmgr_collector.collect.assert_called_once_with(since=mock_now) calls = [call("key1", 1000000, 1), call("key1", 1000020, 2), call("key2", 1000000, 3), call("key2", 1000020, 4)] self._mock_db.add.assert_has_calls(calls, any_order=True)
def test_collect(self, _datetime_cls, _perfmgr_coll_cls): mock_now = MagicMock() _datetime_cls.now.return_value = mock_now collector = StatsCollector(self._mock_db) mock_perfmgr_collector = MagicMock() _perfmgr_coll_cls.return_value = mock_perfmgr_collector mock_perfmgr_collector.collect.return_value = { "key1": [(1000000, 1), (1000020, 2)], "key2": [(1000000, 3), (1000020, 4)] } collector.configure_collectors() collector.collect() mock_perfmgr_collector.collect.assert_called_once_with(since=mock_now) calls = [ call("key1", 1000000, 1), call("key1", 1000020, 2), call("key2", 1000000, 3), call("key2", 1000020, 4) ] self._mock_db.add.assert_has_calls(calls, any_order=True)