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)