add integration test
Added integration test