Add basic integration test

Add basic integration test