def _check_setup(self, cons_dir): inst = SibraExtSteady() inst._get_prev_raw = create_mock() inst.setup = True inst.path_ids = "path ids" iof = create_mock(["cons_dir_flag"]) iof.cons_dir_flag = cons_dir hof = create_mock(["egress_if", "ingress_if"]) path = create_mock(["get_hof", "get_iof"]) path.get_iof.return_value = iof path.get_hof.return_value = hof spkt = create_mock(["path"]) spkt.path = path req = create_mock(["add_hop"]) inst.req_block = req # Call inst._add_hop("key", spkt) # Tests if cons_dir: req.add_hop.assert_called_once_with( hof.ingress_if, hof.egress_if, inst._get_prev_raw.return_value, "key", "path ids") else: req.add_hop.assert_called_once_with( hof.egress_if, hof.ingress_if, inst._get_prev_raw.return_value, "key", "path ids")
def test_non_setup(self, super_addhop): inst = SibraExtSteady() inst.setup = False # Call inst._add_hop("key") # Tests super_addhop.assert_called_once_with(inst, "key")