Create more varied requests in seed script
This commit is contained in:
@@ -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__":
|
||||
|
Reference in New Issue
Block a user