[DEV] Set chromedriver path with env variable (#1123)
parent
e43bc93047
commit
61883437cc
|
@ -21,7 +21,12 @@ type WebDriverSession struct {
|
||||||
|
|
||||||
// StartWebDriverWithProxy create a selenium session.
|
// StartWebDriverWithProxy create a selenium session.
|
||||||
func StartWebDriverWithProxy(proxy string, port int) (*WebDriverSession, error) {
|
func StartWebDriverWithProxy(proxy string, port int) (*WebDriverSession, error) {
|
||||||
service, err := selenium.NewChromeDriverService("/usr/bin/chromedriver", port)
|
driverPath := os.Getenv("CHROMEDRIVER_PATH")
|
||||||
|
if driverPath == "" {
|
||||||
|
driverPath = "/usr/bin/chromedriver"
|
||||||
|
}
|
||||||
|
|
||||||
|
service, err := selenium.NewChromeDriverService(driverPath, port)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
Loading…
Reference in New Issue