Create more varied requests in seed script

This commit is contained in:
richard-dds
2018-08-08 13:42:48 -04:00
parent 5f59b9f24a
commit df93db4cd8
3 changed files with 63 additions and 8 deletions

View File

@@ -8,14 +8,26 @@ sys.path.append(parent_dir)
from atst.app import make_config, make_app
from atst.domain.users import Users
from atst.domain.requests import Requests
from atst.domain.exceptions import AlreadyExistsError
from tests.factories import RequestFactory
from atst.routes.dev import _DEV_USERS as DEV_USERS
def seed_db():
users = [Users.create(**dev_user) for (_, dev_user) in DEV_USERS.items()]
users = []
for dev_user in DEV_USERS.values():
try:
user = Users.create(**dev_user)
users.append(user)
except AlreadyExistsError:
pass
for user in users:
[Requests.create(user, {}) for _ in range(5)]
for dollar_value in [1, 200, 3000, 40000, 500000, 1000000]:
request = Requests.create(
user, RequestFactory.build_request_body(user, dollar_value)
)
Requests.submit(request)
if __name__ == "__main__":