From bba355db9de8cbb71a7b61a09fa76eca43c292d9 Mon Sep 17 00:00:00 2001
From: Trent Palmer <trenttdually@gmail.com>
Date: Thu, 26 Aug 2021 09:09:10 -0700
Subject: [PATCH] add test_filterby_features.py

---
 test/test_filterby_features.py | 91 ++++++++++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 test/test_filterby_features.py

diff --git a/test/test_filterby_features.py b/test/test_filterby_features.py
new file mode 100644
index 0000000..7753147
--- /dev/null
+++ b/test/test_filterby_features.py
@@ -0,0 +1,91 @@
+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
+from test.test_filterby_tags import MATCH
+
+
+class TestFilterByFeatures(TestSelenium, TestCase):
+    def setUp(self):
+        super(TestFilterByFeatures, self).setUp()
+        for x in [
+                'plus-button',
+                'button-for-showing-columns',
+                'features-column-selection-input',
+                'button-for-showing-sort-option',
+                'sortByName',
+                'button-for-filter-by-features',
+        ]:
+            self.driver.find_element_by_id(x).click()
+
+    def test_filterby_features(self):
+        filterby_features_inputs = self.get_filter_by_features_inputs()
+
+        for feature in filterby_features_inputs:
+            self.driver.find_element_by_id(
+                f"{feature}-feature-selection-input").click()
+
+            self.themes = [x for x in get_themes_orderedby_cname() if
+                           x.features_list is not None and feature
+                           in literal_eval(x.features_list)]
+            self.update_features_available()
+
+            featureSelectionRow = self.driver.find_element_by_id(
+                'featureSelectionRow')
+            buttons = BeautifulSoup(
+                featureSelectionRow.get_attribute(
+                    'innerHTML'), features='lxml').findAll('button')
+
+            for button in buttons:
+                label_text = button.find('label').text
+                button_feature = MATCH.search(label_text).group(1)
+                button_theme_count = int(MATCH.search(
+                    label_text).group(2)[2:-1])
+                self.assertEqual(
+                    button_theme_count,
+                    self.features_available[button_feature]['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]),
+                    ],
+                    [
+                        self.themes[i].cname,
+                        self.themes[i].commit_date[:10],
+                        str(self.themes[i].stargazers_count),
+                        self.themes[i].commit_sha[:6],
+                        set(','.join(literal_eval(
+                            self.themes[i].features_list))),
+                    ],
+                )
+
+            self.driver.find_element_by_id(
+                f"{feature}-feature-selection-input").click()
+
+    def get_filter_by_features_inputs(self):
+        div = self.driver.find_element_by_id('featureSelectionRow')
+        return [x.get_attribute('id')[:-24]
+                for x in div.find_elements_by_tag_name('input')]
+
+    def update_features_available(self):
+        self.features_available = {}
+        for theme in self.themes:
+            for feature in literal_eval(theme.features_list):
+                if feature not in self.features_available:
+                    self.features_available[feature] = {'num_themes': 1}
+                else:
+                    self.features_available[feature]['num_themes'] += 1