commit 8684b9287f0f8154c6397f83ebd7824f7dec01e5
author Dan Shi <dshi@chromium.org> Tue Oct 06 13:29:18 2015 -0700
committer chrome-bot <chrome-bot@chromium.org> Tue Oct 06 17:15:31 2015 -0700
tree d28f7084164d342f4536e18bb7763dd9f4523815
parent 6e29a2d722d1f1847ff6599bf9c8d062e202b64e

[autotest] Increase timeout for connecting TKO db.

Also log stats to counter tko_db_con_error if connection attempt timed out.

BUG=chromium:539984
TEST=None

Change-Id: I228493e83297b9894c285b472052b88a44864a1e
Reviewed-on: https://chromium-review.googlesource.com/304351
Commit-Ready: Dan Shi <dshi@chromium.org>
Tested-by: Dan Shi <dshi@chromium.org>
Reviewed-by: Fang Deng <fdeng@chromium.org>
Reviewed-by: Dan Shi <dshi@chromium.org>

tko/db.py

diff --git a/tko/db.py b/tko/db.py
index 4c611f2..5c06475 100644
--- a/tko/db.py
+++ b/tko/db.py
@@ -106,9 +106,13 @@
             self.con.close()
             self.con = None
 
-        # create the db connection and cursor
-        self.con = self.connect(self.host, self.database,
-                                self.user, self.password, self.port)
+        try:
+            # create the db connection and cursor
+            self.con = self.connect(self.host, self.database,
+                                    self.user, self.password, self.port)
+        except:
+            autotest_stats.Counter('tko_db_con_error').increment()
+            raise
         self.cur = self.con.cursor()