94 lines
3.5 KiB
Python
94 lines
3.5 KiB
Python
from test.test_selenium import TestSelenium
|
|
from unittest import TestCase
|
|
from test.database import get_themes_orderedby_cname
|
|
from ast import literal_eval
|
|
from bs4 import BeautifulSoup
|
|
import re
|
|
MATCH = re.compile(r'^(.*)(\s\(\d*\))$')
|
|
|
|
|
|
class TestFilterByTags(TestSelenium, TestCase):
|
|
def setUp(self):
|
|
super(TestFilterByTags, self).setUp()
|
|
for x in [
|
|
'plus-button',
|
|
'button-for-showing-columns',
|
|
'tags-column-selection-input',
|
|
'button-for-showing-sort-option',
|
|
'sortByName',
|
|
'button-for-filter-by-tags',
|
|
]:
|
|
self.driver.find_element_by_id(x).click()
|
|
self.themes = get_themes_orderedby_cname()
|
|
|
|
def test_filter_by_tags(self):
|
|
filterby_tags_inputs = self.get_filter_by_tags_inputs()
|
|
|
|
for tag in filterby_tags_inputs:
|
|
self.driver.find_element_by_id(
|
|
f"{tag}-tag-selection-input").click()
|
|
|
|
themes = [x for x in self.themes if
|
|
x.tags_list is not None and tag
|
|
in literal_eval(x.tags_list)]
|
|
self.update_tags_available(themes)
|
|
|
|
tagSelectionRow = self.driver.find_element_by_id('tagSelectionRow')
|
|
buttons = BeautifulSoup(
|
|
tagSelectionRow.get_attribute(
|
|
'innerHTML'), features='lxml').findAll('button')
|
|
|
|
self.assertEqual(len(buttons), len(self.tags_available))
|
|
|
|
for button in buttons:
|
|
label_text = button.find('label').text
|
|
button_tag = MATCH.search(label_text).group(1)
|
|
button_theme_count = int(MATCH.search(
|
|
label_text).group(2)[2:-1])
|
|
self.assertEqual(
|
|
button_theme_count,
|
|
self.tags_available[button_tag]['num_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[0],
|
|
tds_txt[1],
|
|
tds_txt[2],
|
|
tds_txt[3],
|
|
set(tds_txt[4]),
|
|
],
|
|
[
|
|
themes[i].cname,
|
|
themes[i].commit_date[0:10],
|
|
str(themes[i].stargazers_count),
|
|
themes[i].commit_sha[0:6],
|
|
set(','.join(literal_eval(themes[i].tags_list))),
|
|
],
|
|
)
|
|
|
|
self.driver.find_element_by_id(
|
|
f"{tag}-tag-selection-input").click()
|
|
|
|
def get_filter_by_tags_inputs(self):
|
|
div = self.driver.find_element_by_id('tagSelectionRow')
|
|
return [x.get_attribute('id')[:-20]
|
|
for x in div.find_elements_by_tag_name('input')]
|
|
|
|
def update_tags_available(self, themes):
|
|
self.tags_available = {}
|
|
for theme in themes:
|
|
for tag in literal_eval(theme.tags_list):
|
|
if tag not in self.tags_available:
|
|
self.tags_available[tag] = {'num_themes': 1}
|
|
else:
|
|
self.tags_available[tag]['num_themes'] += 1
|