blob: 6e89ee67df19c05c059f089530497981ec5f0ec2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#!/usr/bin/env python
import os
import unittest
from tests.test_custom_dict import BaseCustomDictTestCase
try:
from requests_cache.backends.dynamodb import DynamoDbDict
except ImportError:
print("DynamoDb not installed")
else:
# boto3 will accept any values for creds, but they still need to be present
os.environ['AWS_ACCESS_KEY_ID'] = 'placeholder'
os.environ['AWS_SECRET_ACCESS_KEY'] = 'placeholder'
class WrapDynamoDbDict(DynamoDbDict):
def __init__(self, namespace, collection_name='dynamodb_dict_data', **options):
options['endpoint_url'] = 'http://0.0.0.0:8000'
super().__init__(namespace, collection_name, **options)
class DynamoDbDictTestCase(BaseCustomDictTestCase, unittest.TestCase):
dict_class = WrapDynamoDbDict
pickled_dict_class = WrapDynamoDbDict
if __name__ == '__main__':
unittest.main()
|