diff --git a/shipyard_airflow/control/configdocs/configdocs_helper.py b/shipyard_airflow/control/configdocs/configdocs_helper.py index 5fe5580e..5ee6646a 100644 --- a/shipyard_airflow/control/configdocs/configdocs_helper.py +++ b/shipyard_airflow/control/configdocs/configdocs_helper.py @@ -565,7 +565,7 @@ class ConfigdocsHelper(object): Returns the status object representing the deckhand validation results """ - dh_validations = self._get_deckhand_validations(revision_id) + dh_validations = self._get_deckhand_validation_errors(revision_id) error_count = len(dh_validations) return ConfigdocsHelper._format_validations_to_status( dh_validations, error_count) diff --git a/shipyard_airflow/plugins/drydock_operators.py b/shipyard_airflow/plugins/drydock_operators.py index 26a5deb6..f1104502 100644 --- a/shipyard_airflow/plugins/drydock_operators.py +++ b/shipyard_airflow/plugins/drydock_operators.py @@ -231,6 +231,11 @@ class DryDockOperator(BaseOperator): logging.info('No Action to Perform') @shipyard_service_token + def _auth_gen(self): + # Generator method for the Drydock Session to use to get the + # auth headers necessary + return [('X-Auth-Token', self.svc_token)] + def drydock_session_client(self, drydock_svc_endpoint): # Initialize Variables drydock_url = None @@ -245,7 +250,7 @@ class DryDockOperator(BaseOperator): logging.info("Build DryDock Session") dd_session = session.DrydockSession(drydock_url.hostname, port=drydock_url.port, - token=self.svc_token) + auth_gen=self._auth_gen()) # Raise Exception if we are not able to get a drydock session if dd_session: