Merge pull request #104 from lightsofapollo/support-put

Add support for .put closing issue #103
shepherdwind · Jun 11, 2018 · 8733300 · 8733300
2 parents 4775b74 + e43b8d3
commit 8733300
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/compile/references.js b/src/compile/references.js
@@ -265,7 +265,8 @@ module.exports = function(Velocity, utils) {
       } else if (id === 'size') {
 
         return getSize(baseRef);
-
+      } else if (id === 'put') {
+        return baseRef[this.getLiteral(property.args[0])] = this.getLiteral(property.args[1]);
       } else {
 
         ret = baseRef[id];

diff --git a/tests/compile.js b/tests/compile.js
@@ -694,6 +694,18 @@ describe('Compile', function() {
     });
   });
 
+  describe('assignment via .put', function () {
+    it('should set a key to an object', function() {
+      var vm = `
+        #set($foo = {})
+        #set($test = $foo.put('foo', 'bar'))
+        $foo["foo"]
+      `;
+      var expected = 'bar';
+      assert.equal(render(vm).trim(), expected)
+    });
+  });
+
   describe('Object|Array#toString', function() {
     it('simple object', function() {
       var vm = '$data';