def test_remove_resource_references_and_empty_remaining_parents(self):
constraint_section = etree.fromstring(
"""
<constraints>
<rsc_ticket id="t1" ticket="tA">
<resource_set id="rs1">
<resource_ref id="rA"/>
</resource_set>
<resource_set id="rs2">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rA"/>
<resource_ref id="rB"/>
</resource_set>
<resource_set id="rs4">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
"""
)
self.assertTrue(
ticket.remove_with_resource_set(
constraint_section, ticket_key="tA", resource_id="rA"
)
)
assert_xml_equal(
"""
<constraints>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rB"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
""",
etree.tostring(constraint_section).decode(),
)
def test_remove_resource_references_and_empty_remaining_parents(self):
constraint_section = etree.fromstring("""
<constraints>
<rsc_ticket id="t1" ticket="tA">
<resource_set id="rs1">
<resource_ref id="rA"/>
</resource_set>
<resource_set id="rs2">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rA"/>
<resource_ref id="rB"/>
</resource_set>
<resource_set id="rs4">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
""")
self.assertTrue(ticket.remove_with_resource_set(
constraint_section,
ticket_key="tA",
resource_id="rA"
))
assert_xml_equal(
"""
<constraints>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rB"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
""",
etree.tostring(constraint_section).decode()
)
def remove(env, ticket_key, resource_id):
"""
remove all ticket constraint from resource
If resource is in resource set with another resources then only resource ref
is removed. If resource is alone in resource set whole constraint is removed.
"""
constraint_section = get_constraints(env.get_cib())
any_plain_removed = ticket.remove_plain(constraint_section, ticket_key,
resource_id)
any_with_resource_set_removed = ticket.remove_with_resource_set(
constraint_section, ticket_key, resource_id)
env.push_cib()
return any_plain_removed or any_with_resource_set_removed
def test_remove_nothing_when_no_matching_found(self):
constraint_section = etree.fromstring("""
<constraints>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rB"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
""")
self.assertFalse(
ticket.remove_with_resource_set(constraint_section,
ticket_key="tA",
resource_id="rA"))
def test_remove_nothing_when_no_matching_found(self):
constraint_section = etree.fromstring("""
<constraints>
<rsc_ticket id="t2" ticket="tA">
<resource_set id="rs3">
<resource_ref id="rB"/>
</resource_set>
</rsc_ticket>
<rsc_ticket id="t3" ticket="tB">
<resource_set id="rs5">
<resource_ref id="rA"/>
</resource_set>
</rsc_ticket>
</constraints>
""")
self.assertFalse(ticket.remove_with_resource_set(
constraint_section,
ticket_key="tA",
resource_id="rA"
))
def remove(env, ticket_key, resource_id):
"""
remove all ticket constraint from resource
If resource is in resource set with another resources then only resource ref
is removed. If resource is alone in resource set whole constraint is removed.
"""
cib = env.get_cib()
constraint_section = get_constraints(cib)
any_plain_removed = ticket.remove_plain(
constraint_section,
ticket_key,
resource_id
)
any_with_resource_set_removed = ticket.remove_with_resource_set(
constraint_section,
ticket_key,
resource_id
)
env.push_cib(cib)
return any_plain_removed or any_with_resource_set_removed
