diff options
| -rw-r--r-- | google_assistant.py | 1 | ||||
| -rwxr-xr-x | 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() |