def test_get_reffs_contextual(self):
""" Ensure getReffs works with context """
endpoint = CTS(self.url)
endpoint.getValidReff = mock.MagicMock(return_value=GET_VALID_REFF)
text = Text("urn:cts:latinLit:phi1294.phi002.perseus-lat2", retriever=endpoint)
passage = Passage(
urn="urn:cts:latinLit:phi1294.phi002.perseus-lat2:1",
resource=GET_PASSAGE,
retriever=endpoint
)
passage.getReffs()
endpoint.getValidReff.assert_called_with(
urn="urn:cts:latinLit:phi1294.phi002.perseus-lat2:1",
level=2
)
def test_first_urn_whenreference(self):
endpoint = CTS(self.url)
endpoint.getFirstUrn = mock.MagicMock(return_value=Get_FIRST)
text = Text("urn:cts:latinLit:phi1294.phi002.perseus-lat2", retriever=endpoint)
passage = Passage(
urn="urn:cts:latinLit:phi1294.phi002.perseus-lat2:1",
resource=GET_PASSAGE,
retriever=endpoint
)
first = passage.getFirstUrn("1.1")
endpoint.getFirstUrn.assert_called_with(
"urn:cts:latinLit:phi1294.phi002.perseus-lat2:1.1"
)
self.assertEqual(
first, "1.pr",
"Parsing should be done and getFirstUrn should treat correctly full urn"
)
