15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Copyright (C) 2010 Google Inc. All rights reserved. 25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Copyright (C) 2010 Gabor Rapcsanyi (rgabor@inf.u-szeged.hu), University of Szeged 35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Redistribution and use in source and binary forms, with or without 55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# modification, are permitted provided that the following conditions are 65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# met: 75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Redistributions of source code must retain the above copyright 95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# notice, this list of conditions and the following disclaimer. 105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Redistributions in binary form must reproduce the above 115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# copyright notice, this list of conditions and the following disclaimer 125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# in the documentation and/or other materials provided with the 135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# distribution. 145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Neither the name of Google Inc. nor the names of its 155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# contributors may be used to endorse or promote products derived from 165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# this software without specific prior written permission. 175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)"""Unit tests for manager.py.""" 315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import sys 335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import time 347757ec2eadfa2dd8ac2aeed0a4399e9b07ec38cbBen Murdochimport webkitpy.thirdparty.unittest2 as unittest 355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.common.host_mock import MockHost 37926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles)from webkitpy.layout_tests.controllers.manager import Manager 385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.layout_tests.models import test_expectations 39926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles)from webkitpy.layout_tests.models.test_run_results import TestRunResults 405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.thirdparty.mock import Mock 415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.tool.mocktool import MockOptions 425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)class ManagerTest(unittest.TestCase): 455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_needs_servers(self): 46926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) def get_manager(): 475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) port = Mock() # FIXME: Use a tighter mock. 485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) port.TEST_PATH_SEPARATOR = '/' 495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) manager = Manager(port, options=MockOptions(http=True, max_locked_shards=1), printer=Mock()) 505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return manager 515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 52926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) manager = get_manager() 53926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertFalse(manager.needs_servers(['fast/html'])) 545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 55926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) manager = get_manager() 56926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertTrue(manager.needs_servers(['http/tests/misc'])) 575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 58591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def test_servers_started(self): 59591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def get_manager(port): 60591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager = Manager(port, options=MockOptions(http=True, max_locked_shards=1), printer=Mock()) 615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return manager 625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 63591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def start_http_server(number_of_servers=None): 64591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.http_started = True 655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 66591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def start_websocket_server(): 67591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.websocket_started = True 68591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 69591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def stop_http_server(): 70591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.http_stopped = True 71591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 72591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch def stop_websocket_server(): 73591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.websocket_stopped = True 74591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 75591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch host = MockHost() 76591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch port = host.port_factory.get('test-mac-leopard') 77591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch port.start_http_server = start_http_server 78591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch port.start_websocket_server = start_websocket_server 79591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch port.stop_http_server = stop_http_server 80591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch port.stop_websocket_server = stop_websocket_server 81591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 82591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.http_started = self.http_stopped = self.websocket_started = self.websocket_stopped = False 83591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager = get_manager(port) 84591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._start_servers(['http/tests/foo.html']) 85591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_started, True) 86591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_started, False) 87591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._stop_servers() 88591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_stopped, True) 89591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_stopped, False) 90591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 91591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.http_started = self.http_stopped = self.websocket_started = self.websocket_stopped = False 92591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._start_servers(['http/tests/websocket/foo.html']) 93591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_started, True) 94591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_started, True) 95591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._stop_servers() 96591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_stopped, True) 97591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_stopped, True) 985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 99591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.http_started = self.http_stopped = self.websocket_started = self.websocket_stopped = False 100591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._start_servers(['fast/html/foo.html']) 101591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_started, False) 102591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_started, False) 103591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch manager._stop_servers() 104591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.http_stopped, False) 105591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch self.assertEqual(self.websocket_stopped, False) 1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_look_for_new_crash_logs(self): 109926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) def get_manager(): 1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) host = MockHost() 1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) port = host.port_factory.get('test-mac-leopard') 1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) manager = Manager(port, options=MockOptions(test_list=None, http=True, max_locked_shards=1), printer=Mock()) 1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return manager 1145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) host = MockHost() 1155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) port = host.port_factory.get('test-mac-leopard') 1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) tests = ['failures/expected/crash.html'] 1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) expectations = test_expectations.TestExpectations(port, tests) 118926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) run_results = TestRunResults(expectations, len(tests)) 119926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) manager = get_manager() 120926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) manager._look_for_new_crash_logs(run_results, time.time()) 121