Actually just fix the test

Author: hssyoo

awscli/telemetry.py

@@ -111,8 +111,8 @@ def _create_host_id_table(self):
 
     def _ensure_host_id(self):
         cur = self.execute(self._CHECK_HOST_ID)
-        host_id_ct = cur.fetchone()
-        if host_id_ct and host_id_ct[0] == 0:
+        host_id_ct = cur.fetchone()[0]
+        if host_id_ct == 0:
             self.execute(
                 self._INSERT_HOST_ID,
                 # Hardcode `0` as primary key to ensure

tests/functional/test_telemetry.py

@@ -108,20 +108,25 @@ def test_ensure_database_setup(self, session_conn):
         assert cursor.fetchall() == [('session',), ('host_id',)]
 
     def test_timeout_does_not_raise_exception(self, session_conn):
+        test_query = """
+                SELECT name
+                FROM sqlite_master
+                WHERE type='table'
+                AND name='session';
+        """
+
         class FakeConnection(sqlite3.Connection):
             def execute(self, query, *parameters):
                 # Simulate timeout by always raising.
-                raise sqlite3.OperationalError()
+                if query == test_query:
+                    raise sqlite3.OperationalError()
+                # Mock host id count query.
+                cur = MagicMock()
+                cur.fetchone.return_value = (1,)
+                return cur
 
         fake_conn = CLISessionDatabaseConnection(FakeConnection(":memory:"))
-        cursor = fake_conn.execute(
-            """
-                SELECT name
-                FROM sqlite_master
-                WHERE type='table'
-                AND name='session';
-            """
-        )
+        cursor = fake_conn.execute(test_query)
         assert cursor.fetchall() == []