Add Tests for Tree Client