def test_6_event_query(self):
# get_event without limit and type
pt_1 = GeniusEventQuery.get_event(self.session)
assert_result = len(list(pt_1))
self.assertEqual(assert_result, 20)
# get_event with limit
pt_2 = GeniusEventQuery.get_event(self.session, None, 10)
assert_result = len(list(pt_2))
self.assertEqual(assert_result, 10)
# get_event with type
pt_3 = GeniusEventQuery.get_event(self.session, "add")
for pt in pt_3:
self.assertEqual(pt.event_type, "add")
pt.event_type = "more"
self.assertEqual(pt.event_type, "more")
def test_6_event_query(self):
# get_event without limit and type
pt_1 = GeniusEventQuery.get_event(self.session)
assert_result = len(list(pt_1))
self.assertEqual(assert_result, 20)
# get_event with limit
pt_2 = GeniusEventQuery.get_event(self.session, None, 10)
assert_result = len(list(pt_2))
self.assertEqual(assert_result, 10)
# get_event with type
pt_3 = GeniusEventQuery.get_event(self.session, "add")
for pt in pt_3:
self.assertEqual(pt.event_type, "add")
pt.event_type = "more"
self.assertEqual(pt.event_type, "more")
def test_3_add_event(self):
for x in range(self.event_num):
self.session.flush()
input_type = self.event_type[random.randint(0, self.index_type-1)]
tmpx = random.randint(0, self.index_device-1)
input_device = GeniusDeviceQuery.get_by_name(self.session, self.test_device[tmpx])
tmp = random.randint(0, self.index_tstname-1)
input_user = GeniusUserQuery.get_by_name(self.session, self.test_name[tmp])
input_event = GeniusEvent(event_type=input_type, device_id=input_device.id, user_id=input_user.id)
try:
self.session.flush()
self.session.add(input_event)
self.session.commit()
except Exception as e:
self.session.rollback()
print(e)
query_res = GeniusEventQuery.get_all(self.session)
assert_result = len(query_res)
self.assertEqual(assert_result, self.event_num)
def test_3_add_event(self):
for x in range(self.event_num):
self.session.flush()
input_type = self.event_type[random.randint(0, self.index_type - 1)]
tmpx = random.randint(0, self.index_device - 1)
input_device = GeniusDeviceQuery.get_by_name(self.session, self.test_device[tmpx])
tmp = random.randint(0, self.index_tstname - 1)
input_user = GeniusUserQuery.get_by_name(self.session, self.test_name[tmp])
input_event = GeniusEvent(event_type=input_type, device_id=input_device.id, user_id=input_user.id)
try:
self.session.flush()
self.session.add(input_event)
self.session.commit()
except Exception as e:
self.session.rollback()
print(e)
query_res = GeniusEventQuery.get_all(self.session)
assert_result = len(query_res)
self.assertEqual(assert_result, self.event_num)
def event_query(self):
return GeniusEventQuery(DBSession)
