Example #1
0
 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)
Example #2
0
 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)