diff --git a/src/classes/SmtpapiTest.cls b/src/classes/SmtpapiTest.cls index 2d0811a..912ac55 100644 --- a/src/classes/SmtpapiTest.cls +++ b/src/classes/SmtpapiTest.cls @@ -25,19 +25,22 @@ private class SmtpapiTest { @isTest static void testAddSubstitution() { Smtpapi.Header header = new Smtpapi.Header(); - List vals = new List { 'val' }; - header.addSubstitution('sub', vals); - system.assertEquals('{\"sub\":{\"sub\":[\"val\"]}}', header.jsonString()); + List firstNames = new List { 'Mike', 'Peter', 'Greg', 'Bobby' }; + header.addSubstitution('-firstName-', firstNames); + List lastNames = new List { 'Brady1', 'Brady2', 'Brady3', 'Brady4' }; + header.addSubstitution('-lastName-', lastNames); + system.assertEquals('{\"sub\":{\"-lastName-\":[\"Brady1\",\"Brady2\",\"Brady3\",\"Brady4\"],\"-firstName-\":[\"Mike\",\"Peter\",\"Greg\",\"Bobby\"]}}', header.jsonString()); } @isTest static void testSetSubstitutions() { Smtpapi.Header header = new Smtpapi.Header(); - List vals = new List { 'val' }; - Map substitutions = new Map{ 'sub' => vals }; + List firstNames = new List { 'Mike', 'Peter', 'Greg', 'Bobby' }; + List lastNames = new List { 'Brady1', 'Brady2', 'Brady3', 'Brady4' }; + Map> substitutions = new Map>{ '-firstName-' => firstNames, '-lastName-' => lastNames }; header.setSubstitutions(substitutions); - system.assertEquals('{\"sub\":{\"sub\":[\"val\"]}}', header.jsonString()); + system.assertEquals('{\"sub\":{\"-lastName-\":[\"Brady1\",\"Brady2\",\"Brady3\",\"Brady4\"],\"-firstName-\":[\"Mike\",\"Peter\",\"Greg\",\"Bobby\"]}}', header.jsonString()); } @isTest static void testAddUniqueArg() { @@ -115,4 +118,4 @@ private class SmtpapiTest { header.setFilters(filters); system.assertEquals('{\"filters\":{\"footer\":{\"settings\":{\"enable\":\"1\",\"text/plain\":\"You can haz footers!\"}}}}', header.jsonString()); } -} \ No newline at end of file +}