pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #514 from dod-ccpo/fix-remove-sample-data

Browse Source 
update script for removing sample data to use new csp report fixtures
This commit is contained in:
dandds 2019-01-04 09:58:42 -05:00 committed by GitHub
commit a35889ddb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
script/remove_sample_data.py
View File

@ -27,7 +27,7 @@ from atst.models.mixins import AuditableMixin
from atst.domain.environments import Environments from atst.domain.environments import Environments
from atst.domain.exceptions import NotFoundError from atst.domain.exceptions import NotFoundError
from atst.domain.reports import MONTHLY_SPEND_AARDVARK, MONTHLY_SPEND_BELUGA from atst.domain.csp.reports import MockReportingProvider
from atst.domain.requests import Requests from atst.domain.requests import Requests
from atst.domain.users import Users from atst.domain.users import Users
from atst.domain.workspaces import Workspaces from atst.domain.workspaces import Workspaces
@ -66,9 +66,9 @@ def create_demo_workspace(name, data):
workspace = Requests.approve_and_create_workspace(request) workspace = Requests.approve_and_create_workspace(request)
Workspaces.update(workspace, { "name": name }) Workspaces.update(workspace, { "name": name })
for name in data: for mock_project in data["projects"]:
project = Project(workspace=workspace, name=name, description='') project = Project(workspace=workspace, name=mock_project.name, description='')
env_names = [env for env in data[name]] env_names = [env.name for env in mock_project.environments]
envs = Environments.create_many(project, env_names) envs = Environments.create_many(project, env_names)
db.session.add(project) db.session.add(project)
db.session.commit() db.session.commit()
@ -153,5 +153,5 @@ if __name__ == "__main__":
app = make_app(config) app = make_app(config)
with app.app_context(): with app.app_context():
remove_sample_data() remove_sample_data()
create_demo_workspace('Aardvark', MONTHLY_SPEND_AARDVARK) create_demo_workspace('Aardvark', MockReportingProvider.REPORT_FIXTURE_MAP["Aardvark"])
create_demo_workspace('Beluga', MONTHLY_SPEND_BELUGA) create_demo_workspace('Beluga', MockReportingProvider.REPORT_FIXTURE_MAP["Beluga"])