def test_wait_for_provision_state_not_found(self): baremetal_client = mock.Mock() baremetal_client.node.get.return_value = None result = utils.wait_for_provision_state(baremetal_client, 'UUID', "available") self.assertEqual(result, True)
def test_wait_for_provision_state_fail(self): baremetal_client = mock.Mock() baremetal_client.node.get.return_value = mock.Mock( provision_state="not what we want") result = utils.wait_for_provision_state(baremetal_client, 'UUID', "available", loops=1, sleep=0.01) self.assertEqual(result, False)