From 0f3131cf21d7e4e810d33f6b50b94280714a4705 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Tue, 11 Apr 2017 15:17:18 +0100 Subject: [PATCH] Refactor test_letter_jobs --- tests/app/main/views/test_letter_jobs.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/app/main/views/test_letter_jobs.py b/tests/app/main/views/test_letter_jobs.py index af95b509e..aa63c7bf3 100644 --- a/tests/app/main/views/test_letter_jobs.py +++ b/tests/app/main/views/test_letter_jobs.py @@ -52,9 +52,7 @@ def test_get_letter_jobs_returns_list_of_all_letter_jobs(logged_in_platform_admi page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') assert page.h1.string == 'Letter jobs' - table = page.find('table') - table_body = table.find('tbody') - rows = table_body.find_all('tr') + rows = page.select('table tbody tr') assert len(rows) == len(valid_letter_jobs) @@ -85,9 +83,7 @@ def test_post_letter_jobs_select_1_letter_job_submits_1_job(logged_in_platform_a page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') - table = page.find('table') - table_body = table.find('tbody') - rows = table_body.find_all('tr') + rows = page.select('table tbody tr') assert len(rows) == len(valid_letter_jobs)