Add unit tests for Proxy