Index: bin/cbuildbot_comm_unittest.py |
diff --git a/bin/cbuildbot_comm_unittest.py b/bin/cbuildbot_comm_unittest.py |
deleted file mode 100755 |
index 4c95fb6ed072227983ea3aa90bc458c58a866075..0000000000000000000000000000000000000000 |
--- a/bin/cbuildbot_comm_unittest.py |
+++ /dev/null |
@@ -1,101 +0,0 @@ |
-#!/usr/bin/python |
- |
-# Copyright (c) 2010 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. |
- |
-"""Units tests for cbuildbot_comm commands.""" |
- |
-import cbuildbot_comm |
-import sys |
-import threading |
-import time |
-import unittest |
- |
-_TEST_CONFIG = {'test_slave' : |
- {'master' : False, |
- 'hostname' : 'localhost', |
- 'important' : True}, |
- 'test_master' : |
- {'master' : True, |
- 'important' : False |
- } |
- } |
- |
-# Reduce timeouts. |
-cbuildbot_comm._HEARTBEAT_TIMEOUT = 2 |
-cbuildbot_comm._MAX_TIMEOUT = 6 |
- |
-class _MasterSendBadStatus(threading.Thread): |
- |
- def __init__(self, test_class): |
- threading.Thread.__init__(self) |
- self.test_class = test_class |
- |
- def run(self): |
- # Sleep for heartbeat timeout to let slave start up. |
- time.sleep(2) |
- return_value = cbuildbot_comm._SendCommand('localhost', 'bad-command', |
- 'args') |
- self.test_class.assertEqual(return_value, |
- cbuildbot_comm._STATUS_COMMAND_REJECTED) |
- |
-class _MasterCheckStatusThread(threading.Thread): |
- |
- def __init__(self, config, expected_return, test_class): |
- threading.Thread.__init__(self) |
- self.config = config |
- self.expected_return = expected_return |
- self.test_class = test_class |
- |
- def run(self): |
- return_value = cbuildbot_comm.HaveSlavesCompleted(self.config) |
- self.test_class.assertEqual(return_value, self.expected_return) |
- |
- |
-class CBuildBotCommTest(unittest.TestCase): |
- |
- def testSlaveComplete(self): |
- print >> sys.stderr, '\n>>> Running testSlaveComplete\n' |
- # Master should check statuses in another thread. |
- master_thread = _MasterCheckStatusThread(_TEST_CONFIG, True, self) |
- master_thread.start() |
- |
- return_value = cbuildbot_comm.PublishStatus( |
- cbuildbot_comm.STATUS_BUILD_COMPLETE) |
- self.assertEqual(return_value, True) |
- |
- def testMasterTimeout(self): |
- print >> sys.stderr, '\n>>> Running testMasterTimeout\n' |
- return_value = cbuildbot_comm.HaveSlavesCompleted(_TEST_CONFIG) |
- self.assertEqual(return_value, False) |
- |
- def testSlaveTimeout(self): |
- print >> sys.stderr, '\n>>> Running testSlaveTimeout\n' |
- return_value = cbuildbot_comm.PublishStatus( |
- cbuildbot_comm.STATUS_BUILD_COMPLETE) |
- self.assertEqual(return_value, False) |
- |
- def testSlaveFail(self): |
- print >> sys.stderr, '\n>>> Running testSlaveFail\n' |
- # Master should check statuses in another thread. |
- master_thread = _MasterCheckStatusThread(_TEST_CONFIG, False, self) |
- master_thread.start() |
- |
- return_value = cbuildbot_comm.PublishStatus( |
- cbuildbot_comm.STATUS_BUILD_FAILED) |
- self.assertEqual(return_value, True) |
- |
- def testBadCommand(self): |
- print >> sys.stderr, '\n>>> Running testSendBadCommand\n' |
- # Master should check statuses in another thread. |
- master_thread = _MasterSendBadStatus(self) |
- master_thread.start() |
- |
- return_value = cbuildbot_comm.PublishStatus( |
- cbuildbot_comm.STATUS_BUILD_COMPLETE) |
- self.assertEqual(return_value, False) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |