def test_get(self):
rs = ReadStatus('N')
self.assertEqual([(0, 'N')], rs.data)
self.assertEqual(rs.get(0), "N")
self.assertEqual(rs.get(1), "N")
self.assertEqual(rs.get(2), "N")
self.assertEqual(rs.get(100), "N")
rs.set(1, 'N')
self.assertEqual(rs.get(1), "N")
self.assertEqual(rs.get(2), "N")
self.assertEqual(rs.get(100), "N")
self.assertEqual(rs.get_range(xrange(1, 6)), ["N"] * 5)
def _initialize_data(self, user_id):
if not user_id in self.read_status:
try:
session = model.Session()
read_status = session.query(model.ReadStatus).filter_by(user_id=user_id).one()
if read_status.read_status_numbers == None:
self.read_status[user_id] = read_status.read_status_data
else:
self.read_status[user_id] = ReadStatus.from_string(read_status.read_status_numbers, read_status.read_status_markers)
session.close()
return True, 'OK'
except InvalidRequestError:
self.read_status[user_id] = ReadStatus()
session.close()
return True, 'OK'
else:
return False, 'ALREADY_ADDED'
def test_getset(self):
d = defaultdict(lambda: 'N')
rs = ReadStatus('N')
for i in range(10):
n = random.randint(0, 1000)
v = chr(random.randint(ord('A'), ord('Z')))
d[n] = v
rs.set(n, v)
self.assert_(len(rs.data) <= 1002)
self.assert_(d[n] == rs.get(n))
d = defaultdict(lambda: 'N')
rs = ReadStatus('N')
log = []
for i in range(20000):
n = random.randint(0, 10)
v = chr(random.randint(ord('A'), ord('Z')))
log.append((n, v))
d[n] = v
rs.set(n, v)
self.assert_(len(rs.data) <= 12)
self.assert_(d[n] == rs.get(n))
d = defaultdict(lambda: 'N')
rs = ReadStatus('N')
for i in range(10000):
n = abs(int(random.gauss(500, 10)))
v = chr(random.randint(ord('A'), ord('C')))
d[n] = v
rs.set(n, v)
self.assert_(d[n] == rs.get(n))
d = defaultdict(lambda: 'N')
rs = ReadStatus('N')
for i in range(10000):
n = abs(int(random.gauss(500000000, 100000000)))
v = chr(random.randint(ord('A'), ord('C')))
d[n] = v
rs.set(n, v)
self.assert_(d[n] == rs.get(n))
def test_set(self):
rs = ReadStatus('N')
self.assertEqual([(0, 'N')], rs.data)
rs.set(1, 'V')
self.assertEqual(rs.data,
[(0, 'N'), (1, 'V'), (2, 'N')])
rs.set(0, 'X')
self.assertEqual(rs.data,
[(0, 'X'), (1, 'V'), (2, 'N')])
rs.set(2, 'V')
self.assertEqual(rs.data,
[(0, 'X'), (1, 'V'), (3, 'N')])
self.assertEqual(rs.get(1), "V")
self.assertEqual(rs.get(2), "V")
self.assertEqual(rs.get(100), "N")
self.assertEqual(rs.get_range(xrange(1, 6)),
['V', 'V', 'N', 'N', 'N'])
rs = ReadStatus('N')
rs.set(5, 'K')
rs.set(0, 'X')
self.assertEqual(rs.get_range(xrange(0, 6)),
['X', 'N', 'N', 'N', 'N', 'K'])
