async def test_creation_request_options_era_being_called(self): # pylint: disable=protected-access test_config = CosmosDbConfig( endpoint="https://localhost:8081", masterkey= "C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==", database="test-db", container="bot-storage", database_creation_options={"OfferThroughput": 1000}, container_creation_options={"OfferThroughput": 500}, ) test_id = "1" client = get_mock_client(identifier=test_id) storage = CosmosDbStorage(test_config, client) storage.database = test_id assert storage._get_or_create_database(doc_client=client, id=test_id), test_id client.CreateDatabase.assert_called_with( {"id": test_id}, test_config.database_creation_options) assert storage._get_or_create_container(doc_client=client, container=test_id), test_id client.CreateContainer.assert_called_with( "dbs/" + test_id, {"id": test_id}, test_config.container_creation_options)
async def test_creation_request_options_era_being_called(self): test_config = CosmosDbConfig( endpoint='https://localhost:8081', masterkey= 'C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==', database='test-db', container='bot-storage', database_creation_options={'OfferThroughput': 1000}, container_creation_options={'OfferThroughput': 500}) test_id = '1' client = get_mock_client(id=test_id) storage = CosmosDbStorage(test_config, client) storage.db = test_id assert storage._get_or_create_database(doc_client=client, id=test_id), test_id client.CreateDatabase.assert_called_with( {'id': test_id}, test_config.database_creation_options) assert storage._get_or_create_container(doc_client=client, container=test_id), test_id client.CreateContainer.assert_called_with( 'dbs/' + test_id, {'id': test_id}, test_config.container_creation_options)