MVP-138: Part 3 unit tests

[BradyMitch]

🎯 Summary

MVP-138

• Added more test coverage

🔰 Checklist

• I have read and agree with the following checklist and am following the guidelines in our Code of Conduct document.

• I have performed a self-review of my code.
• I have commented my code, particularly in hard-to-understand areas.
• I have made corresponding changes to the documentation where required.
• I have tested my changes to the best of my ability.
• I have consulted with the team if introducing a new dependency.
• My changes generate no new warnings.

[TaylorFries]

tests run as expected