def setUp(self): # set up some test variables self.test_end_point = 'https://test_endpoint:443' self.test_master_key = 'magic_test_key' self.test_database_name = 'test_database_name' self.test_collection_name = 'test_collection_name' self.test_database_default = 'test_database_default' self.test_collection_default = 'test_collection_default' db.merge_conn( Connection( conn_id='azure_cosmos_test_key_id', conn_type='azure_cosmos', login=self.test_end_point, password=self.test_master_key, extra=json.dumps( { 'database_name': self.test_database_default, 'collection_name': self.test_collection_default, } ), ) )