add test_filterby_tags.py

2024-12-04 08:41:29 -08:00 · 2021-08-25 20:13:18 -07:00 · 2021-08-25 20:13:18 -07:00 · 15c3001e6b
commit 15c3001e6b
parent 797104f132
1 changed files with 93 additions and 0 deletions
--- a/test/test_filterby_tags.py
+++ b/test/test_filterby_tags.py
@ -0,0 +1,93 @@
+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