gae_ts_mon: instrument Cloud Endpoint methods

infra_libs/ts_mon/common/test/http_metrics_test.py

Index: infra_libs/ts_mon/common/test/http_metrics_test.py
diff --git a/infra_libs/ts_mon/common/test/http_metrics_test.py b/infra_libs/ts_mon/common/test/http_metrics_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..f7995e0e82a20a8b6e440dc1bcc6146cd1475e83
--- /dev/null
+++ b/infra_libs/ts_mon/common/test/http_metrics_test.py
@@ -0,0 +1,44 @@
+# 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 unittest
+
+import mock
+
+from infra_libs.ts_mon.common import http_metrics
+from infra_libs.ts_mon.common import interface
+from infra_libs.ts_mon.common import targets
+
+
+class TestHttpMetrics(unittest.TestCase):
+  def setUp(self):
+    super(TestHttpMetrics, self).setUp()
+    target = targets.TaskTarget('test_service', 'test_job',
+                                'test_region', 'test_host')
+    self.mock_state = interface.State(target=target)
+    self.state_patcher = mock.patch('infra_libs.ts_mon.common.interface.state',
+                                    new=self.mock_state)
+    self.state_patcher.start()
+
+  def tearDown(self):
+    self.state_patcher.stop()
+    super(TestHttpMetrics, self).tearDown()
+
+  def test_update_http_server_metrics(self):
+    http_metrics.update_http_server_metrics(
+        '/', 200, 125.4,
+        request_size=100, response_size=200, user_agent='Chrome')
+    fields = {'status': 200, 'name': '/', 'is_robot': False}
+    self.assertEqual(1, http_metrics.server_response_status.get(fields))
+    self.assertEqual(125.4, http_metrics.server_durations.get(fields).sum)
+    self.assertEqual(100, http_metrics.server_request_bytes.get(fields).sum)
+    self.assertEqual(200, http_metrics.server_response_bytes.get(fields).sum)
+
+  def test_update_http_server_metrics_no_sizes(self):
+    http_metrics.update_http_server_metrics('/', 200, 125.4)
+    fields = {'status': 200, 'name': '/', 'is_robot': False}
+    self.assertEqual(1, http_metrics.server_response_status.get(fields))
+    self.assertEqual(125.4, http_metrics.server_durations.get(fields).sum)
+    self.assertIsNone(http_metrics.server_request_bytes.get(fields))
+    self.assertIsNone(http_metrics.server_response_bytes.get(fields))