Chromium Code Reviews| Index: appengine/findit/model/flake/flake_swarming_task.py |
| diff --git a/appengine/findit/model/flake/flake_swarming_task.py b/appengine/findit/model/flake/flake_swarming_task.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..52267c6c9cbd0fa7658a69dca7256b33d2b4ae9a |
| --- /dev/null |
| +++ b/appengine/findit/model/flake/flake_swarming_task.py |
| @@ -0,0 +1,58 @@ |
| +# 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. |
| + |
| +from google.appengine.ext import ndb |
| + |
| +from common import constants |
| +from model import analysis_status |
| +from model.base_swarming_task import BaseSwarmingTask |
| +from model.base_build_model import BaseBuildModel |
| + |
| +class FlakeSwarmingTask(BaseSwarmingTask, BaseBuildModel): |
| + """Represents a swarming task for a step w/candidate flaky tests. |
| + """ |
| + |
| + @staticmethod |
| + def CreateSwarmingTaskId( |
| + master_name, builder_name, build_number, |
| + step_name, test_name): # pragma: no cover |
| + return '%s/%s/%s/%s/%s' % (master_name, builder_name, |
| + build_number, step_name, test_name) |
| + |
| + @staticmethod |
| + def _CreateKey( |
| + master_name, builder_name, build_number, |
| + step_name, test_name): # pragma: no cover |
| + return ndb.Key('FlakeSwarmingTask', |
| + FlakeSwarmingTask.CreateSwarmingTaskId( |
| + master_name, builder_name, build_number, |
| + step_name, test_name)) |
| + |
| + @staticmethod |
| + def Create( |
| + master_name, builder_name, build_number, |
| + step_name, test_name): # pragma: no cover |
| + return FlakeSwarmingTask(key=FlakeSwarmingTask._CreateKey( |
| + master_name, builder_name, build_number, step_name, test_name)) |
| + |
| + @ndb.ComputedProperty |
| + def step_name(self): |
| + return self.key.pairs()[0][1].split('/')[3] |
| + |
| + @ndb.ComputedProperty |
| + def test_name(self): |
| + return self.key.pairs()[0][1].split('/')[4] |
| + |
| + |
| + @staticmethod |
| + def Get( |
| + master_name, builder_name, build_number, |
| + step_name, test_name): # pragma: no cover |
| + return FlakeSwarmingTask._CreateKey( |
| + master_name, builder_name, build_number, step_name, test_name).get( |
|
lijeffrey
2016/07/19 22:32:21
nit: 4 spaces after opening ( if the following stu
caiw
2016/07/20 18:11:00
Done.
|
| +) |
| + # In how many runs did the test succeed? |
|
lijeffrey
2016/07/19 22:32:21
nit: Be explicit what "succeeded" means. Is it the
caiw
2016/07/20 18:11:00
Done.
|
| + successes = ndb.IntegerProperty(default=0, indexed=False) |
| + # How many times did we rerun the test? |
|
lijeffrey
2016/07/19 22:32:21
Nit: How many times the test was rerun
caiw
2016/07/20 18:11:00
Done.
|
| + tries = ndb.IntegerProperty(default=0, indexed=False) |