From fcbd92704a91a6bf0409cc2510e7f34f434cde69 Mon Sep 17 00:00:00 2001
From: Michael Lange <dingoeatingfuzz@gmail.com>
Date: Tue, 30 Jul 2019 17:20:44 -0700
Subject: [PATCH] Test coverage for streaming file component

---
 .../templates/components/streaming-file.hbs   |   2 +-
 ui/tests/integration/streaming-file-test.js   | 111 ++++++++++++++++++
 2 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100644 ui/tests/integration/streaming-file-test.js

diff --git a/ui/app/templates/components/streaming-file.hbs b/ui/app/templates/components/streaming-file.hbs
index 690dd5915..fa97a815e 100644
--- a/ui/app/templates/components/streaming-file.hbs
+++ b/ui/app/templates/components/streaming-file.hbs
@@ -1 +1 @@
-<code>{{logger.output}}</code>
\ No newline at end of file
+<code data-test-output>{{logger.output}}</code>
\ No newline at end of file
diff --git a/ui/tests/integration/streaming-file-test.js b/ui/tests/integration/streaming-file-test.js
new file mode 100644
index 000000000..f281ffed6
--- /dev/null
+++ b/ui/tests/integration/streaming-file-test.js
@@ -0,0 +1,111 @@
+import { run } from '@ember/runloop';
+import { find, settled } from '@ember/test-helpers';
+import { module, test } from 'qunit';
+import { setupRenderingTest } from 'ember-qunit';
+import hbs from 'htmlbars-inline-precompile';
+import Pretender from 'pretender';
+import { logEncode } from '../../mirage/data/logs';
+import fetch from 'nomad-ui/utils/fetch';
+import Log from 'nomad-ui/utils/classes/log';
+
+const { assign } = Object;
+
+const stringifyValues = obj =>
+  Object.keys(obj).reduce((newObj, key) => {
+    newObj[key] = obj[key].toString();
+    return newObj;
+  }, {});
+
+const makeLogger = (url, params) =>
+  Log.create({
+    url,
+    params,
+    plainText: true,
+    logFetch: url => fetch(url).then(res => res),
+  });
+
+module('Integration | Component | streaming file', function(hooks) {
+  setupRenderingTest(hooks);
+
+  hooks.beforeEach(function() {
+    this.server = new Pretender(function() {
+      this.get('/file/endpoint', () => [200, {}, 'Hello World']);
+      this.get('/file/stream', () => [200, {}, logEncode(['Hello World'], 0)]);
+    });
+  });
+
+  hooks.afterEach(function() {
+    this.server.shutdown();
+  });
+
+  const commonTemplate = hbs`
+    {{streaming-file logger=logger mode=mode isStreaming=isStreaming}}
+  `;
+
+  test('when mode is `head`, the logger signals head', async function(assert) {
+    const url = '/file/endpoint';
+    const params = { path: 'hello/world.txt', offset: 0, limit: 50000 };
+    this.setProperties({
+      logger: makeLogger(url, params),
+      mode: 'head',
+      isStreaming: false,
+    });
+
+    await this.render(commonTemplate);
+    await settled();
+
+    const request = this.server.handledRequests[0];
+    assert.equal(this.server.handledRequests.length, 1, 'One request made');
+    assert.equal(request.url.split('?')[0], url, `URL is ${url}`);
+    assert.deepEqual(
+      request.queryParams,
+      stringifyValues(assign({ origin: 'start' }, params)),
+      'Query params are correct'
+    );
+    assert.equal(find('[data-test-output]').textContent, 'Hello World');
+  });
+
+  test('when mode is `tail`, the logger signals tail', async function(assert) {
+    const url = '/file/endpoint';
+    const params = { path: 'hello/world.txt', limit: 50000 };
+    this.setProperties({
+      logger: makeLogger(url, params),
+      mode: 'tail',
+      isStreaming: false,
+    });
+
+    await this.render(commonTemplate);
+    await settled();
+
+    const request = this.server.handledRequests[0];
+    assert.equal(this.server.handledRequests.length, 1, 'One request made');
+    assert.equal(request.url.split('?')[0], url, `URL is ${url}`);
+    assert.deepEqual(
+      request.queryParams,
+      stringifyValues(assign({ origin: 'end', offset: 50000 }, params)),
+      'Query params are correct'
+    );
+    assert.equal(find('[data-test-output]').textContent, 'Hello World');
+  });
+
+  test('when mode is `streaming` and `isStreaming` is true, streaming starts', async function(assert) {
+    const url = '/file/stream';
+    const params = { path: 'hello/world.txt', limit: 50000 };
+    this.setProperties({
+      logger: makeLogger(url, params),
+      mode: 'streaming',
+      isStreaming: true,
+    });
+
+    assert.ok(true);
+
+    run.later(run, run.cancelTimers, 500);
+
+    await this.render(commonTemplate);
+    await settled();
+
+    const request = this.server.handledRequests[0];
+    assert.equal(request.url.split('?')[0], url, `URL is ${url}`);
+    assert.equal(find('[data-test-output]').textContent, 'Hello World');
+  });
+});