quickstart.test.js

/**
 * Copyright 2017, Google, Inc.
 * 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 proxyquire = require(`proxyquire`).noPreserveCache();
const sinon = require(`sinon`);
const test = require(`ava`);
const tools = require(`@google-cloud/nodejs-repo-tools`);

const dns = proxyquire(`@google-cloud/dns`, {})();

const uuid = require(`uuid`);
const zoneName = `test-${uuid().substring(0, 13)}`;

test.before(async () => {
  await dns.createZone(zoneName, {
    dnsName: `${process.env.GCLOUD_PROJECT}.appspot.com.`
  });
});

test.after.always(async () => {
  try {
    await dns.zone(zoneName).delete();
  } catch (err) {} // ignore error
});

test.beforeEach(tools.stubConsole);
test.afterEach.always(tools.restoreConsole);

test.cb(`should list zones`, (t) => {
  const dnsMock = {
    getZones: () => {
      return dns.getZones()
        .then(([zones]) => {
          t.true(Array.isArray(zones));

          // Listing is eventually consistent, give the indexes time to update
          setTimeout(() => {
            try {
              t.true(console.log.called);
              t.deepEqual(console.log.getCall(0).args, [`Zones:`]);
              zones.forEach((zone, i) => {
                t.deepEqual(console.log.getCall(i + 1).args, [zone.name]);
              });
              t.end();
            } catch (err) {
              t.end(err);
            }
          }, 200);

          return [zones];
        });
    }
  };

  setTimeout(() => {
    proxyquire(`../quickstart`, {
      '@google-cloud/dns': sinon.stub().returns(dnsMock)
    });
  }, 5000);
});