def test_create_thread_local(self): self.init_database() def create_user_thread(low, hi): for _ in range(low, hi): User.create(name="u%d" % i) User.get_connection_resolver().disconnect() threads = [] for i in range(self.threads): threads.append( threading.Thread(target=create_user_thread, args=(i * 10, i * 10 + 10))) [t.start() for t in threads] [t.join() for t in threads] self.assertEqual(User.select().count(), self.threads * 10)
def reader_thread(q, num): for _ in range(num): data_queue.put(User.select().count())