Example #1
0
class TestCache(unittest.TestCase):
    def setUp(self):
        self.cache = Cache(AWS_DEFAULT_REGION, TABLE_NAME)

    def testTrue(self):
        assert True

    def testGetCacheKey(self):
        key = self.cache.get_cache_key('/profile', 89078977)
        self.assertEqual(len(key), Cache.MAX_DIGITS)

    def testSetDoc(self):
        self.cache.set_doc('/check', 69097, { 'hello': 'world' } )
        doc = self.cache.get_doc('/check', 69097)

    def testGetDoc(self):
        dico = { 'hello': 'world' }
        self.cache.set_doc('/check', 69987, dico)
        doc = self.cache.get_doc('/check', 69987)
        assert doc is not None
        self.assertEqual(doc, dico)

    def testBase62(self):
        o = self.cache.base62_encode(0)
        self.assertEqual(o,'a')

    def testDocExist(self):
        doc = self.cache.get_doc('/my-way', 45345369987)
        assert doc is None
Example #2
0
class TestCache(unittest.TestCase):
    def setUp(self):
        self.cache = Cache(AWS_DEFAULT_REGION, TABLE_NAME, ENDPOINT_URL,
                           AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
        client = boto3.resource('dynamodb',
                                region_name=AWS_DEFAULT_REGION,
                                endpoint_url=ENDPOINT_URL,
                                aws_access_key_id=AWS_ACCESS_KEY_ID,
                                aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
        client.create_table(AttributeDefinitions=[{
            'AttributeName': 'cache_key',
            'AttributeType': 'S'
        }],
                            TableName=TABLE_NAME,
                            KeySchema=[{
                                'AttributeName': 'cache_key',
                                'KeyType': 'HASH'
                            }],
                            ProvisionedThroughput={
                                'ReadCapacityUnits': 128,
                                'WriteCapacityUnits': 128
                            })

    def tearDown(self):
        client = boto3.resource('dynamodb',
                                region_name=AWS_DEFAULT_REGION,
                                endpoint_url=ENDPOINT_URL,
                                aws_access_key_id=AWS_ACCESS_KEY_ID,
                                aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
        table = client.Table(TABLE_NAME)
        table.delete()

    def testTrue(self):
        assert True

    def testGetCacheKey(self):
        key = self.cache.get_cache_key('/profile', 89078977)
        self.assertEqual(len(key), Cache.MAX_DIGITS)

    def testSetDoc(self):
        self.cache.set_doc('/check', 69097, {'hello': 'world'})
        # doc = self.cache.get_doc('/check', 69097)

    def testGetDoc(self):
        dico = {'hello': 'world'}
        self.cache.set_doc('/check', 69987, dico)
        doc = self.cache.get_doc('/check', 69987)
        assert doc is not None
        self.assertEqual(doc, dico)

    def testBase62(self):
        o = self.cache.base62_encode(0)
        self.assertEqual(o, 'a')

    def testDocExist(self):
        doc = self.cache.get_doc('/my-way', 45345369987)
        assert doc is None