Adds unittest.

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