ChromeDriver test replacements for SafebrowserOn/SafebrowserOff

tools/chrome_proxy/webdriver/safebrowsing.py

Index: tools/chrome_proxy/webdriver/safebrowsing.py
diff --git a/tools/chrome_proxy/webdriver/safebrowsing.py b/tools/chrome_proxy/webdriver/safebrowsing.py
new file mode 100644
index 0000000000000000000000000000000000000000..cdfe2ec055b98801ded6574b14d8fbc9f7400f34
--- /dev/null
+++ b/tools/chrome_proxy/webdriver/safebrowsing.py
@@ -0,0 +1,33 @@
+# Copyright 2017 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 common
+from common import TestDriver
+from common import IntegrationTest
+from common import AndroidOnly
+from common import NotAndroid
+
+
+class SafeBrowsing(IntegrationTest):
+
+  @AndroidOnly
+  def testSafeBrowsingOn(self):
+    with TestDriver() as t:
+      t.AddChromeArg('--enable-spdy-proxy-auth')
+      t.LoadURL('http://testsafebrowsing.appspot.com/s/malware.html')
+      responses = t.GetHTTPResponses()
+      self.assertEqual(0, len(responses))
+
+  @NotAndroid
+  def testSafeBrowsingOff(self):
+    with TestDriver() as t:
+      t.AddChromeArg('--enable-spdy-proxy-auth')
+      t.LoadURL('http://testsafebrowsing.appspot.com/s/malware.html')
+      responses = t.GetHTTPResponses()
+      self.assertEqual(1, len(responses))
+      for response in responses:
+        self.assertHasChromeProxyViaHeader(response)
+
+if __name__ == '__main__':
+  IntegrationTest.RunAllTests()