From 657bf8de364665a4a79e9fc4100adc11581508f8 Mon Sep 17 00:00:00 2001
From: Zosoled <zosoled@codecow.xyz>
Date: Thu, 4 Jul 2024 00:33:58 -0700
Subject: [PATCH] Add System.Date.addMonths()

---
 .../Date/addMonths/System_Date_addMonths.cls  | 42 +++++++++++
 .../System_Date_addMonths.cls-meta.xml        |  9 +++
 .../addMonths/System_Date_addMonths_TEST.cls  | 70 +++++++++++++++++++
 .../System_Date_addMonths_TEST.cls-meta.xml   |  9 +++
 4 files changed, 130 insertions(+)
 create mode 100644 src/System/Date/addMonths/System_Date_addMonths.cls
 create mode 100644 src/System/Date/addMonths/System_Date_addMonths.cls-meta.xml
 create mode 100644 src/System/Date/addMonths/System_Date_addMonths_TEST.cls
 create mode 100644 src/System/Date/addMonths/System_Date_addMonths_TEST.cls-meta.xml

diff --git a/src/System/Date/addMonths/System_Date_addMonths.cls b/src/System/Date/addMonths/System_Date_addMonths.cls
new file mode 100644
index 0000000..c7cd860
--- /dev/null
+++ b/src/System/Date/addMonths/System_Date_addMonths.cls
@@ -0,0 +1,42 @@
+/*
+ * SPDX-FileCopyrightText: 2024 Chris Duncan <chris@flowpex.dev>
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+global class System_Date_addMonths {
+
+    @InvocableMethod(label='System.Date.addMonths()' category='Flowpex' iconName='slds:utility:dayview' description='Adds the specified number of additional months to a Date.')
+
+    global static List<Response> System_Date_addMonths (List<Request> requests) {
+        List<Response> responses = new List<Response>();
+        for (Request req : requests) {
+            Response res = new Response();
+            try {
+                res.newDate = req.oldDate.addMonths(req.additionalMonths);
+            } catch (Exception e) {
+                System.debug(e);
+            }
+            responses.add(res);
+        }
+        return responses;
+    }
+
+    global class Request {
+        @InvocableVariable(required='true' label='Date' description='Date to be incremented.')
+        global Date oldDate;
+
+        @InvocableVariable(required='true' label='Additional Months' description='Number of months to be added to the Date.')
+        global Integer additionalMonths;
+
+        global Request () {}
+        global Request (Date d, Integer a) {
+            this.oldDate = d;
+            this.additionalMonths = a;
+        }
+    }
+
+    global class Response {
+        @InvocableVariable(label='New Date' description='Date incremented by a number of months.')
+        global Date newDate;
+    }
+}
\ No newline at end of file
diff --git a/src/System/Date/addMonths/System_Date_addMonths.cls-meta.xml b/src/System/Date/addMonths/System_Date_addMonths.cls-meta.xml
new file mode 100644
index 0000000..7d03d55
--- /dev/null
+++ b/src/System/Date/addMonths/System_Date_addMonths.cls-meta.xml
@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+SPDX-FileCopyrightText: 2024 Chris Duncan <chris@flowpex.dev>
+SPDX-License-Identifier: GPL-3.0-or-later
+-->
+<ApexClass xmlns="http://soap.sforce.com/2006/04/metadata">
+    <apiVersion>61.0</apiVersion>
+    <status>Active</status>
+</ApexClass>
\ No newline at end of file
diff --git a/src/System/Date/addMonths/System_Date_addMonths_TEST.cls b/src/System/Date/addMonths/System_Date_addMonths_TEST.cls
new file mode 100644
index 0000000..256975e
--- /dev/null
+++ b/src/System/Date/addMonths/System_Date_addMonths_TEST.cls
@@ -0,0 +1,70 @@
+/*
+ * SPDX-FileCopyrightText: 2024 Chris Duncan <chris@flowpex.dev>
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+@isTest
+private class System_Date_addMonths_TEST {
+
+    @isTest
+    static void testNoArgs () {
+        List<System_Date_addMonths.Request> requests = new List<System_Date_addMonths.Request>();
+
+        Test.startTest();
+        List<System_Date_addMonths.Response> responses = System_Date_addMonths.System_Date_addMonths(requests);
+        Test.stopTest();
+
+        Assert.areEqual(0, responses.size());
+    }
+
+    @isTest
+    static void testEmptyRequest () {
+        List<System_Date_addMonths.Request> requests = new List<System_Date_addMonths.Request>();
+        System_Date_addMonths.Request req = new System_Date_addMonths.Request();
+        requests.add(req);
+
+        Test.startTest();
+        List<System_Date_addMonths.Response> responses = System_Date_addMonths.System_Date_addMonths(requests);
+        Test.stopTest();
+
+        Assert.areEqual(1, responses.size());
+        Assert.isNull(responses[0].newDate);
+    }
+
+    @isTest
+    static void testBlankArgs () {
+        System_Date_addMonths.Request req0 = new System_Date_addMonths.Request(null, null);
+        System_Date_addMonths.Request req1 = new System_Date_addMonths.Request(Date.today(), null);
+        System_Date_addMonths.Request req2 = new System_Date_addMonths.Request(null, 1);
+        List<System_Date_addMonths.Request> requests = new List<System_Date_addMonths.Request>{req0, req1, req2};
+
+        Test.startTest();
+        List<System_Date_addMonths.Response> responses = System_Date_addMonths.System_Date_addMonths(requests);
+        Test.stopTest();
+
+        Assert.areEqual(3, responses.size());
+        Assert.isNull(responses[0].newDate);
+        Assert.isNull(responses[1].newDate);
+        Assert.isNull(responses[2].newDate);
+    }
+
+    @isTest
+    static void testManyArgs () {
+        Date today = Date.today();
+
+        Test.startTest();
+        List<System_Date_addMonths.Request> requests = new List<System_Date_addMonths.Request>();
+        for (Integer i = 0; i < 3; i++) {
+            System_Date_addMonths.Request req = new System_Date_addMonths.Request(today, i);
+            requests.add(req);
+        }
+        List<System_Date_addMonths.Response> responses = System_Date_addMonths.System_Date_addMonths(requests);
+        Test.stopTest();
+
+        Assert.areEqual(3, responses.size());
+        for (Integer i = 0; i < 3; i++) {
+            Date d = today.addMonths(i);
+            Assert.areEqual(d, responses[i].newDate);
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/System/Date/addMonths/System_Date_addMonths_TEST.cls-meta.xml b/src/System/Date/addMonths/System_Date_addMonths_TEST.cls-meta.xml
new file mode 100644
index 0000000..7d03d55
--- /dev/null
+++ b/src/System/Date/addMonths/System_Date_addMonths_TEST.cls-meta.xml
@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+SPDX-FileCopyrightText: 2024 Chris Duncan <chris@flowpex.dev>
+SPDX-License-Identifier: GPL-3.0-or-later
+-->
+<ApexClass xmlns="http://soap.sforce.com/2006/04/metadata">
+    <apiVersion>61.0</apiVersion>
+    <status>Active</status>
+</ApexClass>
\ No newline at end of file
-- 
2.52.0