def test_host_equality(self):
leaf1 = Scheduler("sc1", LEAF_SCHEDULER_TYPE)
h1 = Host("host1", "address1", 1234)
leaf1.add_child(h1)
h1p = Host("host1", "address1", 1234)
h2 = Host("host2", "address2", 1235)
# Same host
assert_that(h1 == h1, is_(True))
# One host has parent, the other doesnt
assert_that(h1 == h1p, is_(False))
# Different hosts
assert_that(h1 == h2, is_(False))
def test_scheduler_equality(self):
root = Scheduler("root", ROOT_SCHEDULER_TYPE)
leaf1 = Scheduler("sc1", LEAF_SCHEDULER_TYPE)
root.add_child(leaf1)
leaf2 = Scheduler("sc1", LEAF_SCHEDULER_TYPE)
# One leaf has parent, the other doesnt
assert_that(leaf1 == leaf2, is_(False))
# Compare a leaf to a root
assert_that(leaf1 == root, is_(False))
# compare two leafs that have an equivalent parents
root2 = Scheduler("root", ROOT_SCHEDULER_TYPE)
root2.add_child(leaf2)
assert_that(leaf1 == leaf2, is_(True))
# compare two leafs that have different owners
h1 = Host("h1", "addr", 1234)
leaf1.owner = h1
assert_that(leaf1 == leaf2, is_(False))
# compare two leafs with the same owner
leaf2.owner = h1
assert_that(leaf1 == leaf2, is_(True))
# compare two roots with the same children and grand children
leaf1.add_child(h1)
leaf2.add_child(h1)
assert_that(root == root2, is_(True))
# compare one root that has more leafs than the other
leaf3 = Scheduler("sc3", LEAF_SCHEDULER_TYPE)
root2.add_child(leaf3)
assert_that(root == root2, is_(False))
del root2.children[leaf3.id]
# compare one tree that has more hosts than the other
leaf2.add_child(Host("h2", "addr", 12345))
assert_that(root == root2, is_(False))