Set up valid integration test
Complete implementation