From 75aebc5b24af6fff0f8c23aeab3272bbd6a63700 Mon Sep 17 00:00:00 2001
From: Alan Hamlett <alan.hamlett@gmail.com>
Date: Thu, 23 Nov 2017 10:00:01 -0800
Subject: [PATCH] test exceptions handled on commit and close

---
 tests/test_offlinequeue.py | 85 +++++++++++++++++++++++++++++++++++---
 1 file changed, 79 insertions(+), 6 deletions(-)

diff --git a/tests/test_offlinequeue.py b/tests/test_offlinequeue.py
index 0c96c18..16c898f 100644
--- a/tests/test_offlinequeue.py
+++ b/tests/test_offlinequeue.py
@@ -526,7 +526,7 @@ class OfflineQueueTestCase(TestCase):
                 self.assertEquals(None, saved_heartbeat)
 
     @log_capture()
-    def test_push_handles_exception_on_query(self, logs):
+    def test_push_handles_exception_on_execute(self, logs):
         logging.disable(logging.NOTSET)
 
         with NamedTemporaryFile() as fh:
@@ -541,14 +541,87 @@ class OfflineQueueTestCase(TestCase):
                 entity = 'tests/samples/codefiles/twolinefile.txt'
                 config = 'tests/samples/configs/good_config.cfg'
 
-                mock_exec = mock.MagicMock()
-                mock_exec.execute.side_effect = sqlite3.Error('')
-
                 mock_conn = mock.MagicMock()
-                mock_conn.execute.side_effect = sqlite3.Error('')
+
+                mock_c = mock.MagicMock()
+                mock_c.execute.side_effect = sqlite3.Error('')
 
                 with mock.patch('wakatime.offlinequeue.Queue.connect') as mock_connect:
-                    mock_connect.return_value = (mock_conn, mock_exec)
+                    mock_connect.return_value = (mock_conn, mock_c)
+
+                    args = ['--file', entity, '--config', config, '--time', now]
+                    execute(args)
+
+                    response.status_code = 201
+                    execute(args)
+
+                queue = Queue(None, None)
+                saved_heartbeat = queue.pop()
+                self.assertEquals(None, saved_heartbeat)
+
+                self.assertNothingPrinted()
+
+    @log_capture()
+    def test_push_handles_exception_on_commit(self, logs):
+        logging.disable(logging.NOTSET)
+
+        with NamedTemporaryFile() as fh:
+            with mock.patch('wakatime.offlinequeue.Queue._get_db_file') as mock_db_file:
+                mock_db_file.return_value = fh.name
+
+                response = Response()
+                response.status_code = 500
+                self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
+
+                now = u(int(time.time()))
+                entity = 'tests/samples/codefiles/twolinefile.txt'
+                config = 'tests/samples/configs/good_config.cfg'
+
+                mock_conn = mock.MagicMock()
+                mock_conn.commit.side_effect = sqlite3.Error('')
+
+                mock_c = mock.MagicMock()
+                mock_c.fetchone.return_value = None
+
+                with mock.patch('wakatime.offlinequeue.Queue.connect') as mock_connect:
+                    mock_connect.return_value = (mock_conn, mock_c)
+
+                    args = ['--file', entity, '--config', config, '--time', now]
+                    execute(args)
+
+                    response.status_code = 201
+                    execute(args)
+
+                queue = Queue(None, None)
+                saved_heartbeat = queue.pop()
+                self.assertEquals(None, saved_heartbeat)
+
+                self.assertNothingPrinted()
+
+    @log_capture()
+    def test_push_handles_exception_on_close(self, logs):
+        logging.disable(logging.NOTSET)
+
+        with NamedTemporaryFile() as fh:
+            with mock.patch('wakatime.offlinequeue.Queue._get_db_file') as mock_db_file:
+                mock_db_file.return_value = fh.name
+
+                response = Response()
+                response.status_code = 500
+                self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
+
+                now = u(int(time.time()))
+                entity = 'tests/samples/codefiles/twolinefile.txt'
+                config = 'tests/samples/configs/good_config.cfg'
+
+                mock_conn = mock.MagicMock()
+                mock_conn.close.side_effect = sqlite3.Error('')
+
+                mock_c = mock.MagicMock()
+                mock_c.fetchone.return_value = None
+
+                with mock.patch('wakatime.offlinequeue.Queue.connect') as mock_connect:
+                    mock_connect.return_value = (mock_conn, mock_c)
 
                     args = ['--file', entity, '--config', config, '--time', now]
                     execute(args)