def setUp(self):
self.db = PostgresAccess()
print('\n')
class TestSequenceFunctions(unittest.TestCase):
def setUp(self):
self.db = PostgresAccess()
print('\n')
def gen_to_list(self, gen):
return gen.make_into_list()
def sequential_inserter(self, width):
"""Generator needs to be setup before calling this."""
while True:
try:
if width:
self.db.run_insert_w()
else:
self.db.run_insert_h()
except StopIteration:
print("Insertion Completed")
break
except:
print("An error occurred during the insertion")
break
def insert_query_all(self, width):
"""Test sequential inserting and query_all for 101 records in 101
streams. Because MySQL does not place a guarantee on query return
order, this test is rather slow."""
gen = self.db.init_insert(101, 101, width, True)
compareresult = self.gen_to_list(gen)
self.sequential_inserter(width)
#compareresult.pop(0) #test the test
#compareresult += ['LOL'] #test the test
result = self.db.run_query_all(debug=True)
if False:
print(result)
print(compareresult)
self.assertEqual(len(result), len(compareresult))
for x in compareresult:
self.assert_(x in result)
def test_query_all_width(self):
"""A simultaneous test of query_all and width-wise sequential
insertion."""
self.insert_query_all(True)
print("test_query_all_width passed")
def test_query_all_height(self):
"""A simultaneous test of query_all and height-wise sequential
insertion."""
self.insert_query_all(False)
print("test_query_all_height passed")
def test_query(self):
"""Test query over a range of records/streams"""
# want to check 1) length of result and 2) that all values in result
# are in the generator, although it would be pretty hard for them not
# to be
width = True #we'll only do one here since it really doesn't matter
gen = self.db.init_insert(101, 101, width, True)
compareresult = self.gen_to_list(gen)
self.sequential_inserter(width)
records = 10
streams = 10
result = self.db.query(records, streams, True)
self.assertEqual(len(result), records*streams)
for x in result:
self.assert_(x in compareresult)
print("test_query passed")
def test_query_single(self):
"""Test query of a single stream"""
width = True #doesn't really matter, just pick one
gen = self.db.init_insert(101, 101, width, True)
compareresult = self.gen_to_list(gen)
self.sequential_inserter(width)
records = 100
streamid = 4
result = self.db.query_single(records, streamid, True)
self.assertEqual(len(result), records)
for x in result:
self.assert_(x in compareresult)
print("test_query_single passed")
