Adds key value integration tests

This PR primarily adds KV integration tests, but also fixes some parts of the existing KV integration

For tests this PR adds a reset method for the container
nit - we should discuss if we should document test helper methods.