Refactor integration tests - use before block