PrivacyScanData/Target/Azure/test_azure_cosmos_10.py

def test_insert_documents(self, mock_cosmos):
        test_id1 = str(uuid.uuid4())
        test_id2 = str(uuid.uuid4())
        test_id3 = str(uuid.uuid4())
        documents = [
            {'id': test_id1, 'data': 'data1'},
            {'id': test_id2, 'data': 'data2'},
            {'id': test_id3, 'data': 'data3'},
        ]

        hook = AzureCosmosDBHook(azure_cosmos_conn_id='azure_cosmos_test_key_id')
        returned_item = hook.insert_documents(documents)
        expected_calls = [
            mock.call()
            .get_database_client('test_database_name')
            .get_container_client('test_collection_name')
            .create_item({'data': 'data1', 'id': test_id1}),
            mock.call()
            .get_database_client('test_database_name')
            .get_container_client('test_collection_name')
            .create_item({'data': 'data2', 'id': test_id2}),
            mock.call()
            .get_database_client('test_database_name')
            .get_container_client('test_collection_name')
            .create_item({'data': 'data3', 'id': test_id3}),
        ]
        logging.getLogger().info(returned_item)
        mock_cosmos.assert_any_call(self.test_end_point, {'masterKey': self.test_master_key})
        mock_cosmos.assert_has_calls(expected_calls, any_order=True)