fourteen-testing-api - During model transform preserve the argument passed to this.subject

Author: jbryson3

__testfixtures__/fourteen-testing-api/setup-model-test.input2.js

@@ -0,0 +1,14 @@
+import { expect } from 'chai';
+import { describe, it } from 'mocha';
+import { setupModelTest } from 'ember-mocha';
+
+describe('Unit | Model | rental', function() {
+  setupModelTest('rental', {
+    needs: []
+  });
+  // Replace this with your real tests.
+  it('exists', function() {
+    let model = this.subject({num: 5, obj: {key: 'val'}, arr: [1,'two']});
+    expect(model).to.be.ok;
+  });
+});

__testfixtures__/fourteen-testing-api/setup-model-test.output2.js

@@ -0,0 +1,12 @@
+import { expect } from 'chai';
+import { describe, it } from 'mocha';
+import { setupTest } from 'ember-mocha';
+
+describe('Unit | Model | rental', function() {
+  setupTest();
+  // Replace this with your real tests.
+  it('exists', function() {
+    let model = this.owner.lookup('service:store').createRecord('rental', {num: 5, obj: {key: 'val'}, arr: [1,'two']});
+    expect(model).to.be.ok;
+  });
+});

fourteen-testing-api.js

@@ -502,6 +502,9 @@ module.exports = function(file, api) {
               )
         );
       } else if (subjectType === 'model') {
+        let createRecordArg = p.node.arguments[0] ?
+              p.node.arguments[0] :  /* the argument provided to this.subject() */
+              j.objectExpression([]) /* empty object expression {} */;
         p.replace(
               j.callExpression(
                 j.memberExpression(
@@ -516,7 +519,7 @@ module.exports = function(file, api) {
                 ),
                 // creating an empty object expression {} as the 2nd argument here
                 // because setupModelTests shouldn't need store dependencies
-                [j.literal(subjectName), j.objectExpression([])].filter(Boolean)
+                [j.literal(subjectName), createRecordArg].filter(Boolean)
               )
         );
       } else {