#!/usr/bin/env python # Copyright 2014 The LUCI Authors. All rights reserved. # Use of this source code is governed under the Apache License, Version 2.0 # that can be found in the LICENSE file. import os import sys import unittest import test_env_api test_env_api.setup_test_env() import bot class TestBot(unittest.TestCase): def test_bot(self): obj = bot.Bot( None, {'dimensions': {'foo': ['bar']}}, 'https://localhost:1', '1234-1a2b3c4-tainted-joe', 'base_dir', None) self.assertEqual({'foo': ['bar']}, obj.dimensions) self.assertEqual( os.path.join(obj.base_dir, 'swarming_bot.zip'), obj.swarming_bot_zip) self.assertEqual('1234-1a2b3c4-tainted-joe', obj.server_version) self.assertEqual('base_dir', obj.base_dir) def test_attribute_updates(self):
#!/usr/bin/env python # Copyright 2014 The LUCI Authors. All rights reserved. # Use of this source code is governed by the Apache v2.0 license that can be # found in the LICENSE file. import os import sys import unittest import threading THIS_FILE = os.path.abspath(__file__) import test_env_api test_env_api.setup_test_env() import bot class TestBot(unittest.TestCase): def test_bot(self): obj = bot.Bot( {'dimensions': {'foo': 'bar'}}, 'https://localhost:1', '1234-1a2b3c4-tainted-joe', 'base_dir', None) self.assertEqual({'foo': 'bar'}, obj.dimensions) self.assertEqual( os.path.join(os.path.dirname(THIS_FILE), 'swarming_bot.zip'), obj.swarming_bot_zip) self.assertEqual('1234-1a2b3c4-tainted-joe', obj.server_version)
def setup_test_env(): """Sets up the environment for bot tests.""" sys.path.insert(0, API_DIR) import test_env_api test_env_api.setup_test_env()