blob: 83ab26fbe6b67ff63d060866284168f3738e2481 [file] [log] [blame]
# Copyright 2018 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Unit tests for swarming_lib module."""
from chromite.cbuildbot import swarming_lib
from chromite.lib import cros_test_lib
class SwarmingLibTest(cros_test_lib.RunCommandTestCase):
"""Unit test of swarming_lib module."""
def testEnv(self):
"""Validate that 'SWARMING_TASK_ID' is removed from swarming cmds."""
swarming_lib.RunSwarmingCommand(['cmd'], 'test-server')
swarming_lib.RunSwarmingCommand(['cmd'], 'test-server',
env={'SWARMING_TASK_ID': 'foo'})
for _, kwargs in self.rc.call_args_list:
self.assertIn('env', kwargs)
self.assertNotIn('SWARMING_TASK_ID', kwargs['env'])