1 /**
2 * Copyright (c) 2002-2015, JWebUnit team.
3 *
4 * This file is part of JWebUnit.
5 *
6 * JWebUnit is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU Lesser General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
10 *
11 * JWebUnit is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with JWebUnit. If not, see <http://www.gnu.org/licenses/>.
18 */
19 package net.sourceforge.jwebunit.webdriver;
20
21 import com.gargoylesoftware.htmlunit.TopLevelWindow;
22 import org.openqa.selenium.htmlunit.HtmlUnitDriver;
23
24 public class FixedHtmlUnitDriver extends HtmlUnitDriver {
25
26 @Override
27 public void close() {
28 if (getCurrentWindow() != null) {
29 if (getWebClient().getWebWindows().size() > 1) {
30 ((TopLevelWindow) getCurrentWindow().getTopWindow()).close();
31 }
32 else {
33 quit();
34 }
35 }
36 }
37
38 }