def test_institution_fetcher():
db_id = "AzureCosmosDbDatabaseId"
collection_id = "AzureCosmosDbInstitutionsCollectionId"
# Get the relevant properties from Application Settings
collection_link = utils.get_collection_link(db_id, collection_id)
client = utils.get_cosmos_client()
institution_fetcher = InstitutionFetcher(client, collection_link)
institution_id = "10007857"
version = 1
return institution_fetcher.get_institution(institution_id, version)
def test_with_all_keys_to_be_deleted(self):
expected_course = {"version": 1}
input_course = {
"_rid": "_rid_test",
"_self": "_self_test",
"_etag": "_etag_test",
"_attachments": "_attachments_test",
"_ts": "_ts_test",
"institution_id": "111",
"version": 1,
}
output_course = InstitutionFetcher.tidy_institution(input_course)
self.assertEqual(expected_course, output_course)
def test_rid_is_deleted(self):
expected_course = {"version": 1}
input_course = {"_rid": "_rid_test", "version": 1}
output_course = InstitutionFetcher.tidy_institution(input_course)
self.assertEqual(expected_course, output_course)