def setUp(self):
self.r_1 = Resource(self.name_1)
self.r_2 = Resource(self.name_2, data=self.data_2)
class ResourceTests(unittest.TestCase):
name_1 = 'name_1'
name_2 = 'name_2'
data_2 = 'data_2'
def setUp(self):
self.r_1 = Resource(self.name_1)
self.r_2 = Resource(self.name_2, data=self.data_2)
def test_constructor(self):
self.assertEqual(self.r_1.name, self.name_1)
self.assertEqual(self.r_1._data, None)
self.assertEqual(self.r_2.name, self.name_2)
self.assertEqual(self.r_2._data, self.data_2)
def test_checkout(self):
self.assertEqual(self.r_2.is_locked, False)
# perform checkout
checkout_2 = self.r_2.checkout()
# checkout results
self.assertEqual(isinstance(checkout_2, dict), True)
self.assertEqual(checkout_2['name'], self.name_2)
self.assertEqual(checkout_2['data'], self.data_2)
self.assertTrue(isinstance(checkout_2['locking_key'], uuid.UUID))
# checkout status
self.assertEqual(self.r_2.is_locked, True)
self.assertRaises(ResourceLockedError, self.r_2.checkout)
def test_checkin(self):
# checkin before checkout raises error.
self.assertRaises(ResourceNotLockedError, self.r_2.checkin)
# perform checkout
checkout_2 = self.r_2.checkout()
# checkin without a key, rasies error
self.assertRaises(InvalidLockingKeyError, self.r_2.checkin)
# checkin with key works
self.assertEqual(self.r_2.is_locked, True)
self.r_2.checkin(key=checkout_2['locking_key'])
self.assertEqual(self.r_2.is_locked, False)
def test_checkin2(self):
checkout_2 = self.r_2.checkout()
data_dict = {'data':'checkin_data',
'change_info':{'with':{},
'using':[]}}
self.assertRaises(AssertionError, self.r_2.checkin, data_dict=data_dict,
key=checkout_2['locking_key'])
# since checkin failed, data should be unchanged...
self.assertEqual(self.r_2.data, self.data_2)
# fix the incomplete data_dict's 'change_info' and checkin
data_dict['change_info']['by'] = 'test' # add last needed thing
self.r_2.checkin(data_dict=data_dict, key=checkout_2['locking_key'])
self.assertEqual(self.r_2.data, 'checkin_data')
self.assertTrue('at' in self.r_2.change_info.keys())
self.assertTrue('change_id' in self.r_2.change_info.keys())