feat: update UnitOfWork + Mocker (#71)

* feat: update UnitOfWork

* dynamic mocking

src/apex/utils/classes/AssertionBuilder.cls

@@ -0,0 +1,178 @@
+public virtual class AssertionBuilder {
+    Mocker.IMockedMethod methodCall;
+
+    Map<String, Object> assertArgs;
+    Integer argSpecificTimes;
+    Integer totalTimes;
+
+    Boolean minimum = false;
+    Boolean maximum = false;
+
+    /**
+     * Begins the assertion building process
+     */
+    public AssertionBuilder begin(Mocker.IMockedMethod methodCall) {
+        this.methodCall = methodCall;
+        return this;
+    }
+
+    /**
+     * @description Asserts the method was called with the supplied args and the number of times with the specific args
+     *
+     * @param args - the args the method was called with
+     */
+    public AssertionBuilder withArgs(Map<String, Object> args) {
+        this.assertArgs = args;
+        this.argSpecificTimes = 0;
+
+        for (Map<String, Object> callArgs : this.methodCall.getCallHistory()) {
+            if (
+                args == callArgs ||
+                args.equals(callArgs) ||
+                System.hashCode(args) == System.hashCode(callArgs)
+            ) {
+                this.argSpecificTimes++;
+            }
+        }
+
+        return this;
+    }
+
+    /**
+     * @description shorthand for withArgs for single arg methods
+     *
+     * @param arg - the arg the method was called with
+     * @param value - the value the method was called with
+     */
+    public AssertionBuilder withArgs(String arg, Object value) {
+        return this.withArgs(new Map<String, Object>{ arg => value });
+    }
+
+    /**
+     * @description Asserts the total number of times the method was called
+     * if withArgs is also called, it will assert the number of times the method was called with the supplied args
+     *
+     * @param times - the number of times the method should have been called
+     */
+    public AssertionBuilder times(Integer times) {
+        this.totalTimes = times;
+        return this;
+    }
+
+    /**
+     * @description Asserts the method was called once
+     */
+    public AssertionBuilder once() {
+        return this.times(1);
+    }
+
+    public AssertionBuilder never() {
+        return this.times(0);
+    }
+
+    public AssertionBuilder atLeast(Integer times) {
+        return this.min().times(times);
+    }
+
+    public AssertionBuilder atMost(Integer times) {
+        return this.max().times(times);
+    }
+
+    public AssertionBuilder max() {
+        if (maximum) {
+            return this;
+        }
+
+        this.minimum = true;
+        return this;
+    }
+
+    public AssertionBuilder min() {
+        if (minimum) {
+            return this;
+        }
+
+        this.maximum = true;
+        return this;
+    }
+
+    /**
+     * @description Finalizes and verifies the assertions set
+     */
+    public virtual void verify() {
+        if (this.assertArgs != null && this.totalTimes != null) {
+            String message =
+                'Method ' +
+                this.methodCall.getMethodName() +
+                ' was called ' +
+                this.argSpecificTimes +
+                ' times with the supplied args';
+
+            if (minimum) {
+                Assert.isTrue(
+                    this.argSpecificTimes >= this.totalTimes,
+                    message
+                );
+                return;
+            }
+
+            if (maximum) {
+                Assert.isTrue(
+                    this.argSpecificTimes <= this.totalTimes,
+                    message
+                );
+                return;
+            }
+
+            Assert.isTrue(this.argSpecificTimes == this.totalTimes, message);
+            return;
+        }
+
+        if (this.assertArgs != null && this.totalTimes == null) {
+            String message =
+                'Method ' +
+                this.methodCall.getMethodName() +
+                ' was called ' +
+                this.argSpecificTimes +
+                ' times with the supplied args';
+            Assert.isTrue(this.argSpecificTimes > 0, message);
+            return;
+        }
+
+        if (this.totalTimes != null && this.assertArgs == null) {
+            String message =
+                'Method ' +
+                this.methodCall.getMethodName() +
+                ' was called ' +
+                this.methodCall.getCallHistory().size() +
+                ' times';
+            if (minimum) {
+                Assert.isTrue(
+                    this.methodCall.getCallHistory().size() >= this.totalTimes,
+                    message
+                );
+                return;
+            }
+
+            if (maximum) {
+                Assert.isTrue(
+                    this.methodCall.getCallHistory().size() <= this.totalTimes,
+                    message
+                );
+                return;
+            }
+
+            Assert.isTrue(
+                this.methodCall.getCallHistory().size() == this.totalTimes,
+                message
+            );
+            return;
+        }
+
+        Assert.isTrue(
+            this.methodCall.getCallHistory().size() > 0,
+            'Method ' + this.methodCall.getMethodName() + ' was never called'
+        );
+        return;
+    }
+}

src/apex/utils/classes/AssertionBuilder.cls-meta.xml

@@ -0,0 +1,5 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<ApexClass xmlns="http://soap.sforce.com/2006/04/metadata">
+    <apiVersion>62.0</apiVersion>
+    <status>Active</status>
+</ApexClass>