Modify DiscoverUrisForSite interface.

Pass in a context-wrapper, instead of the context, to help in unit testing.