def test_get_all_documents(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
_docs = datastore.get_all_documents(Type._ADMIN)
expected = MASTER_CONFIG.get(Type._ADMIN.value).keys()
self.assertEqual(expected, _docs)
def test_list_documents_auth(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
_docs = datastore.list_documents(Type._ADMIN, 'auth')
expected = MASTER_CONFIG.get(Type._ADMIN.value).get('auth')
self.assertEqual(expected.keys(), _docs)
def test_delete_document_key_missing(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore=local_datastore.LocalDatastore()
datastore.delete_document(
type = Type._ADMIN, id = 'auth', key = 'foo')
self.assertEqual(MASTER_CONFIG,
datastore.datastore)
def test_store_new_document(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
datastore.store_document(Type.DV360, id='0000', document={'id': '0000'})
expected = deepcopy(MASTER_CONFIG)
expected.update({'dv360': {'0000': {'id': '0000'}}})
self.open.assert_called_with('datastore.json', 'w')
self.open().write.assert_called_with(json.dumps(expected, indent=2))
def test_update_document_existing(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore=local_datastore.LocalDatastore()
expected = {'api_key': 'new api key'}
datastore.update_document(type=Type._ADMIN, id='auth',
new_data=expected)
self.open().write.assert_called_once()
self.assertEqual(expected.get('api_key'),
datastore.datastore.get(Type._ADMIN.value).get('auth').get('api_key'))
def test_delete_document_key(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore=local_datastore.LocalDatastore()
datastore.delete_document(
type = Type._ADMIN, id = 'auth', key = 'api_key')
self.assertEqual({'administration': {
'auth': {'bHVrZUBza3l3YWxrZXIuY29t': {
'_key': '*****@*****.**',
'access_token': 'access_token',
'refresh_token': 'refresh_token'}}}},
datastore.datastore)
def test_list_documents_none(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
_docs = datastore.list_documents(Type.DV360)
self.assertEqual(None, _docs)
def test_get_document_missing_key(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
self.assertEqual(None, datastore.get_document(Type._ADMIN, 'auth', 'foo'))
def test_get_document_missing_type(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
self.assertEqual(None, datastore.get_document(Type.DV360, '10011'))
def test_get_document_without_key(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
self.assertEqual(MASTER_CONFIG.get(Type._ADMIN.value, 'auth'),
datastore.get_document(Type._ADMIN, 'auth'))
def test_get_document_with_key(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
self.assertEqual({'api_key': 'api_key'},
datastore.get_document(Type._ADMIN, 'auth', 'api_key'))
def test_delete_document_collection(self):
with mock.patch(f'{CLASS_UNDER_TEST}.open', self.open):
datastore = local_datastore.LocalDatastore()
datastore.delete_document(type=Type._ADMIN, id='auth')
self.assertEqual({'administration': {}}, datastore.datastore)