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_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_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)
