X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=tests%2Fgoogle_assistant_test.py;h=857c1e4d332875dfd294ab397308e4b6e690b941;hb=89f305d67e913ea1512e2618a0375359ec925ada;hp=7699337f257a5072205ebeaeddfce940e230eda5;hpb=43635064329197b2f9e822d15e7315ac59141207;p=python_utils.git

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()