Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_init(self):
# Test compress_payload setting
t = ThreadStats(compress_payload=True)
t.start()
assert t.reporter.compress_payload is True
t.stop()
# Default value
t = ThreadStats()
t.start()
assert t.reporter.compress_payload is False
t.stop()
def __init__(self, api_key, app_key, flush_interval=10,
namespace="aplt"):
datadog.initialize(api_key=api_key, app_key=app_key)
self._client = ThreadStats()
self._flush_interval = flush_interval
self._host = get_hostname()
self._namespace = namespace
def test_tags(self):
dog = ThreadStats()
dog.start(roll_up_interval=10, flush_in_thread=False)
reporter = dog.reporter = MemoryReporter()
# Post the same metric with different tags.
dog.gauge('gauge', 10, timestamp=100.0)
dog.gauge('gauge', 15, timestamp=100.0, tags=['env:production', 'db'])
dog.gauge('gauge', 20, timestamp=100.0, tags=['env:staging'])
dog.increment('counter', timestamp=100.0)
dog.increment('counter', timestamp=100.0, tags=['env:production', 'db'])
dog.increment('counter', timestamp=100.0, tags=['env:staging'])
dog.flush(200.0)
metrics = self.sort_metrics(reporter.metrics)
assert_equal(len(metrics), 6)
def test_histogram_percentiles(self):
dog = ThreadStats()
dog.start(roll_up_interval=10, flush_in_thread=False)
reporter = dog.reporter = MemoryReporter()
# Sample all numbers between 1-100 many times. This
# means our percentiles should be relatively close to themselves.
percentiles = list(range(100))
random.shuffle(percentiles) # in place
for i in percentiles:
for j in range(20):
dog.histogram('percentiles', i, 1000.0)
dog.flush(2000.0)
metrics = reporter.metrics
def assert_almost_equal(i, j, e=1):
# Floating point math?
assert abs(i - j) <= e, "%s %s %s" % (i, j, e)
assert_equal(len(metrics), 8)
def test_histogram(self):
dog = ThreadStats()
dog.start(roll_up_interval=10, flush_in_thread=False)
reporter = dog.reporter = MemoryReporter()
# Add some histogram metrics.
dog.histogram('histogram.1', 20, 100.0)
dog.histogram('histogram.1', 30, 105.0)
dog.histogram('histogram.1', 40, 106.0)
dog.histogram('histogram.1', 50, 106.0)
dog.histogram('histogram.1', 30, 110.0)
dog.histogram('histogram.1', 50, 115.0)
dog.histogram('histogram.1', 40, 116.0)
dog.histogram('histogram.2', 40, 100.0)
dog.histogram('histogram.3', 50, 134.0)
def test_default_host_and_device(self):
dog = ThreadStats()
dog.start(roll_up_interval=1, flush_in_thread=False)
reporter = dog.reporter = MemoryReporter()
dog.gauge('my.gauge', 1, 100.0)
dog.flush(1000)
metric = reporter.metrics[0]
assert not metric['device']
assert not metric['host']
def test_distribution(self):
# Create some fake metrics.
dog = ThreadStats()
dog.start(roll_up_interval=10, flush_in_thread=False)
reporter = dog.reporter = MemoryReporter()
dog.distribution('test.dist.1', 20, 100.0)
dog.distribution('test.dist.1', 22, 105.0)
dog.distribution('test.dist.2', 30, 115.0)
dog.distribution('test.dist.3', 30, 125.0)
dog.flush(120.0)
# Assert they've been properly flushed.
dists = self.sort_metrics(reporter.distributions)
assert_equal(len(dists), 2)
(first, second) = dists
assert_equal(first['metric'], 'test.dist.1')
assert_equal(first['points'][0][0], 100.0)
taskdb_kwargs = dict(
host=args.host,
port=args.port,
dbname=args.dbname,
schema=args.schema,
user=args.user,
)
logging.info("Task db config: ")
todo_list = papertasks.TaskDB(
password=args.password,
**taskdb_kwargs
)
# start datadog
datadog.initialize(api_key=os.environ.get("DATADOG_API_KEY"))
stats = datadog.ThreadStats()
stats.start()
datadog_prefix = args.host.split(".")[0]
if datadog_prefix.startswith("spv2-"):
datadog_prefix = datadog_prefix[5:]
datadog_prefix = "spv2.%s." % datadog_prefix
logging.info("Loading model settings ...")
model_settings = settings.default_model_settings
logging.info("Loading token statistics ...")
token_stats = dataprep2.TokenStatistics("model/all.tokenstats3.gz")
logging.info("Loading embeddings ...")
embeddings = dataprep2.CombinedEmbeddings(
token_stats,
dataprep2.GloveVectors(model_settings.glove_vectors),
def stats(self):
instance = ThreadStats()
instance.start()
return instance
cogs.misc.Misc(self),
cogs.characters.Characters(self),
cogs.pets.Pets(self),
cogs.groups.Groups(self),
cogs.user.User(self),
cogs.salary.Salary(self),
cogs.map.Mapping(self),
cogs.backups.Backups(self),
]
for cog in icogs:
self.add_cog(cog)
# self.loop.create_task(self.start_serv())
init_dd(self._auth[3], self._auth[4])
self.stats = ThreadStats()
self.stats.start()
self._first = True