finish adding integration test