From 343e5a52acbb40a62d66ea74822a2450596a9786 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 17 Sep 2019 15:29:33 -0400 Subject: [PATCH] Use session.refresh in tests --- tests/test_jobs.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/tests/test_jobs.py b/tests/test_jobs.py index fd6efc44..09c3e645 100644 --- a/tests/test_jobs.py +++ b/tests/test_jobs.py @@ -76,13 +76,9 @@ def csp(): def test_create_environment_job(session, csp): environment = EnvironmentFactory.create() do_create_environment(csp, environment.id) + session.refresh(environment) - environment_id = environment.id - del environment - - updated_environment = session.query(Environment).get(environment_id) - - assert updated_environment.cloud_id + assert environment.cloud_id def test_create_environment_job_is_idempotent(csp, session): @@ -95,12 +91,9 @@ def test_create_environment_job_is_idempotent(csp, session): def test_create_atat_admin_user(csp, session): environment = EnvironmentFactory.create(cloud_id="something") do_create_atat_admin_user(csp, environment.id) + session.refresh(environment) - environment_id = environment.id - del environment - updated_environment = session.query(Environment).get(environment_id) - - assert updated_environment.root_user_info + assert environment.root_user_info def test_create_environment_baseline(csp, session): @@ -108,12 +101,9 @@ def test_create_environment_baseline(csp, session): root_user_info={"credentials": csp.root_creds()} ) do_create_environment_baseline(csp, environment.id) + session.refresh(environment) - environment_id = environment.id - del environment - updated_environment = session.query(Environment).get(environment_id) - - assert updated_environment.baseline_info + assert environment.baseline_info def test_dispatch_create_environment(session, monkeypatch):