Add workspace in seed.py

This commit is contained in:
richard-dds 2018-08-21 10:48:24 -04:00
parent 92553f3b39
commit 95f7fb2026

View File

@ -8,6 +8,7 @@ sys.path.append(parent_dir)
from atst.app import make_config, make_app from atst.app import make_config, make_app
from atst.domain.users import Users from atst.domain.users import Users
from atst.domain.requests import Requests from atst.domain.requests import Requests
from atst.domain.workspaces import Workspaces
from atst.domain.exceptions import AlreadyExistsError from atst.domain.exceptions import AlreadyExistsError
from tests.factories import RequestFactory from tests.factories import RequestFactory
from atst.routes.dev import _DEV_USERS as DEV_USERS from atst.routes.dev import _DEV_USERS as DEV_USERS
@ -23,11 +24,15 @@ def seed_db():
pass pass
for user in users: for user in users:
requests = []
for dollar_value in [1, 200, 3000, 40000, 500000, 1000000]: for dollar_value in [1, 200, 3000, 40000, 500000, 1000000]:
request = Requests.create( request = Requests.create(
user, RequestFactory.build_request_body(user, dollar_value) user, RequestFactory.build_request_body(user, dollar_value)
) )
Requests.submit(request) Requests.submit(request)
requests.append(request)
Workspaces.create(request[0])
if __name__ == "__main__": if __name__ == "__main__":