notifications.test.js

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

'use strict';

const {SecurityCenterClient} = require('@google-cloud/security-center');
const uuidv1 = require('uuid').v1;
const {assert} = require('chai');
const {describe, it, before, after} = require('mocha');
const {execSync} = require('child_process');
const exec = cmd => execSync(cmd, {encoding: 'utf8'});

// TODO(developers): update for your own environment
const organizationId = '1081635000895';
const orgName = 'organizations/' + organizationId;
const pubsubTopic = 'projects/project-a-id/topics/notifications-sample-topic';

describe('Client with Notifications', async () => {
  const createConfig = 'notif-config-test-node-create' + uuidv1();
  const deleteConfig = 'notif-config-test-node-delete' + uuidv1();
  const getConfig = 'notif-config-test-node-get' + uuidv1();
  const listConfig = 'notif-config-test-node-list' + uuidv1();
  const updateConfig = 'notif-config-test-node-update' + uuidv1();

  before(async () => {
    const client = new SecurityCenterClient();
    async function createNotificationConfig(configId) {
      /*eslint no-unused-vars: ["error", { "varsIgnorePattern": "^_" }]*/
      const [_response] = await client.createNotificationConfig({
        parent: orgName,
        configId: configId,
        notificationConfig: {
          description: 'Sample config for node.js',
          pubsubTopic: pubsubTopic,
          streamingConfig: {filter: 'state = "ACTIVE"'},
        },
      });
    }

    await createNotificationConfig(deleteConfig);
    await createNotificationConfig(getConfig);
    await createNotificationConfig(listConfig);
    await createNotificationConfig(updateConfig);
  });

  after(async () => {
    const client = new SecurityCenterClient();
    async function deleteNotificationConfig(configId) {
      const name = client.organizationNotificationConfigPath(
        organizationId,
        configId
      );
      await client.deleteNotificationConfig({name: name});
    }

    await deleteNotificationConfig(createConfig);
    await deleteNotificationConfig(getConfig);
    await deleteNotificationConfig(listConfig);
    await deleteNotificationConfig(updateConfig);
  });

  it('client can create config', () => {
    const output = exec(
      `node v1/createNotificationConfig.js ${organizationId} ${createConfig} ${pubsubTopic}`
    );
    assert.include(output, createConfig);
    assert.match(output, /Notification config creation succeeded/);
    assert.notMatch(output, /undefined/);
  });

  it('client can delete config', () => {
    const output = exec(
      `node v1/deleteNotificationConfig.js ${organizationId} ${deleteConfig}`
    );
    assert.include(output, 'Notification config deleted');
    assert.notMatch(output, /undefined/);
  });

  it('client can get config', () => {
    const output = exec(
      `node v1/getNotificationConfig.js ${organizationId} ${getConfig}`
    );
    assert.include(output, getConfig);
    assert.match(output, /Notification config/);
    assert.notMatch(output, /undefined/);
  });

  it('client can list configs', () => {
    const output = exec(`node v1/listNotificationConfigs.js ${organizationId}`);
    assert.include(output, listConfig);
    assert.match(output, /Received Notification configs/);
    assert.notMatch(output, /undefined/);
  });

  it('client can update configs', () => {
    const output = exec(
      `node v1/updateNotificationConfig.js ${organizationId} ${updateConfig} ${pubsubTopic}`
    );
    assert.include(output, updateConfig);
    assert.match(output, /notification config update succeeded/);
    assert.notMatch(output, /undefined/);
  });
});