def test_register_cookie_bits_errors(self):
self.assertRaises(TypeError, cookies.register_cookie_bits, 't1', 3,
True)
self.assertRaises(exceptions.OutOfCookieSpaceException,
cookies.register_cookie_bits, 't1', 33)
self.assertRaises(exceptions.OutOfCookieSpaceException,
cookies.register_cookie_bits, 't1', 33, True, 'app')
cookies.register_cookie_bits('t1', 10)
cookies.register_cookie_bits('t1', 10, True, 'app')
self.assertRaises(exceptions.OutOfCookieSpaceException,
cookies.register_cookie_bits, 't2', 23)
self.assertRaises(exceptions.OutOfCookieSpaceException,
cookies.register_cookie_bits, 't2', 23, True, 'app')
def test_extract_value_from_cookie(self):
cookie_value_1 = 17
cookie_value_2 = 20
cookies.register_cookie_bits('test1', 5)
cookies.register_cookie_bits('test2', 5)
cookie, mask = cookies.get_cookie('test1', cookie_value_1)
cookie, mask = cookies.get_cookie('test2',
cookie_value_2,
old_cookie=cookie,
old_mask=mask)
self.assertEqual(cookie_value_1,
cookies.extract_value_from_cookie('test1', cookie))
self.assertEqual(cookie_value_2,
cookies.extract_value_from_cookie('test2', cookie))
def test_get_cookies_errors(self):
cookies.register_cookie_bits('test1', 3)
cookies.register_cookie_bits('test2', 5)
cookies.register_cookie_bits('test3', 5, True, 'app')
cookies.register_cookie_bits('test4', 4, True, 'app')
self.assertRaises(TypeError,
cookies.get_cookie,
'test3',
3,
is_local=True)
self.assertRaises(exceptions.CookieOverflowExcpetion,
cookies.get_cookie, 'test1', 9)
self.assertRaises(exceptions.MaskOverlapException, cookies.get_cookie,
'test2', 9, 0, 0x8)
def test_register_cookie_bits(self):
_cookies = cookies._cookies
used_bits = cookies._cookies_used_bits
cookies.register_cookie_bits('test1', 3)
cookies.register_cookie_bits('test2', 5)
cookies.register_cookie_bits('test3', 5, True, 'app')
cookies.register_cookie_bits('test4', 4, True, 'app')
self.assertEqual(cookies.CookieBitPair(0, 0x7),
_cookies[(cookies.GLOBAL_APP_NAME, 'test1')])
self.assertEqual(cookies.CookieBitPair(3, 0x1f << 3),
_cookies[(cookies.GLOBAL_APP_NAME, 'test2')])
self.assertEqual(cookies.CookieBitPair(32, 0x1f << 32),
_cookies[('app', 'test3')])
self.assertEqual(cookies.CookieBitPair(37, 0xf << (32 + 5)),
_cookies[('app', 'test4')])
self.assertEqual(8, used_bits[cookies.GLOBAL_APP_NAME])
self.assertEqual(9, used_bits['app'])
def test_register_and_get_cookies(self):
cookies.register_cookie_bits('test1', 3)
cookies.register_cookie_bits('test2', 5)
cookies.register_cookie_bits('test3', 5, True, 'app')
cookies.register_cookie_bits('test4', 4, True, 'app')
self.assertEqual((3, 0x7), cookies.get_cookie('test1', 3))
self.assertEqual((5 << 3, 0x1f << 3), cookies.get_cookie('test2', 5))
self.assertEqual((10 << 32, 0x1f << 32),
cookies.get_cookie('test3',
10,
is_local=True,
app_name='app'))
self.assertEqual((13 << (32 + 5) | 10 << 32, 0xf << 37 | 0x1f << 32),
cookies.get_cookie('test4',
13,
old_cookie=10 << 32,
old_mask=0x1f << 32,
is_local=True,
app_name='app'))
cookie, mask = cookies.get_cookie('test1', 2)
self.assertEqual((2, 0x7), cookies.get_cookie('test1', 3, cookie,
mask))
def register_local_cookie_bits(self, name, length):
cookies.register_cookie_bits(name, length, True,
self.__class__.__name__)