66import org .testng .annotations .AfterMethod ;
77import org .testng .annotations .Listeners ;
88
9- import static org .sayem .browser .BrowserType .CHROME ;
10- import static org .sayem .browser .BrowserType .FIREFOX ;
9+ import java .util .stream .Stream ;
10+ 11+ import static java .lang .String .valueOf ;
12+ import static org .sayem .browser .BrowserType .values ;
1113
1214@ Listeners ({BrowserListener .class })
1315public class TestBase {
@@ -16,16 +18,16 @@ public class TestBase {
1618
1719 @ AfterMethod
1820 public void tearDown () {
19- browser .driver ().quit ();
21+ browser .driver ().close ();
2022 }
2123
2224 @ SuppressWarnings ("unchecked" )
2325 protected <T > T getDriver () {
24- if ( System . getProperty ( "browser" ). equalsIgnoreCase ( String . valueOf ( CHROME ))) {
25- browser = CHROME . browser . get ();
26- } else {
27- browser = FIREFOX . browser . get ();
28- }
26+ browser = Stream . of ( values ())
27+ . parallel ()
28+ . filter ( s -> s . name ()
29+ . equalsIgnoreCase ( System . getProperty ( "browser" )))
30+ . findFirst (). get (). driver . get (). browser ();
2931 return (T ) browser ;
3032 }
3133}
0 commit comments