def test_parse(self): p = self._payload([ self._digest([ self._course([ self._thread("t00", [self._item("a"), self._item("b"), self._item("c")]), self._thread("t01", [self._item("d"), self._item("e"), self._item("f")]), self._thread("t02", [self._item("g"), self._item("h"), self._item("i")]), ]), self._course([ self._thread("t10", [self._item("j"), self._item("k"), self._item("l")]), self._thread("t11", [self._item("m"), self._item("n"), self._item("o")]), self._thread("t12", [self._item("p"), self._item("q"), self._item("r")]), ]), ]), self._digest([ self._course([ self._thread("t20", [self._item("A"), self._item("B"), self._item("C")]), self._thread("t21", [self._item("D"), self._item("E"), self._item("F")]), self._thread("t22", [self._item("G"), self._item("H"), self._item("I")]), ]), self._course([ self._thread("t30", [self._item("J"), self._item("K"), self._item("L")]), self._thread("t31", [self._item("M"), self._item("N"), self._item("O")]), self._thread("t32", [self._item("P"), self._item("Q"), self._item("R")]), ]), ]), ]) digest_count = 0 for __, parsed_digest in process_cs_response(p, make_user_info(p)): self.assertIsNotNone(self._find_raw_digest(parsed_digest, p)) digest_count += 1 self.assertEqual(digest_count, len(p))
def _process_cs_response_with_user_info(self, data): mock_user_info = make_user_info(data) return process_cs_response(data, mock_user_info)