def test_publish_failure_intervals(self, _graphite_pub_cls, _periodic_cls): failed_try_count = 5 mock_thread = MagicMock() _periodic_cls.return_value = mock_thread publisher = StatsPublisher(self._mock_db, failed_try_count, 1) publisher._publisher_thread = MagicMock() mock_graphite_publisher = MagicMock() mock_graphite_publisher.publish = MagicMock() mock_graphite_publisher.publish.return_value = False _graphite_pub_cls.return_value = mock_graphite_publisher publisher.configure_publishers() for i in range(failed_try_count): metrics = {"key1": [(1000000, 1), (1000020, 2)]} self._mock_db.get_keys.return_value = metrics.keys() self._mock_db.get_values_since.return_value = metrics["key1"] assert_that(publisher.failed_count, is_(i)) publisher.publish() assert_that(publisher.failed_count, is_(0)) # After fail_count cycle through, it should get incremented on next failure again. publisher.publish() assert_that(publisher.failed_count, is_(1)) # Successful publish should reset the failed_count to 0 mock_graphite_publisher.publish.return_value = True publisher.publish() assert_that(publisher.failed_count, is_(0))
def test_publish(self, _graphite_pub_cls): publisher = StatsPublisher(self._mock_db) mock_graphite_publisher = MagicMock() _graphite_pub_cls.return_value = mock_graphite_publisher publisher.configure_publishers() metrics = {"key1": [(1000000, 1), (1000020, 2)]} self._mock_db.get_keys.return_value = metrics.keys() self._mock_db.get_values_since.return_value = metrics["key1"] publisher.publish() self._mock_db.get_values_since.assert_called_once_with(0, "key1") mock_graphite_publisher.publish.assert_called_once_with(metrics) self._mock_db.get_values_since.reset_mock() mock_graphite_publisher.publish.reset_mock() metrics = {"key1": [(1000040, 5)]} self._mock_db.get_values_since.return_value = metrics["key1"] publisher.publish() # verify retrieval is from last seen time stamp self._mock_db.get_values_since.assert_called_once_with(1000020, "key1") mock_graphite_publisher.publish.assert_called_once_with(metrics)
def test_configure_publishers(self, _graphite_pub_cls): publisher = StatsPublisher(self._mock_db) mock_publisher = MagicMock() _graphite_pub_cls.return_value = mock_publisher publisher.configure_publishers() _graphite_pub_cls.assert_called_once_with(host_id="fake-id", carbon_host='1.1.1.1') assert_that(publisher._publishers, contains(mock_publisher))
def test_configure_publishers(self, _graphite_pub_cls): publisher = StatsPublisher(self._mock_db) mock_publisher = MagicMock() _graphite_pub_cls.return_value = mock_publisher publisher.configure_publishers() _graphite_pub_cls.assert_called_once_with(hostname="hostname", carbon_host='1.1.1.1', carbon_port=1111, host_tags="abc") assert_that(publisher._publishers, contains(mock_publisher))
def test_start_stop_publishing(self, _periodic_cls): mock_thread = MagicMock() _periodic_cls.return_value = mock_thread publisher = StatsPublisher(self._mock_db) publisher.start_publishing() assert_that(mock_thread.daemon, is_(True)) mock_thread.start.assert_called_once_with() _periodic_cls.assert_called_once_with( publisher.publish, StatsPublisher.DEFAULT_PUBLISH_INTERVAL_SECS) publisher.stop_publishing() mock_thread.stop.assert_called_once_with()