def setUp(self): db.merge_conn( Connection( conn_id='azure_container_instance_test', conn_type='azure_container_instances', login='login', password='key', extra=json.dumps({'tenantId': 'tenant_id', 'subscriptionId': 'subscription_id'}), ) ) self.resources = ResourceRequirements(requests=ResourceRequests(memory_in_gb='4', cpu='1')) with patch( 'azure.common.credentials.ServicePrincipalCredentials.__init__', autospec=True, return_value=None ): with patch('azure.mgmt.containerinstance.ContainerInstanceManagementClient'): self.hook = AzureContainerInstanceHook(conn_id='azure_container_instance_test')