From 6e8e3458d9641394b8f061d55ba569a72e6a8493 Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Mon, 7 Feb 2022 10:55:09 -0800 Subject: [PATCH] Adds unittest. --- google_assistant.py | 1 + tests/google_assistant_test.py | 13 +++++++++++-- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/google_assistant.py b/google_assistant.py index ec5f6a4..b0aabf3 100644 --- a/google_assistant.py +++ b/google_assistant.py @@ -70,6 +70,7 @@ def ask_google(cmd: str, *, recognize_speech=True) -> GoogleResponse: audio_transcription: Optional[str] = "" if r.status_code == 200: j = r.json() + logger.debug(j) success = bool(j["success"]) response = j["response"] if success else j["error"] if success: diff --git a/tests/google_assistant_test.py b/tests/google_assistant_test.py index 7699337..857c1e4 100755 --- a/tests/google_assistant_test.py +++ b/tests/google_assistant_test.py @@ -8,8 +8,7 @@ import unittest_utils # Needed for --unittests_ignore_perf flag class TestGoogleAssistant(unittest.TestCase): - def test_basic_functionality(self): - + def test_failure_case(self): with patch('requests.post') as mock: response = MagicMock() response.status_code = 404 @@ -20,6 +19,16 @@ class TestGoogleAssistant(unittest.TestCase): self.assertEqual('', ret.audio_transcription) self.assertEqual('', ret.audio_url) + def test_success_case(self): + with patch('requests.post') as mock: + response = MagicMock() + response.status_code = 200 + json = {'response': 'LGTM', 'audio': '', 'success': True} + response.json = MagicMock(return_value=json) + mock.return_value = response + ret = google_assistant.ask_google('Is this thing working?', recognize_speech=False) + self.assertTrue(ret.success) + if __name__ == '__main__': unittest.main() -- 2.45.2