def test_shard(self):
"""Tests internal counter structure. """
name = 'testcounter'
shards = ndb.get_multi(counters.ShardConfig.all_keys(name))
counters.increment(name)
config = counters.ShardConfig.get_by_id(name)
self.assertTrue(config is not None)
shards = ndb.get_multi(counters.ShardConfig.all_keys(name))
self.assertTrue(len(shards) > 1) # this will be gt 1 but all but 1 are None
shards = filter(None, shards)
self.assertEqual(len(shards), 1) # there should be only 1 shard since only one increment
def test_many_increment(self):
name = 'testcounter'
for i in range(0, 100):
counters.increment(name)
self.assertEqual(counters.get_count(name), 100)
def test_increment(self):
"""Happy path """
counters.increment('testcounter')
self.assertEqual(counters.get_count('testcounter'), 1)
counters.increment('testcounter', delta=5)
self.assertEqual(counters.get_count('testcounter'), 6)
