Added unit tests for secret type variables

postmanlabs · Sep 25, 2023 · 520d5aa · 520d5aa
1 parent 56f36ab
commit 520d5aa
Show file tree

Hide file tree

Showing 3 changed files with 68 additions and 2 deletions.
diff --git a/test/unit/mutation-tracker.test.js b/test/unit/mutation-tracker.test.js
@@ -163,8 +163,8 @@ describe('MutationTracker', function () {
         it('should not track invalid mutation format', function () {
             var tracker = new MutationTracker();
 
-            // expected signature is two parameters
-            tracker.track('set', 'foo', 'bar', 'baz');
+            // expected signature is three parameters
+            tracker.track('set', 'foo', 'bar', 'secret', 'baz');
 
             expect(tracker.count()).to.eql(0);
         });

diff --git a/test/unit/variable-scope.test.js b/test/unit/variable-scope.test.js
@@ -1315,6 +1315,18 @@ describe('VariableScope', function () {
             expect(scope.mutations.count()).to.equal(1);
         });
 
+        it('should track set operations with datatype when datatype is secret', function () {
+            var scope = new VariableScope();
+
+            scope.enableTracking();
+
+            scope.set('foo', 'bar', 'secret');
+
+            expect(scope).to.have.property('mutations');
+            expect(scope.mutations.count()).to.equal(1);
+            expect(scope.mutations.stream[0].length).to.equal(3);
+        });
+
         it('should track unset operations', function () {
             var scope = new VariableScope({
                 values: [{
@@ -1379,6 +1391,34 @@ describe('VariableScope', function () {
             expect(scope1.values).to.eql(scope2.values);
         });
 
+        it('should be capable of being replayed with secret datatype', function () {
+            var initialState = {
+                    values: [{
+                        key: 'foo',
+                        value: 'foo'
+                    }, {
+                        key: 'bar',
+                        value: 'bar'
+                    }]
+                },
+                scope1 = new VariableScope(initialState),
+                scope2 = new VariableScope(initialState);
+
+            scope1.enableTracking();
+
+            // add a new key
+            scope1.set('baz', 'baz', 'secret');
+            // update a key
+            scope1.set('foo', 'foo updated', 'secret');
+            // remove a key
+            scope1.unset('bar');
+
+            // replay mutations on a different object
+            scope1.mutations.applyOn(scope2);
+
+            expect(scope1.values).to.eql(scope2.values);
+        });
+
         it('should be serialized', function () {
             var scope = new VariableScope(),
                 serialized,

diff --git a/test/unit/variable.test.js b/test/unit/variable.test.js
@@ -128,6 +128,18 @@ describe('Variable', function () {
         });
     });
 
+    it('should prepopulate value and type when passed to the constructor (secret)', function () {
+        let v = new Variable({
+            value: 'Picard',
+            type: 'secret'
+        });
+
+        expect(v).to.deep.include({
+            value: 'Picard',
+            type: 'secret'
+        });
+    });
+
     it('should typecast value during construction when type is provided (number)', function () {
         var v = new Variable({
             value: '108',
@@ -227,6 +239,20 @@ describe('Variable', function () {
         expect(v.get()).to.equal(jsonValue);
     });
 
+    it('should support any data type if type is set to `secret`', function () {
+        var v = new Variable({ type: 'secret' }),
+            jsonValue = { iam: 'json' };
+
+        v.set('Picard');
+        expect(v.get()).to.equal('Picard');
+
+        v.set(3.142);
+        expect(v.get()).to.equal(3.142);
+
+        v.set(jsonValue);
+        expect(v.get()).to.equal(jsonValue);
+    });
+
     it('should type cast values to the specific type set', function () {
         var v = new Variable({
             type: 'string'