From 95f7fb202639021eb4f66b360339919d7ad6770b Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 21 Aug 2018 10:48:24 -0400 Subject: [PATCH] Add workspace in seed.py --- script/seed.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/script/seed.py b/script/seed.py index d865b9d7..7ddc7cf7 100644 --- a/script/seed.py +++ b/script/seed.py @@ -8,6 +8,7 @@ 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.workspaces import Workspaces from atst.domain.exceptions import AlreadyExistsError from tests.factories import RequestFactory from atst.routes.dev import _DEV_USERS as DEV_USERS @@ -23,11 +24,15 @@ def seed_db(): pass for user in users: + requests = [] for dollar_value in [1, 200, 3000, 40000, 500000, 1000000]: request = Requests.create( user, RequestFactory.build_request_body(user, dollar_value) ) Requests.submit(request) + requests.append(request) + + Workspaces.create(request[0]) if __name__ == "__main__":