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)