def test_missing_audio(self, mock_hook): mock_hook.return_value.recognize_speech.return_value = True with self.assertRaises(AirflowException) as e: GcpSpeechToTextRecognizeSpeechOperator( project_id=PROJECT_ID, gcp_conn_id=GCP_CONN_ID, config=CONFIG, task_id="id" ).execute(context={"task_instance": mock.Mock()}) err = e.exception self.assertIn("audio", str(err)) mock_hook.assert_not_called()
def test_recognize_speech_green_path(self, mock_hook): mock_hook.return_value.recognize_speech.return_value = True GcpSpeechToTextRecognizeSpeechOperator( project_id=PROJECT_ID, gcp_conn_id=GCP_CONN_ID, config=CONFIG, audio=AUDIO, task_id="id" ).execute(context={"task_instance": mock.Mock()}) mock_hook.assert_called_once_with(gcp_conn_id=GCP_CONN_ID) mock_hook.return_value.recognize_speech.assert_called_once_with( config=CONFIG, audio=AUDIO, retry=None, timeout=None )
def test_missing_config(self, mock_hook): mock_hook.return_value.recognize_speech.return_value = True with self.assertRaises(AirflowException) as e: GcpSpeechToTextRecognizeSpeechOperator( # pylint: disable=no-value-for-parameter project_id=PROJECT_ID, gcp_conn_id=GCP_CONN_ID, audio=AUDIO, task_id="id" ).execute(context={"task_instance": Mock()}) err = e.exception self.assertIn("config", str(err)) mock_hook.assert_not_called()
default_args = {"start_date": dates.days_ago(1)} with models.DAG( "example_gcp_speech", default_args=default_args, schedule_interval=None # Override to match your needs ) as dag: # [START howto_operator_text_to_speech_synthesize] text_to_speech_synthesize_task = GcpTextToSpeechSynthesizeOperator( project_id=GCP_PROJECT_ID, input_data=INPUT, voice=VOICE, audio_config=AUDIO_CONFIG, target_bucket_name=BUCKET_NAME, target_filename=FILENAME, task_id="text_to_speech_synthesize_task", ) # [END howto_operator_text_to_speech_synthesize] # [START howto_operator_speech_to_text_recognize] speech_to_text_recognize_task = GcpSpeechToTextRecognizeSpeechOperator( project_id=GCP_PROJECT_ID, config=CONFIG, audio=AUDIO, task_id="speech_to_text_recognize_task") # [END howto_operator_speech_to_text_recognize] text_to_speech_synthesize_task >> speech_to_text_recognize_task