updated unit tests

[Hongjiequ]

🗣 Description

Updated existing unit tests by adding more testing coverage. Jira ticket: https://cset.atlassian.net/jira/software/c/projects/CPD/boards/20?modal=detail&selectedIssue=CPD-1127

🧪 Testing

All updated tests were ran and passed in tester's local.

📷 Screenshots (if appropriate)

✅ Pre-approval checklist

• [x] This PR has an informative and human-readable title.
• [x] Changes are limited to a single goal - eschew scope creep!
• [x] All future TODOs are captured in issues, which are referenced in code comments.
• [x] All relevant type-of-change labels have been added.
• [x] I have read the CONTRIBUTING document.
• [x] These code changes follow cisagov code standards.
• [x] All relevant repo and/or project documentation has been updated to reflect the changes in this PR.
• [x] Tests have been added and/or modified to cover the changes in this PR.
• [x] All new and existing tests pass.