def test_docfreq_url(self):
def route(url):
self.assertEqual("https://xxx", url)
with open(self._get_path(self.DOCFREQ_PATH), "rb") as fin:
return fin.read()
gcs_backend.requests = FakeRequests(route)
with captured_output() as (out, _, _):
dump_model(self._get_args(input="https://xxx"))
self.assertEqual(out.getvalue(), self.DOCFREQ_DUMP)
def test_setup(self):
with captured_output() as (out, err, log):
root = logging.getLogger()
if len(root.handlers) == 1:
root.handlers.insert(0, logging.StreamHandler())
slogging.setup("INFO", False)
logger = logging.getLogger("test")
logger.info("success")
self.assertIn("test", err.getvalue())
self.assertIn("success", err.getvalue())
self.assertIn("1;36", err.getvalue())
def test_docfreq_id(self):
def route(url):
if gcs_backend.GCSBackend.INDEX_FILE in url:
return '{"models": {"docfreq": {' \
'"f64bacd4-67fb-4c64-8382-399a8e7db52a": ' \
'{"url": "https://xxx"}}}}'.encode()
self.assertEqual("https://xxx", url)
with open(self._get_path(self.DOCFREQ_PATH), "rb") as fin:
return fin.read()
gcs_backend.requests = FakeRequests(route)
with captured_output() as (out, err, _):
dump_model(
self._get_args(input="f64bacd4-67fb-4c64-8382-399a8e7db52a"))
self.assertEqual(out.getvalue(), self.DOCFREQ_DUMP)
self.assertFalse(err.getvalue())
def test_docfreq(self):
with captured_output() as (out, _, _):
dump_model(self._get_args(input=self._get_path(self.DOCFREQ_PATH)))
self.assertEqual(out.getvalue(), self.DOCFREQ_DUMP)