Add tests

[doodspav]

We need test partially so we're sure everything is correct, but also to make sure that changes don't cause regressions or break expected behaviour.