DSAT-87 - Fix policy building

CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## [Unreleased](https://github.com/virtru/protect-and-track/compare/master...HEAD)
 
+- NO-REF: _patch_
+  - Fix policy building by using `policy.builder()`
 - NO-REF: _patch_
   - Use Audit from SDK
 

src/utils/VirtruWrapper.js

@@ -21,9 +21,20 @@ function _pushAction(action) {
  *
  * @param {?object} opts
  */
-function policyBuilder(opts) {
-  const builder = new Virtru.PolicyBuilder(opts);
-  let actions = [`const policy = new Virtru.PolicyBuilder(${opts ? 'policy' : ''})`];
+function policyBuilder(existingPolicy) {
+  if (existingPolicy) {
+    _pushAction({
+      title: 'Get Policy Builder',
+      code: 'const builder = existingPolicy.builder();',
+    });
+  } else {
+    _pushAction({
+      title: 'Create Policy Builder',
+      code: 'const builder = new Virtru.PolicyBuilder();',
+    });
+  }
+  const builder = existingPolicy ? existingPolicy.builder() : new Virtru.PolicyBuilder();
+  let actions = ['const policy = builder'];
   // This proxy records all calls, then logs them to the UI on `build` invocations.
   return new Proxy(builder, {
     get(target, propKey, receiver) {