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

diff --git a/tests/google_assistant_test.py b/tests/google_assistant_test.py
new file mode 100755
index 0000000..7699337
--- /dev/null
+++ b/tests/google_assistant_test.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+
+import unittest
+from unittest.mock import MagicMock, patch
+
+import google_assistant
+import unittest_utils  # Needed for --unittests_ignore_perf flag
+
+
+class TestGoogleAssistant(unittest.TestCase):
+    def test_basic_functionality(self):
+
+        with patch('requests.post') as mock:
+            response = MagicMock()
+            response.status_code = 404
+            mock.return_value = response
+            ret = google_assistant.ask_google('What happens with a 404 response?')
+            self.assertFalse(ret.success)
+            self.assertTrue('failed; code 404' in ret.response)
+            self.assertEqual('', ret.audio_transcription)
+            self.assertEqual('', ret.audio_url)
+
+
+if __name__ == '__main__':
+    unittest.main()