Add unit and e2e sample tests