from base.tests import BaseTestCase
from census.models import Census
from voting.models import QuestionOptionYesNo, Voting, Question, QuestionOption
from django.utils import timezone
from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from mixnet.models import Auth
from selenium import webdriver
from selenium.webdriver.common.by import By
from django.conf import settings
from selenium.webdriver.support.ui import Select
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from base import mods
from django.contrib.auth.models import User
[docs]class VisualizerTestCase(StaticLiveServerTestCase):
"""
Test case for the Visualizer functionality.
This test case includes methods to create various types of votings, simulate user participation,
and check the visualization results.
:cvar base: An instance of BaseTestCase for common test setup.
:cvar driver: The Selenium WebDriver instance for browser automation.
"""
[docs] def create_classic_voting(self):
"""
Create and return a classic voting with a test question and options.
:return: The created Voting instance.
:rtype: Voting
"""
q = Question(desc='test question', type='C')
q.save()
for i in range(5):
opt = QuestionOption(question=q, option='option {}'.format(i + 1))
opt.save()
v = Voting(name='test voting', question=q)
v.save()
a, _ = Auth.objects.get_or_create(
url=settings.BASEURL, defaults={
'me': True, 'name': 'test auth'})
a.save()
v.auths.add(a)
return v
[docs] def create_yesno_voting_started(self):
"""
Create and return a Yes/No voting with a test question and options, and set the start date.
:return: The created Voting instance.
:rtype: Voting
"""
q = Question(desc='Yes/No test question', type='Y')
q.save()
for i in range(5):
opt = QuestionOptionYesNo(
question=q,
option='option {}'.format(
i + 1))
opt.save()
v = Voting(name='test Yes/No voting', question=q)
v.save()
a, _ = Auth.objects.get_or_create(
url=settings.BASEURL, defaults={
'me': True, 'name': 'test auth'})
a.save()
v.auths.add(a)
v.start_date = timezone.now()
return v
[docs] def create_multiple_choice_voting_started(self):
"""
Create and return a multiple-choice voting with a test question and options, and set the start date.
:return: The created Voting instance.
:rtype: Voting
"""
q = Question(desc='test multiple choice question', type='M')
q.save()
for i in range(5):
opt = QuestionOption(question=q, option='option {}'.format(i + 1))
opt.save()
v = Voting(name='test voting', question=q)
v.save()
a, _ = Auth.objects.get_or_create(
url=settings.BASEURL, defaults={
'me': True, 'name': 'test auth'})
a.save()
v.auths.add(a)
v.start_date = timezone.now()
return v
[docs] def create_preference_voting_started(self):
"""
Create and return a preference voting with a test question and options, and set the start date.
:return: The created Voting instance.
:rtype: Voting
"""
q = Question(desc='test multiple choice question', type='R')
q.save()
for i in range(5):
opt = QuestionOption(question=q, option='option {}'.format(i + 1))
opt.save()
v = Voting(name='test voting', question=q)
v.save()
a, _ = Auth.objects.get_or_create(
url=settings.BASEURL, defaults={
'me': True, 'name': 'test auth'})
a.save()
v.auths.add(a)
v.start_date = timezone.now()
return v
[docs] def setUp(self):
"""
Set up the test environment.
Initialize BaseTestCase and create a headless Chrome WebDriver.
:return: None
"""
self.base = BaseTestCase()
self.base.setUp()
options = webdriver.ChromeOptions()
options.headless = True
self.driver = webdriver.Chrome(options=options)
[docs] def tearDown(self):
"""
Tear down the test environment.
Quit the WebDriver and call the tearDown method of the base class.
:return: None
"""
super().tearDown()
self.driver.quit()
self.base.tearDown()
[docs] def test_visualizer_not_started(self):
"""
Test the visualization when the voting is not started.
Access the visualization page, check the voting state, and assert that it is not started.
:return: None
"""
voting = self.create_classic_voting()
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
voting_state = self.driver.find_element(By.TAG_NAME, "h2").text
self.assertEqual(voting_state, "Voting not started")
[docs] def test_visualizer_started_no_census(self):
"""
Test the visualization when the voting is started but there is no census.
Create a voting with a test question and options, set the start date, access the visualization page,
and check the participation information.
:return: None
"""
question = Question(desc='test question', type='C')
question.save()
voting = Voting(
name='test voting',
start_date=timezone.now(),
question_id=question.id)
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
self.assertEqual(
self.driver.find_element(
By.ID, "participation").text, "-")
[docs] def test_visualizer_census_change(self):
"""
Test the visualization when the census changes.
Create a voting with a test question and options, access the visualization page,
record the census information, add new voters, and check if the census information changes.
:return: None
"""
question = Question(desc='test question', type='C')
question.save()
for i in range(5):
opt = QuestionOption(
question=question,
option="option {}".format(
i + 1))
opt.save()
voting = Voting(
name="test voting",
start_date=timezone.now(),
question=question)
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
census_before = self.driver.find_element(By.ID, "census").text
user, created = User.objects.get_or_create(username='testvoter')
user.is_active = True
user.save()
census1 = Census.objects.create(voting_id=voting.id, voter_id=user.id)
user2, created2 = User.objects.get_or_create(username='testvoter2')
user2.is_active = True
user2.save()
census2 = Census.objects.create(voting_id=voting.id, voter_id=user2.id)
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
census_after = self.driver.find_element(By.ID, "census").text
self.assertNotEqual(census_before, census_after)
[docs] def test_visualizer_started_classic_with_census(self):
"""
Test the visualization for a classic voting with a census.
Create a classic voting with a test question, add voters to the census, access the visualization page,
and check the participation percentage.
:return: None
"""
question = Question(desc="test question", type="C")
question.save()
for i in range(5):
opt = QuestionOption(
question=question,
option="option {}".format(
i + 1))
opt.save()
voting = Voting(
name="test voting",
start_date=timezone.now(),
question=question)
voting.save()
user, created = User.objects.get_or_create(username='testvoter')
user.is_active = True
user.save()
census1 = Census.objects.create(voting_id=voting.id, voter_id=user.id)
user2, created2 = User.objects.get_or_create(username='testvoter2')
user2.is_active = True
user2.save()
census2 = Census.objects.create(voting_id=voting.id, voter_id=user2.id)
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
self.assertEqual(
self.driver.find_element(
By.ID,
"participation").text,
"0.0%")
[docs] def test_visualizer_classic_finished(self):
"""
Test the visualization for a finished classic voting.
Create a classic voting with a test question, set start and end dates, access the visualization page,
and check various chart options.
:return: None
"""
question = Question(desc='test question', type='C')
question.save()
voting = Voting(
name='test voting',
start_date=timezone.now(),
end_date=timezone.now() +
timezone.timedelta(
days=1),
postproc=[],
question_id=question.id)
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
chart_select = WebDriverWait(self.driver, 10).until(
EC.presence_of_element_located((By.ID, "chart-select"))
)
chart_select = Select(chart_select)
default_option = chart_select.first_selected_option.text
self.assertEqual(default_option, "Doughnut Chart (Score)")
chart_options = [
("Polar Chart (Score)", "polar-chart-post"),
("Polar Chart (Votes)", "polar-chart-votes"),
("Radar Chart (Score)", "radar-chart-post"),
("Radar Chart (Votes)", "radar-chart-votes"),
("Bar Chart", "bar-chart"),
("Doughnut Chart (Votes)", "doughnut-chart-votes"),
]
for option_text, chart_id in chart_options:
chart_select.select_by_visible_text(option_text)
selected_option = chart_select.first_selected_option.text
self.assertEqual(selected_option, option_text)
chart_element = self.driver.find_element(By.ID, chart_id)
self.assertTrue(chart_element.is_displayed())
[docs] def test_visualizer_started_yesno_with_census(self):
"""
Test the visualization for a Yes/No voting with a census.
Create a Yes/No voting with a test question, add voters to the census, access the visualization page,
and check the participation percentage.
:return: None
"""
voting = self.create_yesno_voting_started()
voting.save()
user, created = User.objects.get_or_create(username='testvoter')
user.is_active = True
user.save()
census1 = Census.objects.create(voting_id=voting.id, voter_id=user.id)
user2, created2 = User.objects.get_or_create(username='testvoter2')
user2.is_active = True
user2.save()
census2 = Census.objects.create(voting_id=voting.id, voter_id=user2.id)
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
self.assertEqual(
self.driver.find_element(
By.ID,
"participation").text,
"0.0%")
[docs] def test_visualizer_yesno_finished(self):
"""
Test the visualization for a finished Yes/No voting.
Create a Yes/No voting with a test question, set start and end dates, access the visualization page,
and check the chart options.
:return: None
"""
voting = self.create_yesno_voting_started()
voting.end_date = timezone.now() + timezone.timedelta(days=1)
voting.postproc = []
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
voting_state = self.driver.find_element(By.TAG_NAME, "h3").text
self.assertEqual(voting_state, "Yes/No voting Results:")
chart_select = WebDriverWait(self.driver, 10).until(
EC.presence_of_element_located((By.ID, "chart-select"))
)
chart_select = Select(chart_select)
default_option = chart_select.first_selected_option.text
self.assertEqual(default_option, "Bar Chart")
chart_option = ("Polar Chart (Score)", "polar-chart-post")
chart_select.select_by_visible_text(chart_option[0])
selected_option = chart_select.first_selected_option.text
self.assertEqual(selected_option, chart_option[0])
chart_element = self.driver.find_element(By.ID, chart_option[1])
self.assertTrue(chart_element.is_displayed())
[docs] def test_visualizer_started_multiple_choice_with_census(self):
"""
Test the visualization for a multiple-choice voting with a census.
Create a multiple-choice voting with a test question, add voters to the census, access the visualization page,
and check the participation percentage.
:return: None
"""
voting = self.create_multiple_choice_voting_started()
voting.save()
user, created = User.objects.get_or_create(username='testvoter')
user.is_active = True
user.save()
census1 = Census.objects.create(voting_id=voting.id, voter_id=user.id)
user2, created2 = User.objects.get_or_create(username='testvoter2')
user2.is_active = True
user2.save()
census2 = Census.objects.create(voting_id=voting.id, voter_id=user2.id)
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
self.assertEqual(
self.driver.find_element(
By.ID,
"participation").text,
"0.0%")
[docs] def test_visualizer_multiple_choice_finished(self):
"""
Test the visualization for a finished multiple-choice voting.
Create a multiple-choice voting with a test question, set start and end dates, access the visualization page,
and check various chart options.
:return: None
"""
voting = self.create_multiple_choice_voting_started()
voting.end_date = timezone.now() + timezone.timedelta(days=1)
voting.postproc = []
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
chart_select = WebDriverWait(self.driver, 10).until(
EC.presence_of_element_located((By.ID, "chart-select"))
)
chart_select = Select(chart_select)
default_option = chart_select.first_selected_option.text
self.assertEqual(default_option, "Doughnut Chart (Score)")
chart_options = [
("Polar Chart (Score)", "polar-chart-post"),
("Polar Chart (Votes)", "polar-chart-votes"),
("Radar Chart (Score)", "radar-chart-post"),
("Radar Chart (Votes)", "radar-chart-votes"),
("Bar Chart", "bar-chart"),
("Doughnut Chart (Votes)", "doughnut-chart-votes"),
]
for option_text, chart_id in chart_options:
chart_select.select_by_visible_text(option_text)
selected_option = chart_select.first_selected_option.text
self.assertEqual(selected_option, option_text)
chart_element = self.driver.find_element(By.ID, chart_id)
self.assertTrue(chart_element.is_displayed())
[docs] def test_visualizer_started_preference_with_census(self):
"""
Test the visualization for a preference voting with a census.
Create a preference voting with a test question, add voters to the census, access the visualization page,
and check the participation percentage.
:return: None
"""
voting = self.create_preference_voting_started()
voting.save()
user, created = User.objects.get_or_create(username='testvoter')
user.is_active = True
user.save()
census1 = Census.objects.create(voting_id=voting.id, voter_id=user.id)
user2, created2 = User.objects.get_or_create(username='testvoter2')
user2.is_active = True
user2.save()
census2 = Census.objects.create(voting_id=voting.id, voter_id=user2.id)
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
self.assertEqual(
self.driver.find_element(
By.ID,
"participation").text,
"0.0%")
[docs] def test_visualizer_preference_finished(self):
"""
Test the visualization for a finished preference voting.
Create a preference voting with a test question, set start and end dates, access the visualization page,
and check various chart options.
:return: None
"""
voting = self.create_preference_voting_started()
voting.end_date = timezone.now() + timezone.timedelta(days=1)
voting.postproc = []
voting.save()
self.driver.get(f'{self.live_server_url}/visualizer/{voting.pk}/')
chart_select = WebDriverWait(self.driver, 10).until(
EC.presence_of_element_located((By.ID, "chart-select"))
)
chart_select = Select(chart_select)
default_option = chart_select.first_selected_option.text
self.assertEqual(default_option, "Doughnut Chart (Score)")
chart_options = [
("Polar Chart (Score)", "polar-chart-post"),
("Polar Chart (Votes)", "polar-chart-votes"),
("Radar Chart (Score)", "radar-chart-post"),
("Radar Chart (Votes)", "radar-chart-votes"),
("Bar Chart", "bar-chart"),
("Doughnut Chart (Votes)", "doughnut-chart-votes"),
]
for option_text, chart_id in chart_options:
chart_select.select_by_visible_text(option_text)
selected_option = chart_select.first_selected_option.text
self.assertEqual(selected_option, option_text)
chart_element = self.driver.find_element(By.ID, chart_id)
self.assertTrue(chart_element.is_displayed())