Suggestion for integration tests with database