Mocking host

browserless · Aug 23, 2018 · 54b160a · 54b160a
1 parent aa73447
commit 54b160a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.npmrc b/.npmrc
@@ -0,0 +1 @@
+chromedriver_version=2.39
diff --git a/Dockerfile b/Dockerfile
@@ -6,7 +6,6 @@ ENV HOST=0.0.0.0
 ENV PORT=3000
 ENV application_directory=/usr/src/app
 ENV font_directory=/usr/share/fonts/noto
-ENV CHROMEDRIVER_VERSION=2.39
 
 # Build Args
 ARG USE_CHROME_STABLE

diff --git a/src/webdriver-provider.ts b/src/webdriver-provider.ts
@@ -45,7 +45,7 @@ export class WebDriver {
       getPort().then(async (port) => {
         const chromeProcess = chromeDriver.start([
           '--url-base=wd/hub',
-          '--whitelisted-ips',
+          '--whitelisted-ips=127.0.0.1',
           `--port=${port}`,
           '--verbose',
         ]);
@@ -85,6 +85,7 @@ export class WebDriver {
           await sleep(25);
 
           proxy.web(req, res, (error) => {
+            req.headers.host = '127.0.0.1:3000';
             debug(`Issue in webdriver: ${error.message}`);
             res.end();
             done();