Refactor tests

Consider splitting the tests by modules they've testing – sources are growing...

- Basic unit tests checks implementation 
- More sophisticated integration tests verifies a feature simulating real-world application (embedding, CLI, etc.)
- Use mocks to speed up the execution process