Add unit tests as well as e2e tests

[JohannesLichtenberger]

We need a good test coverage to avoid regressions.

[mosheduminer]

For unit tests, #262 should probably be done first.