def test_base_save(tmpdb, tmpdb_reverse, section, orig_data): db = data.DataStore(tmpdb) db.save_to_file(tmpdb_reverse) with shelve.open(tmpdb_reverse) as saved_db: d = saved_db[section] assert d == orig_data
def test_data_object():
'''Test data abstraction layer.'''
dyn = boto3.client('dynamodb')
for table in ('pending', 'users'):
dyn.create_table(
TableName='%s_%s' % (APPNAME, table),
AttributeDefinitions=[
{
'AttributeName': 'email',
'AttributeType': 'S'
},
],
KeySchema=[{
'AttributeName': 'email',
'KeyType': 'S'
}],
ProvisionedThroughput={
'ReadCapacityUnits': 5,
'WriteCapacityUnits': 5
},
)
data_obj = data.DataStore()
assert data_obj.get_validated_users() == {'response': []}
assert data_obj.get_pending_users() == {'response': []}
def test_base_load(tmpdb, tmpdb_reverse, section, orig_data): db = data.DataStore(tmpdb) db.load_from_file(tmpdb_reverse) d = db.get_dump(section) assert d == orig_data
def test_base_clear(tmpdb):
db = data.DataStore(tmpdb)
db.clear(['first'])
d = db.get_dump('second')
assert d == basetest.data1
def test_base_keys(tmpdb, key, section, orig_data): db = data.DataStore(tmpdb) key_object = db.get_key(section, key) assert key_object == orig_data[key]
def test_base_reset(tmpdb):
db = data.DataStore(tmpdb)
db.reset('second')
d = db.get_dump('second')
assert d == basetest.data1
def test_base_init(tmpdb, section, orig_data): db = data.DataStore(tmpdb) d = db.get_dump(section) assert d == orig_data
def test_base_creation(tmpfile, section, orig_data):
db = data.DataStore(tmpfile, ddate={'first': basetest.data1, 'second': basetest.data2}, dsection='first')
d = db.db[section]
assert d == orig_data
