From 1da336827ba3e61ed8002f07a0dad928f2ad0643 Mon Sep 17 00:00:00 2001 From: Pea Tyczynska Date: Thu, 13 Aug 2020 11:00:28 +0100 Subject: [PATCH] Make agreement test work with both error message classes --- tests/app/main/views/test_agreement.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/app/main/views/test_agreement.py b/tests/app/main/views/test_agreement.py index 8ee11e211..00591dfc0 100644 --- a/tests/app/main/views/test_agreement.py +++ b/tests/app/main/views/test_agreement.py @@ -281,6 +281,7 @@ def test_accept_agreement_page_populates( 'on_behalf_of_email': '', }, [ + 'Select an option', 'Error: Must be a number', ], ), @@ -345,7 +346,7 @@ def test_accept_agreement_page_validates( _expected_status=200, ) assert [ - error.text.strip() for error in page.select('.govuk-error-message') + error.text.strip() for error in page.select('.govuk-error-message, .error-message') ] == expected_errors