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..5ef9180be45fbe1dc212f2bb2f9f49719280e976 |
| --- /dev/null |
| +++ b/appengine/findit/model/flake/flake_swarming_task.py |
| @@ -0,0 +1,62 @@ |
| +# 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.progress import Progress |
| +from model.base_build_model import BaseBuildModel |
| + |
| +class FlakeSwarmingTask(Progress, 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( |
| +) |
| + # In how many runs did the test succeed? |
| + successes = ndb.IntegerProperty(default=0, indexed=False) |
| + # How many times did we rerun the test? |
| + tries = ndb.IntegerProperty(default=0, indexed=False) |
| + # Identifier for code build. |
| + git_hash = ndb.StringProperty(indexed=False) |
|
stgao
2016/07/14 18:01:33
In other code, we use revision instead.
|
| + # Another identifier for code build. |
| + commit_position = ndb.IntegerProperty(indexed=False) |