Chop off platform name from step name

appengine/findit/waterfall/test/try_job_util_test.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.
 
 from datetime import datetime
 from datetime import timedelta
 
 from common.waterfall import failure_type
 from model import analysis_status
 from model.wf_analysis import WfAnalysis
@@ skipping 1250 matching lines @@
                 {'source': 'b.cc', 'target': 'b.o'},
             ]
         }
     }
 
     self.assertEqual(
         try_job_util._GetFailedTargetsFromSignals(
             signals, 'master1', 'builder1'),
         ['b.o'])
 
+  def testRemovePlatformFromStepName(self):
+    self.assertEqual('a_tests',
+        try_job_util._RemovePlatformFromStepName('a_tests on Platform'))
+    self.assertEqual('a_tests',
+        try_job_util._RemovePlatformFromStepName('a_tests on Other-Platform'))
+    self.assertEqual('a_tests',
+        try_job_util._RemovePlatformFromStepName('a_tests'))
+
   def testGetSuspectedCLsWithFailuresNoHeuristicResult(self):
     heuristic_result = None
     expected_suspected_revisions = []
     self.assertEqual(
         expected_suspected_revisions,
         sorted(try_job_util.GetSuspectedCLsWithFailures(heuristic_result)))
 
   def testGetSuspectedCLsWithFailuresEmptyHeuristicResult(self):
     heuristic_result = {}
     expected_suspected_revisions = []
@@ skipping 115 matching lines @@
         try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))
 
   def testShouldBailOutforOutdatedBuild(self):
     yesterday = datetime.utcnow() - timedelta(days=1)
     build = WfBuild.Create('m', 'b', 1)
     build.start_time = yesterday
     self.assertTrue(try_job_util._ShouldBailOutForOutdatedBuild(build))
 
     build.start_time = yesterday + timedelta(hours=1)
     self.assertFalse(try_job_util._ShouldBailOutForOutdatedBuild(build))