Use python's unittest framework.

tools/chrome_proxy/webdriver/simple_smoke.py

 # Copyright 2016 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
+import sys
 import time
+import unittest
 
-from common import IntegrationTest
+import common
 from common import TestDriver
 
 
-class SimpleSmoke(IntegrationTest):
+class SimpleSmoke(unittest.TestCase):
 
   # Simple example integration test.
-  def TestCheckPageWithProxy(self):
+  def testCheckPageWithProxy(self):
     with TestDriver() as t:
       t.AddChromeArg('--enable-spdy-proxy-auth')
       t.SetURL('http://check.googlezip.net/test.html')
       t.LoadPage()
       print 'Document Title: ', t.ExecuteJavascriptStatement('document.title',
         timeout=1)
       time.sleep(5)
       responses = t.GetHTTPResponses()
       for response in responses:
         print "URL: %s, ViaHeader: %s, XHR: %s" % (response.url,
           response.ResponseHasViaHeader(), response.WasXHR())
 
   # Show how to get a histogram.
-  def TestPingbackHistogram(self):
+  def testPingbackHistogram(self):
     with TestDriver() as t:
       t.AddChromeArg('--enable-spdy-proxy-auth')
       t.SetURL('http://check.googlezip.net/test.html')
       t.LoadPage()
       t.LoadPage()
       print t.GetHistogram('DataReductionProxy.Pingback.Attempted')
 
 if __name__ == '__main__':
-  test = SimpleSmoke()
-  test.RunAllTests()
+  # Give a different argv so that unittest won't get in the way of our argparse.

[sclittle, 2016/12/08 23:14:00]

nit: pardon my ignorance, but I don't know what this means :) could you explain
it a bit more perhaps?

Is it that sys.argv has a bunch of our integration-test specific args and we
don't want to pass those to unittest.main?

[robertogden, 2016/12/08 23:19:30]

How's that?

+  unittest.main(argv=[sys.argv[0]], verbosity=2)