Add tests for PE id validation

2018-07-19 11:30:03 -04:00
parent 3f8b584ff8
commit 355d2ed36a
4 changed files with 111 additions and 16 deletions
--- a/atst/forms/financial.py
+++ b/atst/forms/financial.py
@@ -31,7 +31,7 @@ class FinancialForm(ValidatedForm):
    @tornado.gen.coroutine
    def perform_extra_validation(self, existing_request, fundz_client):
        valid = True
-        if existing_request['pe_id'] != self.pe_id.data:
+        if not existing_request or existing_request.get('pe_id') != self.pe_id.data:
            valid = yield validate_pe_id(self.pe_id, existing_request, fundz_client)
        raise Return(valid)