/**
 * Copyright 2018 Google Inc. All rights reserved.
 * Modifications copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { test as it, expect } from './pageTest';

it('should work', async ({ page, server }) => {
  await page.route('**/*', route => route.fallback());
  await page.goto(server.EMPTY_PAGE);
});

it('should fall back', async ({ page, server }) => {
  const intercepted = [];
  await page.route('**/empty.html', route => {
    intercepted.push(1);
    route.fallback();
  });
  await page.route('**/empty.html', route => {
    intercepted.push(2);
    route.fallback();
  });
  await page.route('**/empty.html', route => {
    intercepted.push(3);
    route.fallback();
  });
  await page.goto(server.EMPTY_PAGE);
  expect(intercepted).toEqual([3, 2, 1]);
});

it('should fall back async', async ({ page, server }) => {
  const intercepted = [];
  await page.route('**/empty.html', async route => {
    intercepted.push(1);
    await new Promise(r => setTimeout(r, 100));
    route.fallback();
  });
  await page.route('**/empty.html', async route => {
    intercepted.push(2);
    await new Promise(r => setTimeout(r, 100));
    route.fallback();
  });
  await page.route('**/empty.html', async route => {
    intercepted.push(3);
    await new Promise(r => setTimeout(r, 100));
    route.fallback();
  });
  await page.goto(server.EMPTY_PAGE);
  expect(intercepted).toEqual([3, 2, 1]);
});

it('should not chain fulfill', async ({ page, server }) => {
  let failed = false;
  await page.route('**/empty.html', route => {
    failed = true;
  });
  await page.route('**/empty.html', route => {
    route.fulfill({ status: 200, body: 'fulfilled' });
  });
  await page.route('**/empty.html', route => {
    route.fallback();
  });
  const response = await page.goto(server.EMPTY_PAGE);
  const body = await response.body();
  expect(body.toString()).toEqual('fulfilled');
  expect(failed).toBeFalsy();
});

it('should not chain abort', async ({ page, server }) => {
  let failed = false;
  await page.route('**/empty.html', route => {
    failed = true;
  });
  await page.route('**/empty.html', route => {
    route.abort();
  });
  await page.route('**/empty.html', route => {
    route.fallback();
  });
  const e = await page.goto(server.EMPTY_PAGE).catch(e => e);
  expect(e).toBeTruthy();
  expect(failed).toBeFalsy();
});

it('should fall back after exception', async ({ page, server }) => {
  await page.route('**/empty.html', route => {
    route.continue();
  });
  await page.route('**/empty.html', async route => {
    try {
      await route.fulfill({ response: {} as any });
    } catch (e) {
      route.fallback();
    }
  });
  await page.goto(server.EMPTY_PAGE);
});

it('should chain once', async ({ page, server }) => {
  await page.route('**/empty.html', route => {
    route.fulfill({ status: 200, body: 'fulfilled one' });
  }, { times: 1 });
  await page.route('**/empty.html', route => {
    route.fallback();
  }, { times: 1 });
  const response = await page.goto(server.EMPTY_PAGE);
  const body = await response.body();
  expect(body.toString()).toEqual('fulfilled one');
});

it('should amend HTTP headers', async ({ page, server }) => {
  const values = [];
  await page.route('**/sleep.zzz', async route => {
    values.push(route.request().headers().foo);
    values.push(await route.request().headerValue('FOO'));
    route.continue();
  });
  await page.route('**/*', route => {
    route.fallback({ headers: { ...route.request().headers(), FOO: 'bar' } });
  });
  await page.goto(server.EMPTY_PAGE);
  const [request] = await Promise.all([
    server.waitForRequest('/sleep.zzz'),
    page.evaluate(() => fetch('/sleep.zzz'))
  ]);
  values.push(request.headers['foo']);
  expect(values).toEqual(['bar', 'bar', 'bar']);
});

