 Write Unit tests

## Specifications

- Tool Version: 1.0.0
- AEM version: n/a

## Additional description
Write unit tests inside existing code base to cover as much of missing logic as possible. 

The goal is to cover Stubway logic and prepare Stubway for upcoming improvements and changes and to assure nothing is broken during implementing new features.