Enable --exclude-tasks option (elastic#160)

Closes elastic/night-rally#159

Author: ebadyano

night_rally/night_rally.py

@@ -389,13 +389,14 @@ class StandardParams:
     """
     Extracts all parameters that are needed for all Rally invocations.
     """
-    def __init__(self, configuration_name, effective_start_date, runtime_jdk, user_tag_setup, race_configs_id=None, test_mode=False):
+    def __init__(self, configuration_name, effective_start_date, runtime_jdk, user_tag_setup, race_configs_id=None, test_mode=False, exclude_tasks=None):
         self.configuration_name = configuration_name
         self.effective_start_date = effective_start_date
         self.runtime_jdk = runtime_jdk
         self.user_tag_setup = user_tag_setup
         self.race_configs_id = race_configs_id
         self.test_mode = test_mode
+        self.exclude_tasks = exclude_tasks
 
     def __call__(self, race_config):
         params = {
@@ -421,6 +422,7 @@ def __call__(self, race_config):
             params["runtime-jdk"] = race_config.runtime_jdk
         else:
             add_if_present(params, "runtime-jdk", self.runtime_jdk)
+        add_if_present(params, "exclude-tasks", race_config.exclude_tasks)
         add_if_present(params, "car-params", race_config.car_params)
         add_if_present(params, "track-params", race_config.track_params)
         add_if_present(params, "elasticsearch-plugins", race_config.plugins)
@@ -533,6 +535,10 @@ def car(self):
     @property
     def car_params(self):
         return self.configuration.get("car-params")
+    
+    @property
+    def exclude_tasks(self):
+        return self.configuration.get("exclude-tasks")
 
     @property
     def plugins(self):

tests/night_rally_test.py

@@ -374,6 +374,52 @@ def test_run_two_oss_challenges_successfully(self, mocked_wait_until_port_is_fre
             ,
             system_call.calls
         )
+        
+    @mock.patch('night_rally.night_rally.wait_until_port_is_free', return_value=True)
+    def test_exclude_tasks_option(self, mocked_wait_until_port_is_free):
+        system_call = RecordingSystemCall(return_value=False)
+
+        tracks = [
+            {
+                "track": "geonames",
+                "flavors": [
+                    {
+                        "name": "oss",
+                        "licenses": [
+                            {
+                                "name": "oss",
+                                "configurations": [
+                                    {
+                                        "name": "geonames-append-1node",
+                                        "challenge": "append-no-conflicts",
+                                        "car": "defaults",
+                                        "exclude-tasks": "delete,type:search"
+                                    }
+                                ]
+                            }
+                        ]
+                    }
+                ]
+            }
+        ]
+
+        start_date = datetime.datetime(2016, 1, 1)
+        race_configs_id = os.path.basename(get_random_race_configs_id())
+        params = [night_rally.StandardParams("nightly", start_date, 8, "bare", race_configs_id=race_configs_id)]
+        cmd = night_rally.NightlyCommand(params, start_date)
+        night_rally.run_rally(tracks, None, ["localhost"], cmd, skip_ansible=True, system=system_call)
+        self.assertEqual(
+            [
+                "rally --skip-update --configuration-name=\"nightly\" --quiet --target-host=\"localhost:39200\" "
+                "--effective-start-date=\"2016-01-01 00:00:00\" --track-repository=\"default\" --track=\"geonames\" "
+                "--challenge=\"append-no-conflicts\" --car=\"defaults\" --client-options=\"timeout:240\" "
+                "--user-tag=\"name:geonames-append-1node,setup:bare,race-configs-id:{},license:oss\" --runtime-jdk=\"8\" "
+                "--exclude-tasks=\"delete,type:search\" --pipeline=\"from-sources-complete\" "
+                "--revision=\"@2016-01-01T00:00:00Z\"".format(race_configs_id)
+            ]
+            ,
+            system_call.calls
+        )
 
     @mock.patch('night_rally.night_rally.wait_until_port_is_free', return_value=True)
     def test_overwrite_runtime_jdk_successfully(self, mocked_wait_until_port_is_free):