Refactor registry set calls

stone-payments · Dec 27, 2018 · be754ab · be754ab
1 parent 1b16caa
commit be754ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/packages/components/Printable/Printable.test.js b/packages/components/Printable/Printable.test.js
@@ -52,7 +52,9 @@ it('should return a resolved promise when `print({ print_to_paper: false })` and
 });
 
 it('should display a retry dialog in case of a printing error', () => {
-  Registry.set('$Printer.panel.shouldFail', true);
+  Registry.set(draft => {
+    draft.$Printer.panel.shouldFail = true;
+  });
 
   printable = newPrintable();
   printable.print();
@@ -67,7 +69,9 @@ it('should display a retry dialog in case of a printing error', () => {
 });
 
 it('should dispatch a "finish" event in case of not wanting to retry printing', () => {
-  Registry.set('$Printer.panel.shouldFail', true);
+  Registry.set(draft => {
+    draft.$Printer.panel.shouldFail = true;
+  });
 
   printable = newPrintable();
   printable.print();

diff --git a/packages/pos/drivers/cancellation/simulation.js b/packages/pos/drivers/cancellation/simulation.js
@@ -29,10 +29,9 @@ export function setup(Cancellation) {
       : Registry.get().$Cancellation.cancelledAmount;
 
   Cancellation.doCancellation = () => {
-    Registry.set(
-      '$Cancellation.cancelledAmount',
-      Registry.get().$Payment.authorizedAmount,
-    );
+    Registry.set(draft => {
+      draft.$Cancellation.cancelledAmount = draft.$Payment.authorizedAmount;
+    });
     Cancellation.cancellationDone();
   };
 }