def test_client(self, mock_cosmos): hook = AzureCosmosDBHook( azure_cosmos_conn_id='azure_cosmos_test_key_id') self.assertIsNone(hook._conn) self.assertIsInstance(hook.get_conn(), CosmosClient)
def test_client(self, mock_cosmos): hook = AzureCosmosDBHook( azure_cosmos_conn_id='azure_cosmos_test_key_id') assert hook._conn is None assert isinstance(hook.get_conn(), CosmosClient)