blob: e3325d2c7badaec823fd9d8c19d1392deb409be2 [file] [log] [blame]
# -*- coding: utf-8 -*-
# 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 __future__ import print_function
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'])