it('should delete header with undefined value', async ({ page, server, browserName }) => {
  await page.goto(server.PREFIX + '/empty.html');
  server.setRoute('/something', (request, response) => {
    response.writeHead(200, { 'Access-Control-Allow-Origin': '*' });
    response.end('done');
  });
  let interceptedRequest;
  await page.route('**/*', (route, request) => {
    interceptedRequest = request;
    route.continue();
  });
  await page.route(server.PREFIX + '/something', async (route, request) => {
    const headers = await request.allHeaders();
    route.fallback({
      headers: {
        ...headers,
        foo: undefined
      }
    });
  });
  const [text, serverRequest] = await Promise.all([
    page.evaluate(async url => {
      const data = await fetch(url, {
        headers: {
          foo: 'a',
          bar: 'b',
        }
      });
      return data.text();
    }, server.PREFIX + '/something'),
    server.waitForRequest('/something')
  ]);
  expect(text).toBe('done');
  expect(interceptedRequest.headers()['foo']).toEqual(undefined);
  expect(interceptedRequest.headers()['bar']).toEqual('b');
  expect(serverRequest.headers.foo).toBeFalsy();
  expect(serverRequest.headers.bar).toBe('b');
});

it('should amend method', async ({ page, server }) => {
  const sRequest = server.waitForRequest('/sleep.zzz');
  await page.goto(server.EMPTY_PAGE);

  let method: string;
  await page.route('**/*', route => {
    method = route.request().method();
    route.continue();
  });
  await page.route('**/*', route => route.fallback({ method: 'POST' }));

  const [request] = await Promise.all([
    page.waitForRequest('**/sleep.zzz'),
    page.evaluate(() => fetch('/sleep.zzz'))
  ]);
  expect(method).toBe('POST');
  expect(request.method()).toBe('POST');
  expect((await sRequest).method).toBe('POST');
});

it('should override request url', async ({ page, server }) => {
  const serverRequest = server.waitForRequest('/global-var.html');

  let url: string;
  await page.route('**/global-var.html', route => {
    url = route.request().url();
    route.continue();
  });

  await page.route('**/foo', route => route.fallback({ url: server.PREFIX + '/global-var.html' }));

  const response = await page.goto(server.PREFIX + '/foo');
  expect(url).toBe(server.PREFIX + '/global-var.html');
  expect(response.request().url()).toBe(server.PREFIX + '/global-var.html');
  expect(response.url()).toBe(server.PREFIX + '/global-var.html');
  expect(await page.evaluate(() => window['globalVar'])).toBe(123);
  expect((await serverRequest).method).toBe('GET');
});

it.describe('post data', () => {
  it('should amend post data', async ({ page, server }) => {
    await page.goto(server.EMPTY_PAGE);
    let postData: string;
    await page.route('**/*', route => {
      postData = route.request().postData();
      route.continue();
    });
    await page.route('**/*', route => {
      route.fallback({ postData: 'doggo' });
    });
    const [serverRequest] = await Promise.all([
      server.waitForRequest('/sleep.zzz'),
      page.evaluate(() => fetch('/sleep.zzz', { method: 'POST', body: 'birdy' }))
    ]);
    expect(postData).toBe('doggo');
    expect((await serverRequest.postBody).toString('utf8')).toBe('doggo');
  });

  it('should amend binary post data', async ({ page, server }) => {
    await page.goto(server.EMPTY_PAGE);
    const arr = Array.from(Array(256).keys());
    let postDataBuffer: Buffer;
    await page.route('**/*', route => {
      postDataBuffer = route.request().postDataBuffer();
      route.continue();
    });
    await page.route('**/*', route => {
      route.fallback({ postData: Buffer.from(arr) });
    });
    const [serverRequest] = await Promise.all([
      server.waitForRequest('/sleep.zzz'),
      page.evaluate(() => fetch('/sleep.zzz', { method: 'POST', body: 'birdy' }))
    ]);
    const buffer = await serverRequest.postBody;
    expect(postDataBuffer.length).toBe(arr.length);
    expect(buffer.length).toBe(arr.length);
    for (let i = 0; i < arr.length; ++i) {
      expect(buffer[i]).toBe(arr[i]);
      expect(postDataBuffer[i]).toBe(arr[i]);
    }
  });
});