update tests for Flask

atst/forms/financial.py

@@ -1,12 +1,11 @@
 import re
-import tornado
-from tornado.gen import Return
 from wtforms.fields.html5 import EmailField
 from wtforms.fields import StringField, SelectField
 from wtforms.form import Form
 from wtforms.validators import Required, Email
 
 from atst.domain.exceptions import NotFoundError
+from atst.domain.pe_numbers import PENumbers
 
 from .fields import NewlineListField
 from .forms import ValidatedForm
@@ -40,9 +39,9 @@ def suggest_pe_id(pe_id):
     return None
 
 
-def validate_pe_id(field, existing_request, pe_numbers_repo):
+def validate_pe_id(field, existing_request):
     try:
-        pe_number = pe_numbers_repo.get(field.data)
+        pe_number = PENumbers.get(field.data)
     except NotFoundError:
         suggestion = suggest_pe_id(field.data)
         error_str = (
@@ -50,17 +49,17 @@ def validate_pe_id(field, existing_request, pe_numbers_repo):
             "If you have double checked it you can submit anyway. "
             "Your request will need to go through a manual review."
         ).format('Did you mean "{}"? '.format(suggestion) if suggestion else "")
-        field.errors.append(error_str)
+        field.errors += (error_str,)
         return False
 
     return True
 
 
 class FinancialForm(ValidatedForm):
-    def perform_extra_validation(self, existing_request, pe_numbers_repo):
+    def perform_extra_validation(self, existing_request):
         valid = True
         if not existing_request or existing_request.get("pe_id") != self.pe_id.data:
-            valid = yield validate_pe_id(self.pe_id, existing_request, pe_numbers_repo)
+            valid = validate_pe_id(self.pe_id, existing_request)
         return valid
 
     task_order_id = StringField(