[Findit] Refactoring trigger swarming task pipelines

appengine/findit/model/test/wf_swarming_task_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.
 
 import unittest
 
 from model.wf_swarming_task import WfSwarmingTask
 
 
 class WfSwarmingTaskTest(unittest.TestCase):
+
   def testClassifiedTests(self):
-    task =  WfSwarmingTask.Create('m', 'b', 121, 'browser_tests')
+    task = WfSwarmingTask.Create('m', 'b', 121, 'browser_tests')
     task.tests_statuses = {
         'TestSuite1.test1': {
             'total_run': 2,
             'SUCCESS': 2
         },
         'TestSuite1.test2': {
             'total_run': 4,
             'SUCCESS': 2,
             'FAILURE': 2
         },
@@ skipping 13 matching lines @@
 
     expected_classified_tests = {
         'flaky_tests': ['TestSuite1.test2', 'TestSuite1.test1'],
         'reliable_tests': ['TestSuite1.test3', 'TestSuite1.test4'],
         'unknown_tests': ['TestSuite1.test5']
     }
 
     self.assertEqual(expected_classified_tests, task.classified_tests)
     self.assertEqual(expected_classified_tests['reliable_tests'],
                      task.reliable_tests)
+
+  def testStepName(self):
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = 123
+    expected_step_name = 's'
+    task = WfSwarmingTask.Create(
+        master_name, builder_name, build_number, expected_step_name)
+    self.assertEqual(expected_step_name, task.step_name)