コード例 #1
0
    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))
コード例 #2
0
    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))
コード例 #3
0
    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))
コード例 #4
0
    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))
コード例 #5
0
    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))
コード例 #6
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)
コード例 #7
0
    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()
コード例 #8
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)
コード例 #9
0
    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()
コード例 #10
0
    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))