diff options
Diffstat (limited to 'tests')
| -rwxr-xr-x | tests/google_assistant_test.py | 13 |
1 files changed, 11 insertions, 2 deletions
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() |