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())
