diff --git a/test/test_filterby_license.py b/test/test_filterby_license.py new file mode 100644 index 0000000..d5cdb64 --- /dev/null +++ b/test/test_filterby_license.py @@ -0,0 +1,51 @@ +from test.test_selenium import TestSelenium +from unittest import TestCase +from test.database import get_themes_orderedby_cname +from bs4 import BeautifulSoup + + +class TestFilterByLicense(TestSelenium, TestCase): + def setUp(self): + super(TestFilterByLicense, self).setUp() + for x in [ + 'plus-button', + 'button-for-showing-columns', + 'license-column-selection-input', + 'button-for-showing-sort-option', + 'sortByName', + 'button-for-filter-by-license', + ]: + self.driver.find_element_by_id(x).click() + self.themes = get_themes_orderedby_cname() + + def test_filter_by_license(self): + filterby_lic_inputs = [x.get_attribute('id')[:-24] + for x in self.driver.find_element_by_id( + 'licenseSelectionRow').find_elements_by_tag_name('input')] + + for license in filterby_lic_inputs: + self.driver.find_element_by_id( + f"{license}-license-selection-input").click() + themes = list( + filter(lambda x: x.theme_license == license, self.themes) + ) + results_table_div = self.driver.find_element_by_id('results') + rows = BeautifulSoup(results_table_div.get_attribute( + 'innerHTML'), features='lxml').find('table').findAll('tr') + + for i, row in enumerate(rows[1:]): + tds = row.findAll('td') + tds_txt = [x.text for x in tds] + + self.assertEqual( + tds_txt, + [ + themes[i].cname, + themes[i].commit_date[0:10], + str(themes[i].stargazers_count), + themes[i].commit_sha[0:6], + themes[i].theme_license, + ], + ) + self.driver.find_element_by_id( + f"{license}-license-selection-input").click()