def test_sucess_create(self):
resource_xml = '<primitive id="resourceA" class="service" type="exim"/>'
cib = (
self.create_cib()
.append_to_first_tag_name('resources', resource_xml)
)
env = Env(self.mock_logger, self.mock_reporter, cib_data=str(cib))
ticket_command.create(env, "ticketA", "resourceA", {
"loss-policy": "fence",
"rsc-role": "master"
})
assert_xml_equal(
env.get_cib_xml(),
str(cib.append_to_first_tag_name(
'constraints', """
<rsc_ticket
id="ticket-ticketA-resourceA-Master"
rsc="resourceA"
rsc-role="Master"
ticket="ticketA"
loss-policy="fence"
/>
"""
))
)
def test_cib_set(self, mock_get_cib, mock_push_cib):
cib_data = "test cib data"
new_cib_data = "new test cib data"
env = LibraryEnvironment(
self.mock_logger,
self.mock_reporter,
cib_data=cib_data
)
self.assertFalse(env.is_cib_live)
self.assertEqual(cib_data, env.get_cib_xml())
self.assertEqual(0, mock_get_cib.call_count)
env.push_cib_xml(new_cib_data)
self.assertEqual(0, mock_push_cib.call_count)
self.assertEqual(new_cib_data, env.get_cib_xml())
self.assertEqual(0, mock_get_cib.call_count)
def test_cib_not_set(self, mock_get_cib, mock_push_cib):
cib_data = "test cib data"
new_cib_data = "new test cib data"
mock_get_cib.return_value = cib_data
env = LibraryEnvironment(self.mock_logger, self.mock_reporter)
self.assertTrue(env.is_cib_live)
self.assertEqual(cib_data, env.get_cib_xml())
self.assertEqual(1, mock_get_cib.call_count)
env.push_cib_xml(new_cib_data)
self.assertEqual(1, mock_push_cib.call_count)
