1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- # Licensed to the Apache Software Foundation (ASF) under one
- # or more contributor license agreements. See the NOTICE file
- # distributed with this work for additional information
- # regarding copyright ownership. The ASF licenses this file
- # to you under the Apache License, Version 2.0 (the
- # "License"); you may not use this file except in compliance
- # with the License. You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing,
- # software distributed under the License is distributed on an
- # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
- # KIND, either express or implied. See the License for the
- # specific language governing permissions and limitations
- # under the License.
- """Unit tests for Superset"""
- from unittest import TestCase
- from unittest.mock import Mock, patch
- from superset.stats_logger import StatsdStatsLogger
- class StatsdStatsLoggerTest(TestCase):
- def verify_client_calls(self, logger, client):
- logger.incr("foo1")
- client.incr.assert_called_once()
- client.incr.assert_called_with("foo1")
- logger.decr("foo2")
- client.decr.assert_called_once()
- client.decr.assert_called_with("foo2")
- logger.gauge("foo3")
- client.gauge.assert_called_once()
- client.gauge.assert_called_with("foo3")
- logger.timing("foo4", 1.234)
- client.timing.assert_called_once()
- client.timing.assert_called_with("foo4", 1.234)
- def test_init_with_statsd_client(self):
- client = Mock()
- stats_logger = StatsdStatsLogger(statsd_client=client)
- self.verify_client_calls(stats_logger, client)
- def test_init_with_params(self):
- with patch("superset.stats_logger.StatsClient") as MockStatsdClient:
- mock_client = MockStatsdClient.return_value
- stats_logger = StatsdStatsLogger()
- self.verify_client_calls(stats_logger, mock_client)
|