def test_with_revocation(self, skip_peer):
up_pcbms, down_pcbms = self._mk_pcbms()
p = create_mock_full({"hashTreeRoot": b"1234"})
up_asm = create_mock_full({"isd_as()": "1-1",
"iter_pcbms()": up_pcbms,
"p": p})
down_asm = create_mock_full({"isd_as()": "2-1",
"iter_pcbms()": down_pcbms,
"p": p})
up_peer_rev = create_mock()
down_peer_rev = create_mock()
peer_revs = create_mock(["get"])
def get_side_effect(key):
data = {("1-1", 3): up_peer_rev, ("2-1", 3): down_peer_rev}
return data.get(key)
peer_revs.get.side_effect = get_side_effect
def skip_peer_side_effect(rev, ht_root):
if rev in [up_peer_rev, down_peer_rev] and ht_root == b"1234":
return True
return False
skip_peer.side_effect = skip_peer_side_effect
# Call
peers = path_combinator._find_peer_hfs(up_asm, down_asm, peer_revs)
# Tests
ntools.eq_(peers, [(up_pcbms[0].hof(), down_pcbms[0].hof(), 500)])
skip_peer.assert_has_calls(
[call(None, b"1234"), call(up_peer_rev, b"1234")], any_order=True)
def test_with_revocation(self):
up_pcbms, down_pcbms = self._mk_pcbms()
up_asm = create_mock_full({
"isd_as()": "1-ff00:0:300",
"iter_pcbms()": up_pcbms
})
down_asm = create_mock_full({
"isd_as()": "2-ff00:0:321",
"iter_pcbms()": down_pcbms
})
up_peer_rev = create_mock()
down_peer_rev = create_mock()
peer_revs = create_mock(["get"])
def get_side_effect(key):
data = {
("1-ff00:0:300", 3): up_peer_rev,
("2-ff00:0:321", 3): down_peer_rev
}
return data.get(key)
peer_revs.get.side_effect = get_side_effect
# Call
peers = path_combinator._find_peer_hfs(up_asm, down_asm, peer_revs)
# Tests
ntools.eq_(peers, [(up_pcbms[0].hof(), down_pcbms[0].hof(), 500)])
def test(self):
up_pcbms, down_pcbms = self._mk_pcbms()
up_asm = create_mock_full({"isd_as()": "1-ff00:0:300", "iter_pcbms()": up_pcbms})
down_asm = create_mock_full({"isd_as()": "2-ff00:0:321",
"iter_pcbms()": down_pcbms})
peer_revs = create_mock_full({"get()": None})
# Call
ntools.eq_(path_combinator._find_peer_hfs(up_asm, down_asm, peer_revs),
[(up_pcbms[0].hof(), down_pcbms[0].hof(), 500),
(up_pcbms[2].hof(), down_pcbms[1].hof(), 700)])
def test(self):
up_pcbms, down_pcbms = self._mk_pcbms()
p = create_mock_full({"hashTreeRoot": b"1234"})
up_asm = create_mock_full({"isd_as()": "1-1", "iter_pcbms()": up_pcbms,
"p": p})
down_asm = create_mock_full({"isd_as()": "2-1",
"iter_pcbms()": down_pcbms,
"p": p})
peer_revs = create_mock_full({"get()": None})
# Call
ntools.eq_(path_combinator._find_peer_hfs(up_asm, down_asm, peer_revs),
[(up_pcbms[0].hof(), down_pcbms[0].hof(), 500),
(up_pcbms[2].hof(), down_pcbms[1].hof(), 700)])