Add integration test to team model