def test_get_analytics_sample_rate(self): """" Check method for accessing sample rate based on configuration """ ic = IntegrationConfig(self.config, 'foo', analytics_enabled=True, analytics_sample_rate=0.5) self.assertEqual(ic.get_analytics_sample_rate(), 0.5) ic = IntegrationConfig(self.config, 'foo', analytics_enabled=True) self.assertEqual(ic.get_analytics_sample_rate(), 1.0) ic = IntegrationConfig(self.config, 'foo', analytics_enabled=False) self.assertIsNone(ic.get_analytics_sample_rate()) with self.override_env(dict(DD_ANALYTICS_ENABLED='True')): config = Config() ic = IntegrationConfig(config, 'foo') self.assertEqual( ic.get_analytics_sample_rate(use_global_config=True), 1.0) with self.override_env(dict(DD_ANALYTICS_ENABLED='False')): config = Config() ic = IntegrationConfig(config, 'foo') self.assertIsNone( ic.get_analytics_sample_rate(use_global_config=True))
def test_get_analytics_sample_rate_deprecated_name(self): """Check method for accessing sample rate based on configuration""" with self.override_env(dict(DD_FOO_ANALYTICS_ENABLED="True")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertEqual(ic.get_analytics_sample_rate(), 1.0) with self.override_env(dict(DD_TRACE_FOO_ANALYTICS_ENABLED="True")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertEqual(ic.get_analytics_sample_rate(), 1.0) with self.override_env( dict(DD_FOO_ANALYTICS_ENABLED="True", DD_FOO_ANALYTICS_SAMPLE_RATE="0.5")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertEqual(ic.get_analytics_sample_rate(), 0.5) with self.override_env( dict(DD_FOO_ANALYTICS_ENABLED="True", DD_TRACE_FOO_ANALYTICS_SAMPLE_RATE="0.5")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertEqual(ic.get_analytics_sample_rate(), 0.5) with self.override_env(dict(DD_FOO_ANALYTICS_ENABLED="False")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertIsNone(ic.get_analytics_sample_rate()) with self.override_env(dict(DD_TRACE_FOO_ANALYTICS_ENABLED="False")): config = Config() ic = IntegrationConfig(config, "bar", _deprecated_name="foo") self.assertIsNone(ic.get_analytics_sample_rate())