Add tests for insert operations
Minor refactor

Use `shouldReturn`