Index: appengine_apps/trooper_o_matic/appengine_module/trooper_o_matic/test/controller_test.py |
diff --git a/appengine_apps/trooper_o_matic/appengine_module/trooper_o_matic/test/controller_test.py b/appengine_apps/trooper_o_matic/appengine_module/trooper_o_matic/test/controller_test.py |
deleted file mode 100644 |
index 8ad7c1528151e216b4616edef4577f055fbde958..0000000000000000000000000000000000000000 |
--- a/appengine_apps/trooper_o_matic/appengine_module/trooper_o_matic/test/controller_test.py |
+++ /dev/null |
@@ -1,81 +0,0 @@ |
-# Copyright (c) 2014 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 datetime |
-import json |
-import unittest |
-import webapp2 |
-import webtest |
- |
-import endpoints |
-from protorpc import protojson |
- |
-from appengine_module.trooper_o_matic import controller |
-from appengine_module.trooper_o_matic import cron |
-from appengine_module.trooper_o_matic import models |
-from appengine_module.trooper_o_matic import trooper_o_api |
-from appengine_module.trooper_o_matic.test import testing_common |
-from appengine_module.trooper_o_matic.test import test_data |
-from appengine_module.testing_utils import testing |
- |
- |
-def MockNow(): |
- return datetime.datetime(2014, 1, 1, 12) |
- |
- |
-class CoverageTest(testing.AppengineTestCase): |
- |
- def setUp(self): # pylint: disable=E1002 |
- super(CoverageTest, self).setUp() |
- # restricted=False is needed for testing. |
- # Initialized here because setUp() has to be run first. |
- self.app_module = endpoints.api_server( |
- [trooper_o_api.TrooperOMaticAPI], |
- restricted=False |
- ) |
- |
- testing_common.StubUrlfetch(test_data.URLFETCH_RESPONSES, |
- stub=self.testbed.get_stub('urlfetch')) |
- |
- cron.datetime_now = MockNow |
- |
- |
- def _make_api_call(self, method, params=None, status=None): |
- params = params or {} |
- return self.test_app.post_json( |
- '/_ah/spi/TrooperOMaticAPI.%s' % method, |
- params=params, |
- status=status, |
- ) |
- |
- def testCqStats(self): |
- project = 'chromium' |
- cron_app = webtest.TestApp( |
- webapp2.WSGIApplication([ |
- ('/check-cq', cron.CheckCQHandler), |
- ]) |
- ) |
- cron_app.get('/check-cq') |
- cq_data = self._make_api_call( |
- 'cq_stats_get', |
- params={'project': project}, |
- ).json |
- |
- generated = {} |
- for name, klass in (('single_run_data', models.CqStat), |
- ('queue_time_data', models.CqTimeInQueueForPatchStat), |
- ('total_time_data', models.CqTotalTimeForPatchStat)): |
- generated[name] = [ |
- protojson.decode_message(klass.ProtoModel(), json.dumps(x)) |
- for x in cq_data[name]] |
- |
- expected = controller.get_cq_stats(project) |
- for key in expected: |
- expected[key] = [x.ToMessage() for x in expected[key]] |
- |
- self.assertEqual(generated, expected) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |