Unit tests for the log util

ui/tests/unit/helpers/format-bytes-test.js

@@ -1,7 +1,7 @@
 import { module, test } from 'ember-qunit';
 import { formatBytes } from 'nomad-ui/helpers/format-bytes';
 
-module('format-bytes', 'Unit | Helper | format-bytes');
+module('Unit | Helper | format-bytes');
 
 test('formats null/undefined as 0 bytes', function(assert) {
   assert.equal(formatBytes([undefined]), '0 Bytes');

ui/tests/unit/utils/log-test.js

@@ -0,0 +1,155 @@
+import Ember from 'ember';
+import sinon from 'sinon';
+import wait from 'ember-test-helpers/wait';
+import { module, test } from 'ember-qunit';
+import _Log from 'nomad-ui/utils/classes/log';
+
+const { Object: EmberObject, RSVP, run } = Ember;
+
+let startSpy, stopSpy, initSpy, fetchSpy;
+
+const MockStreamer = EmberObject.extend({
+  poll: {
+    isRunning: false,
+  },
+
+  init() {
+    initSpy(...arguments);
+  },
+
+  start() {
+    this.get('poll').isRunning = true;
+    startSpy(...arguments);
+  },
+
+  stop() {
+    this.get('poll').isRunning = true;
+    stopSpy(...arguments);
+  },
+
+  step(chunk) {
+    if (this.get('poll').isRunning) {
+      this.get('write')(chunk);
+    }
+  },
+});
+
+const Log = _Log.extend({
+  init() {
+    this._super();
+    const props = this.get('logStreamer').getProperties('url', 'params', 'logFetch', 'write');
+    this.set('logStreamer', MockStreamer.create(props));
+  },
+});
+
+module('Unit | Util | Log', {
+  beforeEach() {
+    initSpy = sinon.spy();
+    startSpy = sinon.spy();
+    stopSpy = sinon.spy();
+    fetchSpy = sinon.spy();
+  },
+});
+
+const makeMocks = output => ({
+  url: '/test-url/',
+  params: {
+    a: 'param',
+    another: 'one',
+  },
+  logFetch: function() {
+    fetchSpy(...arguments);
+    return RSVP.Promise.resolve({
+      text() {
+        return output;
+      },
+    });
+  },
+});
+
+test('logStreamer is created on init', function(assert) {
+  const log = Log.create(makeMocks(''));
+
+  assert.ok(log.get('logStreamer'), 'logStreamer property is defined');
+  assert.ok(initSpy.calledOnce, 'logStreamer init was called');
+});
+
+test('gotoHead builds the correct URL', function(assert) {
+  const mocks = makeMocks('');
+  const expectedUrl = `${mocks.url}?a=param&another=one&offset=0&origin=start&plain=true`;
+  const log = Log.create(mocks);
+
+  run(() => {
+    log.get('gotoHead').perform();
+    assert.ok(fetchSpy.calledWith(expectedUrl), `gotoHead URL was ${expectedUrl}`);
+  });
+});
+
+test('When gotoHead returns too large of a log, the log is truncated', function(assert) {
+  const longLog = Array(50001)
+    .fill('a')
+    .join('');
+  const truncationMessage =
+    '\n\n---------- TRUNCATED: Click "tail" to view the bottom of the log ----------';
+
+  const mocks = makeMocks(longLog);
+  const log = Log.create(mocks);
+
+  run(() => {
+    log.get('gotoHead').perform();
+  });
+
+  return wait().then(() => {
+    assert.ok(log.get('output').endsWith(truncationMessage), 'Truncation message is shown');
+    assert.equal(
+      log.get('output').length,
+      50000 + truncationMessage.length,
+      'Output is truncated the appropriate amount'
+    );
+  });
+});
+
+test('gotoTail builds the correct URL', function(assert) {
+  const mocks = makeMocks('');
+  const expectedUrl = `${mocks.url}?a=param&another=one&offset=50000&origin=end&plain=true`;
+  const log = Log.create(mocks);
+
+  run(() => {
+    log.get('gotoTail').perform();
+    assert.ok(fetchSpy.calledWith(expectedUrl), `gotoTail URL was ${expectedUrl}`);
+  });
+});
+
+test('startStreaming starts the log streamer', function(assert) {
+  const log = Log.create(makeMocks(''));
+
+  log.startStreaming();
+  assert.ok(startSpy.calledOnce, 'Streaming started');
+  assert.equal(log.get('logPointer'), 'tail', 'Streaming points the log to the tail');
+});
+
+test('When the log streamer calls `write`, the output is appended', function(assert) {
+  const log = Log.create(makeMocks(''));
+  const chunk1 = 'Hello';
+  const chunk2 = ' World';
+  const chunk3 = '\n\nEOF';
+
+  log.startStreaming();
+  assert.equal(log.get('output'), '', 'No output yet');
+
+  log.get('logStreamer').step(chunk1);
+  assert.equal(log.get('output'), chunk1, 'First chunk written');
+
+  log.get('logStreamer').step(chunk2);
+  assert.equal(log.get('output'), chunk1 + chunk2, 'Second chunk written');
+
+  log.get('logStreamer').step(chunk3);
+  assert.equal(log.get('output'), chunk1 + chunk2 + chunk3, 'Third chunk written');
+});
+
+test('stop stops the log streamer', function(assert) {
+  const log = Log.create(makeMocks(''));
+
+  log.stop();
+  assert.ok(stopSpy.calledOnce, 'Streaming stopped');
+});