Add module to mock state in tests
Add module to mock state in tests
Refactored tests to make use of stateMock