def test_get_context(): ticket = Ticket(testutil.create_ticket(ops=["read"])) ctx = Context() ticket.add_context(1, ctx) assert ticket.get_context(1) is ctx
def test_get_context_missing(): ticket = Ticket(testutil.create_ticket(ops=["read"])) with pytest.raises(KeyError): ticket.get_context(1)