[tools/perf] Make trybot bails out early if the benchmark is disabled on trybot platform

tools/perf/core/trybot_command_unittest.py

 # Copyright 2015 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 argparse
 import logging
 import json
 import StringIO
 import unittest
 
 import mock
 
+from telemetry import benchmark
 from telemetry.testing import system_stub
 
 from core import trybot_command
 
 
 class FakeProcess(object):
 
   def __init__(self, expected_responses):
     self._communicate = expected_responses[1:]
     self._poll = expected_responses[0]
@@ skipping 598 matching lines @@
   "target_arch": "ia32",
   "truncate_percent": "0"
 }''''''config = {
   "command": "./tools/perf/run_benchmark --browser=android-chromium --verbose",
   "max_time_minutes": "120",
   "repeat_count": "1",
   "target_arch": "ia32",
   "truncate_percent": "0"
 }''')
     self.assertEquals(cfg.read(), config)
+
+
+
+class IsBenchmarkDisabledOnTrybotPlatformTest(unittest.TestCase):
+
+  def IsBenchmarkDisabled(self, benchmark_class, trybot_name):
+    return trybot_command.Trybot.IsBenchmarkDisabledOnTrybotPlatform(
+        benchmark_class, trybot_name)[0]
+
+  def testBenchmarkIsDisabledAll(self):
+    @benchmark.Disabled('all')
+    class FooBenchmark(benchmark.Benchmark):
+      pass
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'all'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'all-mac'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'android-s5'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'linux'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'winx64ati'))
+
+  def testBenchmarkIsEnabledAll(self):
+    @benchmark.Enabled('all')
+    class FooBenchmark(benchmark.Benchmark):
+      pass
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'all'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'all-mac'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'android-s5'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'linux'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'winx64ati'))
+
+  def testBenchmarkIsDisabledOnMultiplePlatforms(self):
+    @benchmark.Disabled('win', 'mac')
+    class FooBenchmark(benchmark.Benchmark):
+      pass
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'all'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'android-s5'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'linux'))
+
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'all-mac'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'winx64ati'))
+
+  def testBenchmarkIsEnabledOnMultiplePlatforms(self):
+    @benchmark.Enabled('win', 'mac')
+    class FooBenchmark(benchmark.Benchmark):
+      pass
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'all'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'all-mac'))
+    self.assertFalse(self.IsBenchmarkDisabled(FooBenchmark, 'winx64ati'))
+
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'android-s5'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'linux'))
+    self.assertTrue(self.IsBenchmarkDisabled(FooBenchmark, 'all-linux'))