15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CONTENT_PUBLIC_TEST_BROWSER_TEST_BASE_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CONTENT_PUBLIC_TEST_BROWSER_TEST_BASE_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/callback.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 1058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)#include "base/threading/thread.h" 11b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)#include "net/test/spawned_test_server/spawned_test_server.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace base { 15a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)class CommandLine; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class FilePath; 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)namespace net { 2090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)namespace test_server { 2190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)class EmbeddedTestServer; 2290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 2368043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 2468043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles)class RuleBasedHostResolverProc; 2568043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles)} // namespace net 2690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class BrowserTestBase : public testing::Test { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserTestBase(); 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~BrowserTestBase(); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We do this so we can be used in a Task. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddRef() {} 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Release() {} 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Configures everything for an in process browser test, then invokes 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // BrowserMain. BrowserMain ends up invoking RunTestOnMainThreadLoop. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() OVERRIDE; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Restores state configured in SetUp. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDown() OVERRIDE; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override this to add any custom setup code that needs to be done on the 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // main thread after the browser is created and just before calling 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // RunTestOnMainThread(). 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUpOnMainThread() {} 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 50eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch // Override this to add any custom teardown code that needs to be done on the 51eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch // main thread right after RunTestOnMainThread(). 52eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void TearDownOnMainThread() {} 53eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override this to add command line flags specific to your test. 55a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) virtual void SetUpCommandLine(base::CommandLine* command_line) {} 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5768043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // Returns the host resolver being used for the tests. Subclasses might want 5868043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // to configure it inside tests. 5968043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) net::RuleBasedHostResolverProc* host_resolver() { 6068043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) return rule_based_resolver_.get(); 6168043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) } 6268043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We need these special methods because SetUp is the bottom of the stack 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // that winds up calling your test method, so it is not always an option 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to do what you want by overriding it and calling the superclass version. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override this for things you would normally override SetUp for. It will be 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // called before your individual test fixture method is run, but after most 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // of the overhead initialization has occured. 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUpInProcessBrowserTestFixture() {} 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override this for things you would normally override TearDown for. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDownInProcessBrowserTestFixture() {} 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override this rather than TestBody. 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void RunTestOnMainThread() = 0; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is invoked from main after browser_init/browser_main have completed. 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This prepares for the test by creating a new browser, runs the test 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // (RunTestOnMainThread), quits the browsers and returns. 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void RunTestOnMainThreadLoop() = 0; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Sets expected browser exit code, in case it's different than 0 (success). 8546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) void set_expected_exit_code(int code) { expected_exit_code_ = code; } 8646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns the testing server. Guaranteed to be non-NULL. 8890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // TODO(phajdan.jr): Remove test_server accessor (http://crbug.com/96594). 89c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const net::SpawnedTestServer* test_server() const { 90c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return test_server_.get(); 91c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 92c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::SpawnedTestServer* test_server() { return test_server_.get(); } 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Returns the embedded test server. Guaranteed to be non-NULL. 9590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) const net::test_server::EmbeddedTestServer* embedded_test_server() const { 9690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return embedded_test_server_.get(); 9790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 9890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) net::test_server::EmbeddedTestServer* embedded_test_server() { 9990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return embedded_test_server_.get(); 10090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 10190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_POSIX) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is only needed by a test that raises SIGTERM to ensure that a specific 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // codepath is taken. 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DisableSIGTERMHandling() { 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handle_sigterm_ = false; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This function is meant only for classes that directly derive from this 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // class to construct the test server in their constructor. They might need to 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // call this after setting up the paths. Actual test cases should never call 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // this. 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |test_server_base| is the path, relative to src, to give to the test HTTP 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // server. 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void CreateTestServer(const base::FilePath& test_server_base); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 118c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // When the test is running in --single-process mode, runs the given task on 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // the in-process renderer thread. A nested message loop is run until it 120c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // returns. 121c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) void PostTaskToInProcessRendererAndWait(const base::Closure& task); 122c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Call this before SetUp() to cause the test to generate pixel output. 1245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void EnablePixelOutput(); 125c2db58bd994c04d98e4ee2cd7565b71548655fe3Ben Murdoch 1265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Call this before SetUp() to not use GL, but use software compositing 1275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // instead. 1285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void UseSoftwareCompositing(); 1295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Returns true if the test will be using GL acceleration via OSMesa. 1315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bool UsingOSMesa() const; 132c2db58bd994c04d98e4ee2cd7565b71548655fe3Ben Murdoch 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ProxyRunTestOnMainThreadLoop(); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Testing server, started on demand. 137c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<net::SpawnedTestServer> test_server_; 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Embedded test server, cheap to create, started on demand. 14090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) scoped_ptr<net::test_server::EmbeddedTestServer> embedded_test_server_; 14190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 14268043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) // Host resolver used during tests. 14368043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) scoped_refptr<net::RuleBasedHostResolverProc> rule_based_resolver_; 14468043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) 14546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) // Expected exit code (default is 0). 14646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) int expected_exit_code_; 14746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) 1485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // When true, the compositor will produce pixel output that can be read back 1495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // for pixel tests. 1505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bool enable_pixel_output_; 151c2db58bd994c04d98e4ee2cd7565b71548655fe3Ben Murdoch 1525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // When true, do compositing with the software backend instead of using GL. 1535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) bool use_software_compositing_; 154c2db58bd994c04d98e4ee2cd7565b71548655fe3Ben Murdoch 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_POSIX) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool handle_sigterm_; 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CONTENT_PUBLIC_TEST_BROWSER_TEST_BASE_H_ 163