update tests for Flask

tests/domain/test_requests.py

@@ -8,14 +8,14 @@ from tests.factories import RequestFactory
 
 
 @pytest.fixture()
-def requests(db):
-    return Requests(db)
+def requests(session):
+    return Requests()
 
 @pytest.fixture(scope="function")
-def new_request(db):
+def new_request(session):
     created_request = RequestFactory.create()
-    db.add(created_request)
-    db.commit()
+    session.add(created_request)
+    session.commit()
 
     return created_request
 
@@ -31,25 +31,22 @@ def test_nonexistent_request_raises(requests):
         requests.get(uuid4())
 
 
-@pytest.mark.gen_test
 def test_auto_approve_less_than_1m(requests, new_request):
     new_request.body = {"details_of_use": {"dollar_value": 999999}}
-    request = yield requests.submit(new_request)
+    request = requests.submit(new_request)
 
     assert request.status == 'approved'
 
 
-@pytest.mark.gen_test
 def test_dont_auto_approve_if_dollar_value_is_1m_or_above(requests, new_request):
     new_request.body = {"details_of_use": {"dollar_value": 1000000}}
-    request = yield requests.submit(new_request)
+    request = requests.submit(new_request)
 
     assert request.status == 'submitted'
 
 
-@pytest.mark.gen_test
 def test_dont_auto_approve_if_no_dollar_value_specified(requests, new_request):
     new_request.body = {"details_of_use": {}}
-    request = yield requests.submit(new_request)
+    request = requests.submit(new_request)
 
     assert request.status == 'submitted'