Write test to check integration