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)
Пример #2
0
 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)