Refactor Test Suite

[abishekk92]

The current test suite is unmaintainable in the long run for the following reasons.

1. Very large test files
2. Very difficult to get context on what it is testing or understand the expected behaviour
3. Works the opposite of confidence inducing when making changes.

Refactor the test suite, so that

1. Each component is tested individually
2. Each test scenario doesn't have dependency on the other