K-82: General refactor and CLI unit tests

[ms-shankar]

Description

Closes issue #82

Checklist

• An issue was first created before opening this pull request
• The new code follows the kaos contribution guidelines
• I have performed a self-review of my own code
• I have commented on my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests to ensure that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

[aalhour]

Tests need refactoring, there is a lot of state management and conditional logic that will make a test pass or fail based on the state mutations within the test cases.

[aalhour]

Side-effects and managed state at the level of the test case are still present.