Add sample integration test