def test_delete_execute_error_project_id(self, mock_hook): with self.assertRaises(AirflowException): operator = GKEClusterDeleteOperator(location=PROJECT_LOCATION, name=CLUSTER_NAME, task_id=PROJECT_TASK_ID) operator.execute(None) mock_hook.return_value.delete_cluster.assert_not_called()
def test_delete_execute_error_project_id(self, mock_hook): with self.assertRaises(AirflowException): operator = GKEClusterDeleteOperator(location=PROJECT_LOCATION, name=CLUSTER_NAME, task_id=PROJECT_TASK_ID) operator.execute(None) mock_hook.return_value.delete_cluster.assert_not_called()
def test_delete_execute(self, mock_hook): operator = GKEClusterDeleteOperator(project_id=TEST_GCP_PROJECT_ID, name=CLUSTER_NAME, location=PROJECT_LOCATION, task_id=PROJECT_TASK_ID) operator.execute(None) mock_hook.return_value.delete_cluster.assert_called_once_with( name=CLUSTER_NAME, project_id=TEST_GCP_PROJECT_ID)
def test_delete_execute(self, mock_hook): operator = GKEClusterDeleteOperator(project_id=TEST_GCP_PROJECT_ID, name=CLUSTER_NAME, location=PROJECT_LOCATION, task_id=PROJECT_TASK_ID) operator.execute(None) mock_hook.return_value.delete_cluster.assert_called_once_with( name=CLUSTER_NAME)