Add trivial tests to the plugin

It's hard to test lazy collections