integration.test.js

// Copyright 2022 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
//
//      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.

'use strict';

const assert = require('assert');
const crypto = require('crypto');
const supertest = require('supertest');
const functionsFramework = require('@google-cloud/functions-framework/testing');

process.env.SLACK_SECRET = 'testsecret';
const SLACK_TIMESTAMP = Date.now();

require('../index');

const generateSignature = query => {
  const body = JSON.stringify({text: query});

  const buf = Buffer.from(body);
  const plaintext = `v0:${SLACK_TIMESTAMP}:${buf}`;

  const hmac = crypto.createHmac('sha256', process.env.SLACK_SECRET);
  hmac.update(plaintext);
  const ciphertext = hmac.digest('hex');

  return `v0=${ciphertext}`;
};

describe('functions_slack_format functions_slack_request functions_slack_search functions_verify_webhook', () => {
  it('returns search results', async () => {
    const query = 'kolach';
    const server = functionsFramework.getTestServer('kgSearch');
    const response = await supertest(server)
      .post('/')
      .set({
        'x-slack-signature': generateSignature(query),
        'x-slack-request-timestamp': SLACK_TIMESTAMP,
      })
      .send({text: query})
      .expect(200);

    const results = response.body && response.body.attachments;
    assert.ok(results);

    const result = results[0];

    assert.ok(result);
    assert.ok(result.text);
    assert.ok(result.title);
    assert.ok(result.title.toLowerCase().includes('kolach'));
  });

  it('handles non-existent query', async () => {
    const query = 'g1bb3r1shhhhhhh';

    const server = functionsFramework.getTestServer('kgSearch');
    const response = await supertest(server)
      .post('/')
      .set({
        'x-slack-signature': generateSignature(query),
        'x-slack-request-timestamp': SLACK_TIMESTAMP,
      })
      .send({text: query})
      .expect(200);

    const results = response.body && response.body.attachments;
    assert.ok(results);

    const result = results[0];
    assert.ok(result);
    assert.ok(result.text);
    assert.ok(result.text.includes('No results'));
  });

  it('handles empty query', async () => {
    const query = '';

    const server = functionsFramework.getTestServer('kgSearch');
    await supertest(server)
      .post('/')
      .set({
        'x-slack-signature': generateSignature(query),
        'x-slack-request-timestamp': SLACK_TIMESTAMP,
      })
      .send({text: query})
      .expect(400);
  });

  it('fails on missing signature', async () => {
    const query = 'kolach';

    const server = functionsFramework.getTestServer('kgSearch');
    await supertest(server)
      .post('/')
      .set({
        'x-slack-request-timestamp': Date.now().toString(),
      })
      .send({text: query})
      .expect(401);
  });
});