Add tests for client pkg