def test_init_6(self):
from shortening.service.models import db, User, Shortening
from peewee import IntegrityError
# Given...
db = SqliteDatabase(TEST_DB, check_same_thread=False)
db.connect()
# remove all records
Shortening.delete().execute()
# Test...
total = 1000
t0 = time.time()
for x in range(total):
success = False
while not success:
try:
s = Shortening(user=User.get(login="******"), short_url=Shortening.generate_random_shortening(),
long_url="http://google.com/")
s.save()
except IntegrityError:
pass
else:
success = True
# Shortening already exists
t1 = time.time()
diff = t1 - t0
diff_per_row = diff / total
print("Insert records with random name performance test, method 3, set size: %s, time: %s s, per row: %s s" %
(total, diff, diff_per_row))
db.close()
