| # 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']) |
