More cleanup, yey!

[python_utils.git] / tests / google_assistant_test.py

diff --git a/tests/google_assistant_test.py b/tests/google_assistant_test.py
index 7699337f257a5072205ebeaeddfce940e230eda5..857c1e4d332875dfd294ab397308e4b6e690b941 100755 (executable)
--- 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()