From b2becde2aeb37376823d7ed62dfd6fc6e3c2680e Mon Sep 17 00:00:00 2001
From: Eddie Antonio Santos <easantos@ualberta.ca>
Date: Tue, 8 Jan 2019 13:16:04 -0700
Subject: [PATCH] Add capabilities argument to WorkerInternalModel constructor.

---
 .../unit_tests/headless/worker-predict-dummy.js            | 6 ++++--
 common/predictive-text/worker/dummy-model.ts               | 2 +-
 common/predictive-text/worker/worker-interfaces.ts         | 7 +++++--
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/common/predictive-text/unit_tests/headless/worker-predict-dummy.js b/common/predictive-text/unit_tests/headless/worker-predict-dummy.js
index 04d0b2e545..3d05ceea8d 100644
--- a/common/predictive-text/unit_tests/headless/worker-predict-dummy.js
+++ b/common/predictive-text/unit_tests/headless/worker-predict-dummy.js
@@ -8,8 +8,10 @@ var sinon = require('sinon');
 var DummyModel = require('../../build/intermediate').models.DummyModel;
 
 describe('LMLayerWorker dummy model', function() {
-  it('can be instantiated with no arguments', function () {
-    var model = new DummyModel;
+  it('can be instantiated with capabilities', function () {
+    var model = new DummyModel({
+      maxLeftContextCodeUnits: 64,
+    });
     assert.isObject(model);
   });
 });
diff --git a/common/predictive-text/worker/dummy-model.ts b/common/predictive-text/worker/dummy-model.ts
index 4a14130324..d89620edf1 100644
--- a/common/predictive-text/worker/dummy-model.ts
+++ b/common/predictive-text/worker/dummy-model.ts
@@ -33,6 +33,6 @@
  * The Dummy Model that returns nonsensical, but predictable results. 
  */
 LMLayerWorker.models.DummyModel = class DummyModel implements WorkerInternalModel {
-  constructor() {
+  constructor(capabilities: Capabilities) {
   }
 };
diff --git a/common/predictive-text/worker/worker-interfaces.ts b/common/predictive-text/worker/worker-interfaces.ts
index c09aaeac8f..2181beaaea 100644
--- a/common/predictive-text/worker/worker-interfaces.ts
+++ b/common/predictive-text/worker/worker-interfaces.ts
@@ -73,6 +73,9 @@ interface WorkerInternalModel {
  * Constructors that return worker internal models.
  */
 interface WorkerInternalModelConstructor {
-  // TODO: new should take (cap: Capabilities)
-  new(): WorkerInternalModel;
+  /**
+   * WorkerInternalModel instances are all given the keyboard's
+   * capabilities.
+   */
+  new(capabilities: Capabilities): WorkerInternalModel;
 }