darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-08 04:08:25 -04:00
Code Issues Packages Projects Releases Wiki Activity

notify-api-412 use black to enforce python coding style

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-25 09:12:23 -07:00
parent c6eb007386
commit 8c9721d8e2
201 changed files with 31660 additions and 28105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
tests/app/main/views/test_email_preview.py
View File

@@ -4,68 +4,92 @@ import pytest
@pytest.mark.parametrize(
"query_args, result", [
({}, True),
({'govuk_banner': 'false'}, 'false')
]
"query_args, result", [({}, True), ({"govuk_banner": "false"}, "false")]
)
def test_renders(client_request, mocker, query_args, result):
mocker.patch(
"app.main.views.index.HTMLEmailTemplate.__str__", return_value="rendered"
)
mocker.patch('app.main.views.index.HTMLEmailTemplate.__str__', return_value='rendered')
response = client_request.get_response("main.email_template", **query_args)
response = client_request.get_response('main.email_template', **query_args)
assert response.get_data(as_text=True) == 'rendered'
assert response.get_data(as_text=True) == "rendered"
def test_displays_both_branding(client_request, mock_get_email_branding_with_both_brand_type):
def test_displays_both_branding(
client_request, mock_get_email_branding_with_both_brand_type
):
page = client_request.get(
"main.email_template", branding_style="1", _test_page_title=False
)
page = client_request.get('main.email_template', branding_style="1", _test_page_title=False)
mock_get_email_branding_with_both_brand_type.assert_called_once_with('1')
mock_get_email_branding_with_both_brand_type.assert_called_once_with("1")
assert page.find("img", attrs={"src": re.compile("example.png$")})
assert page.select("body > table:nth-of-type(3) table > tr:nth-of-type(1) > td:nth-of-type(2)")[0]\
.get_text().strip() == 'Organization text' # brand text is set
assert (
page.select(
"body > table:nth-of-type(3) table > tr:nth-of-type(1) > td:nth-of-type(2)"
)[0]
.get_text()
.strip()
== "Organization text"
) # brand text is set
def test_displays_org_branding(client_request, mock_get_email_branding):
# mock_get_email_branding has 'brand_type' of 'org'
page = client_request.get('main.email_template', branding_style="1", _test_page_title=False)
page = client_request.get(
"main.email_template", branding_style="1", _test_page_title=False
)
mock_get_email_branding.assert_called_once_with('1')
mock_get_email_branding.assert_called_once_with("1")
assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
assert page.find("img", attrs={"src": re.compile("example.png")})
assert not page.select("body > table > tr > td[bgcolor='#f00']") # banner colour is not set
assert page.select("body > table:nth-of-type(1) > tr:nth-of-type(1) > td:nth-of-type(2)")[0]\
.get_text().strip() == 'Organization text' # brand text is set
assert not page.select(
"body > table > tr > td[bgcolor='#f00']"
) # banner colour is not set
assert (
page.select(
"body > table:nth-of-type(1) > tr:nth-of-type(1) > td:nth-of-type(2)"
)[0]
.get_text()
.strip()
== "Organization text"
) # brand text is set
def test_displays_org_branding_with_banner(
client_request, mock_get_email_branding_with_org_banner_brand_type
):
page = client_request.get('main.email_template', branding_style="1", _test_page_title=False)
page = client_request.get(
"main.email_template", branding_style="1", _test_page_title=False
)
mock_get_email_branding_with_org_banner_brand_type.assert_called_once_with('1')
mock_get_email_branding_with_org_banner_brand_type.assert_called_once_with("1")
assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
assert page.find("img", attrs={"src": re.compile("example.png")})
assert page.select("body > table > tr > td[bgcolor='#f00']") # banner colour is set
assert page.select("body > table table > tr > td > span")[0]\
.get_text().strip() == 'Organization text' # brand text is set
assert (
page.select("body > table table > tr > td > span")[0].get_text().strip()
== "Organization text"
) # brand text is set
def test_displays_org_branding_with_banner_without_brand_text(
client_request, mock_get_email_branding_without_brand_text
):
# mock_get_email_branding_without_brand_text has 'brand_type' of 'org_banner'
page = client_request.get('main.email_template', branding_style="1", _test_page_title=False)
page = client_request.get(
"main.email_template", branding_style="1", _test_page_title=False
)
mock_get_email_branding_without_brand_text.assert_called_once_with('1')
mock_get_email_branding_without_brand_text.assert_called_once_with("1")
assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
assert page.find("img", attrs={"src": re.compile("example.png")})
assert page.select("body > table > tr > td[bgcolor='#f00']") # banner colour is set
assert not page.select("body > table table > tr > td > span") == 0 # brand text is not set
assert (
not page.select("body > table table > tr > td > span") == 0
) # brand text is not set