// Copyright 2019 Google LLC
//
// 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
//
//     https://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.

const assert = require('assert');
const got = require('got');
const {execSync, exec} = require('child_process');
const {GoogleAuth} = require('google-auth-library');
const puppeteer = require('puppeteer');
const auth = new GoogleAuth();

describe('End-to-End Tests', () => {
  let BASE_URL, ID_TOKEN;
  // Retrieve Cloud Run service test config
  const {GOOGLE_CLOUD_PROJECT} = process.env;
  if (!GOOGLE_CLOUD_PROJECT) {
    throw Error('"GOOGLE_CLOUD_PROJECT" env var not found.');
  }
  let {SERVICE_NAME} = process.env;
  if (!SERVICE_NAME) {
    SERVICE_NAME = 'websocket';
    console.log(
      `"SERVICE_NAME" env var not found. Defaulting to "${SERVICE_NAME}"`
    );
  }
  const REGION = 'us-central1';
  let browser, browserPage;
  const {REDISHOST} = process.env;
  if (!REDISHOST) {
    throw Error('"REDISHOST" env var not found.');
  }
  before(async () => {
    // Deploy service using Cloud Build
    const buildCmd =
      `gcloud builds submit --project ${GOOGLE_CLOUD_PROJECT} ` +
      '--config ./test/e2e_test_setup.yaml ' +
      `--substitutions _SERVICE=${SERVICE_NAME},_REGION=${REGION},_REDISHOST=${REDISHOST}`;

    console.log('Starting Cloud Build...');
    execSync(buildCmd);
    console.log('Cloud Build completed.');

    // Retrieve URL of Cloud Run service
    const url = execSync(
      `gcloud run services describe ${SERVICE_NAME} --project=${GOOGLE_CLOUD_PROJECT} ` +
        `--region=${REGION} --format='value(status.url)'`
    );

    BASE_URL = url.toString('utf-8').trim();
    if (!BASE_URL) throw Error('Cloud Run service URL not found');

    // Retrieve ID token for testing
    const client = await auth.getIdTokenClient(BASE_URL);
    const clientHeaders = await client.getRequestHeaders();
    ID_TOKEN = clientHeaders['Authorization'].trim();
    if (!ID_TOKEN) throw Error('Unable to acquire an ID token.');

    // Initialize puppeteer to test service client websockets
    browser = await puppeteer.launch({
      args: ['--no-sandbox', '--disable-setuid-sandbox'],
    });
    browserPage = await browser.newPage();
  });

  after(async () => {
    if (browser) {
      await browser.close();
    }

    const cleanUpCmd =
      `gcloud builds submit --project ${GOOGLE_CLOUD_PROJECT} ` +
      '--config ./test/e2e_test_cleanup.yaml ' +
      `--substitutions _SERVICE=${SERVICE_NAME},_REGION=${REGION}`;

    exec(cleanUpCmd);
  });

  it('can be reached by an HTTP request', async () => {
    const options = {
      headers: {
        Authorization: ID_TOKEN,
      },
      method: 'GET',
      retry: 3,
    };
    const response = await got(BASE_URL, options);
    assert.strictEqual(response.statusCode, 200);
  });

  it('should process chat message', async () => {
    await browserPage.setExtraHTTPHeaders({
      Authorization: ID_TOKEN,
    });
    // Go to service client
    await browserPage.goto(BASE_URL, {
      waitUntil: 'networkidle2',
    });

    let document;
    // Join chat room
    await browserPage.evaluate(() => {
      document.querySelector('#name').value = 'Sundar';
      document.querySelector('#room').value = 'Google';
      document.querySelector('.signin').click();
    });
    await new Promise(resolve => setTimeout(resolve, 100));

    // Send message
    await browserPage.evaluate(() => {
      document.querySelector('#msg').value = 'Welcome!';
      document.querySelector('.send').click();
    });
    await new Promise(resolve => setTimeout(resolve, 100));

    // Confirm message
    const list = await browserPage.$('#messages li');
    const itemText = await browserPage.evaluate(el => el.textContent, list);
    assert.strictEqual(itemText.trim(), 'Sundar: Welcome!');
    // Confirm room
    const room = await browserPage.$('#chatroom h1');
    const roomText = await browserPage.evaluate(el => el.textContent, room);
    assert.strictEqual(roomText, 'Google');
  });
});

@@ -4080,131 +4080,273 @@
     
-const list = await browserPage.$('#messages li');%0A    const itemText = await browserPage.evaluate(el =%3E el.textContent, lis
+let itemText;%0A    for (let i = 0; i %3C 5; i++) %7B%0A      itemText = await browserPage.evaluate(() =%3E %7B%0A        return document.querySelector('#messages li');%0A      %7D);%0A      if (itemText) return;%0A      await sleep(i * 1000); // Linear delay%0A    %7D%0A    assert.ok(itemTex
 t);%0A
@@ -4377,16 +4377,26 @@
 temText.
+innerText.
 trim(), 
@@ -4611,18 +4611,106 @@
 ogle');%0A
-
   %7D);
+%0A%0A  function sleep(ms) %7B%0A    return new Promise(resolve =%3E setTimeout(resolve, ms));%0A  %7D
 %0A%7D);%0A

add retries to test (#2313)