Combined test.engine.telemetry tests into 2 classes

Author: chillaq

tests/engine/test_telemetry.py

@@ -7,7 +7,7 @@
 class TelemetryStorageProducerTests(object):
     """TelemetryStorageProducer test."""
 
-    def test_instances(self):
+    def test_producer_instances(self):
         telemetry_storage = InMemoryTelemetryStorage()
         telemetry_producer = TelemetryStorageProducer(telemetry_storage)
 
@@ -19,9 +19,6 @@ def test_instances(self):
         assert(telemetry_producer._telemetry_init_producer == telemetry_producer.get_telemetry_init_producer())
         assert(telemetry_producer._telemetry_runtime_producer == telemetry_producer.get_telemetry_runtime_producer())
 
-class TelemetryInitProducerTest(object):
-    """TelemetryInitProducer test."""
-
     def test_record_config(self, mocker):
         telemetry_storage = mocker.Mock()
         telemetry_init_producer = TelemetryInitProducer(telemetry_storage)
@@ -30,7 +27,7 @@ def record_config(*args, **kwargs):
             self.passed_config = args[0]
 
         telemetry_storage.record_config.side_effect = record_config
-        telemetry_init_producer.record_config({'bT':0, 'nR':0, 'uC': 0})
+        telemetry_init_producer.record_config({'bT':0, 'nR':0, 'uC': 0}, {})
         assert(self.passed_config == {'bT':0, 'nR':0, 'uC': 0})
 
     def test_record_ready_time(self, mocker):
@@ -58,9 +55,6 @@ def test_record_not_ready_usage(self, mocker):
         telemetry_init_producer.record_not_ready_usage()
         assert(mocker.called)
 
-class TelemetryEvaluationProducerTest(object):
-    """Telemetry evaluation producer test class."""
-
     def test_record_latency(self, mocker):
         telemetry_storage = mocker.Mock()
         telemetry_evaluation_producer = TelemetryEvaluationProducer(telemetry_storage)
@@ -84,10 +78,6 @@ def record_exception(*args, **kwargs):
         telemetry_evaluation_producer.record_exception('method')
         assert(self.passed_method == 'method')
 
-
-class TelemetryRuntimeProducerTest(object):
-    """Telemetry runtime producer test."""
-
     def test_add_tag(self, mocker):
         telemetry_storage = mocker.Mock()
         telemetry_runtime_producer = TelemetryRuntimeProducer(telemetry_storage)
@@ -210,9 +200,6 @@ def test_instances(self):
         assert(telemetry_consumer._telemetry_init_consumer == telemetry_consumer.get_telemetry_init_consumer())
         assert(telemetry_consumer._telemetry_runtime_consumer == telemetry_consumer.get_telemetry_runtime_consumer())
 
-class TelemetryInitConsumerTest(object):
-    """TelemetryInitConsumer test."""
-
     @mock.patch('splitio.storage.inmemmory.InMemoryTelemetryStorage.get_bur_time_outs')
     def test_get_bur_time_outs(self, mocker):
         telemetry_storage = InMemoryTelemetryStorage()
@@ -234,9 +221,6 @@ def get_not_ready_usage(self, mocker):
         telemetry_init_consumer.get_config_stats()
         assert(mocker.called)
 
-class TelemetryEvaluationConsumerTest(object):
-    """TelemetryEvaluationConsumer test."""
-
     @mock.patch('splitio.storage.inmemmory.InMemoryTelemetryStorage.pop_exceptions')
     def pop_exceptions(self, mocker):
         telemetry_storage = InMemoryTelemetryStorage()
@@ -251,9 +235,6 @@ def pop_latencies(self, mocker):
         telemetry_evaluation_consumer.pop_latencies()
         assert(mocker.called)
 
-class TelemetryRuntimeConsumerTest(object):
-    """TelemetryRuntimeConsumer test."""
-
     def test_get_impressions_stats(self, mocker):
         telemetry_storage = mocker.Mock()
         telemetry_runtime_consumer = TelemetryRuntimeConsumer(telemetry_storage)