| Index: appengine/findit/model/flake/test/flake_swarming_task_test.py
|
| diff --git a/appengine/findit/model/flake/test/flake_swarming_task_test.py b/appengine/findit/model/flake/test/flake_swarming_task_test.py
|
| index d87db20a22ca19ae23999a230cd53c2b5810c91e..909f5a33bc7ca91d6a5ddb9345ae0c1a7e769fcc 100644
|
| --- a/appengine/findit/model/flake/test/flake_swarming_task_test.py
|
| +++ b/appengine/findit/model/flake/test/flake_swarming_task_test.py
|
| @@ -2,6 +2,8 @@
|
| # Use of this source code is governed by a BSD-style license that can be
|
| # found in the LICENSE file.
|
|
|
| +from datetime import datetime
|
| +
|
| import unittest
|
|
|
| from model.flake.flake_swarming_task import FlakeSwarmingTask
|
| @@ -13,3 +15,34 @@ class FlakeSwarmingTaskTest(unittest.TestCase):
|
| task = FlakeSwarmingTask.Create('m', 'b', 121, 'browser_tests', 'test1')
|
| self.assertEqual('browser_tests', task.step_name)
|
| self.assertEqual('test1', task.test_name)
|
| +
|
| + def testResultsToDict(self):
|
| + master_name = 'm'
|
| + builder_name = 'b'
|
| + build_number = 121
|
| + step_name = 'browser_tests'
|
| + test_name = 'test1'
|
| + created_time = datetime(2016, 9, 26, 0, 0, 0, 0)
|
| + started_time = datetime(2016, 9, 26, 0, 1, 0, 0)
|
| + completed_time = datetime(2016, 9, 26, 0, 2, 0, 0)
|
| + number_of_iterations = 100
|
| + number_of_passes = 100
|
| +
|
| + task = FlakeSwarmingTask.Create(
|
| + master_name, builder_name, build_number, step_name, test_name)
|
| + task.created_time = created_time
|
| + task.started_time = started_time
|
| + task.completed_time = completed_time
|
| + task.tries = number_of_iterations
|
| + task.successes = number_of_passes
|
| +
|
| + expected_dict = {
|
| + 'created_time': created_time,
|
| + 'started_time': started_time,
|
| + 'completed_time': completed_time,
|
| + 'build_number': build_number,
|
| + 'number_of_iterations': number_of_iterations,
|
| + 'number_of_passes': number_of_passes
|
| + }
|
| +
|
| + self.assertEqual(expected_dict, task.ResultsToDict())
|
|
|