import datetime
from time import sleep
from urllib import parse
from django.contrib.auth.models import User
from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from main.billing import BillingPeriod, RecalculateThread
from main.models import Product, ProductType, IncomingInvoice, Order, Consumption, Inventory, ProductInventory, \
OutgoingInvoice, OutgoingInvoiceProductUserPosition, OutgoingInvoiceProductPosition, UserExtension
HEADLESS = True
FIREFOX = "firefox"
CHROME = "chrome"
driver = FIREFOX
class TestCaseData1(object):
def __init__(self):
self.admin = User.objects.create(username="admin", is_staff=True)
self.admin.set_password("1234")
self.admin.save()
self.user1 = User.objects.create(username="User1", email="bla@bla.bla")
self.user1.set_password("1234")
self.user1.save()
self.user2 = User.objects.create(username="User2", email="bla@bla.bla")
self.user3 = User.objects.create(username="User3", email="bla@bla.bla")
self.product_type_drinks = ProductType.objects.create(name="Drinks")
self.product_water = Product.objects.create(name="Water", product_type=self.product_type_drinks)
self.product_soda = Product.objects.create(name="Soda", product_type=self.product_type_drinks)
self.product_beer = Product.objects.create(name="Beer", product_type=self.product_type_drinks)
incoming_invoice_date = datetime.date(2019, 1, 1)
self.incoming_invoice = IncomingInvoice.objects.create(
invoice_id="inv-001",
date=incoming_invoice_date)
self.incoming_invoice_position1 = Order.objects.create(
incoming_invoice=self.incoming_invoice,
product=self.product_soda,
each_cents=20,
count=20)
self.incoming_invoice_position2 = Order.objects.create(
incoming_invoice=self.incoming_invoice,
product=self.product_beer,
each_cents=80,
count=24)
self.consumption1 = Consumption.objects.create(product=self.product_beer, user=self.user1, count=4, date=datetime.datetime(2019, 1, 4), issued_by=self.admin)
self.consumption2 = Consumption.objects.create(product=self.product_soda, user=self.user1, count=6, date=datetime.datetime(2019, 1, 2), issued_by=self.admin)
self.consumption3 = Consumption.objects.create(product=self.product_beer, user=self.user2, count=4, date=datetime.datetime(2019, 1, 4), issued_by=self.admin)
self.consumption4 = Consumption.objects.create(product=self.product_beer, user=self.user3, count=3, date=datetime.datetime(2019, 1, 4), issued_by=self.admin)
self.inventory1 = Inventory.objects.create(date=datetime.datetime(2019, 1, 20))
self.inventory1_beer = ProductInventory.objects.create(product=self.product_beer, count=24 - 11, inventory=self.inventory1)
self.inventory1_soda = ProductInventory.objects.create(product=self.product_soda, count=20 - 7, inventory=self.inventory1)
period = BillingPeriod(self.inventory1)
period.recalculate_temporary_invoices()
invoice = period.invoices.first()
invoice.is_frozen = True
invoice.save()
invoice.inventory.may_have_changed = True
invoice.inventory.save()
def replace_input_value(web_element, value):
web_element.send_keys(Keys.CONTROL + "a")
web_element.send_keys(value)
class CoverageTest(StaticLiveServerTestCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
if driver == CHROME:
options = webdriver.ChromeOptions()
if HEADLESS:
options.add_argument("--headless")
options.add_argument("--disable-gpu")
cls.selenium = webdriver.Chrome(chrome_options=options)
elif driver == FIREFOX:
options = webdriver.FirefoxOptions()
profile = webdriver.FirefoxProfile()
if HEADLESS:
options.add_argument('-headless')
profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "text/csv")
cls.selenium = webdriver.Firefox(firefox_options=options, firefox_profile=profile)
cls.selenium.implicitly_wait(2)
@classmethod
def tearDownClass(cls):
cls.selenium.quit()
super().tearDownClass()
def setUp(self):
self.scenario_data = TestCaseData1()
self.selenium.get('%s%s' % (self.live_server_url, '/admin/login/?next=/'))
username_input = self.selenium.find_element_by_name("username")
username_input.send_keys('admin')
password_input = self.selenium.find_element_by_name("password")
password_input.send_keys('1234')
self.selenium.find_element_by_xpath('//input[@value="Log in"]').click()
self.wait_for_page()
self.assertEqual(parse.urlparse(self.selenium.current_url).path, "/")
self.recalculate_thread = RecalculateThread()
self.recalculate_thread.start()
def tearDown(self):
self.recalculate_thread.running = False
self.recalculate_thread.join()
self.selenium.get('%s%s' % (self.live_server_url, '/logout/'))
self.assertEqual(parse.urlparse(self.selenium.current_url).path, "/admin/login/")
def get_simple_list_text(self):
return set(element.text
for element
in self.selenium.find_elements_by_xpath(
'//a[@class="list-group-item list-group-item-action"]'))
def wait_for_page(self):
return self.selenium.find_elements_by_class_name("navbar-header")
def test_users(self):
self.selenium.get('%s%s' % (self.live_server_url, '/users/'))
self.assertEqual(len(self.get_simple_list_text()), 4)
self.selenium.find_element_by_xpath('//span[@class="glyphicon glyphicon-plus"]/parent::a').click()
username, mail = "NewUser", "default@default.default"
self.selenium.find_element_by_name("username").send_keys(username)
self.selenium.find_element_by_name("email").send_keys(mail)
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
created_url = self.selenium.current_url
self.assertTrue(User.objects.filter(username=username, email=mail).exists())
self.selenium.get('%s%s' % (self.live_server_url, '/users/'))
users = self.get_simple_list_text()
self.assertIn("NewUser", users)
self.assertIn("admin", users)
self.assertIn("User1", users)
self.assertIn("User2", users)
self.assertIn("User3", users)
self.assertEqual(len(users), 5)
self.selenium.get(created_url)
self.selenium.find_element_by_name("email").send_keys("2")
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
self.assertTrue(User.objects.filter(username=username, email="2" + mail).exists())
self.selenium.get(created_url)
self.selenium.find_element_by_xpath('//button[@data-toggle="modal" and @data-target="#confirm-delete"]').click()
self.selenium.find_element_by_xpath('//button[@type="submit" and @form="delete_form"]').click()
self.wait_for_page()
self.assertTrue(User.objects.filter(username=username, email="2" + mail).exists())
def test_products(self):
self.selenium.get('%s%s' % (self.live_server_url, '/products/'))
self.assertEqual(len(self.get_simple_list_text()), 3)
self.selenium.find_element_by_xpath('//span[@class="glyphicon glyphicon-plus"]/parent::a').click()
self.selenium.find_element_by_name("name").send_keys("Juice")
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
created_url = self.selenium.current_url
self.assertTrue(Product.objects.filter(name="Juice").exists())
self.selenium.get('%s%s' % (self.live_server_url, '/products/'))
products = self.get_simple_list_text()
self.assertIn("Water", products)
self.assertIn("Soda", products)
self.assertIn("Beer", products)
self.assertIn("Juice", products)
self.assertEqual(len(products), 4)
self.selenium.get(created_url)
self.selenium.find_element_by_xpath('//button[@data-toggle="modal" and @data-target="#confirm-delete"]').click()
self.selenium.find_element_by_xpath('//button[@type="submit" and @form="delete_form"]').click()
self.wait_for_page()
self.assertFalse(Product.objects.filter(name="Juice").exists())
def test_incoming_invoices(self):
self.selenium.get('%s%s' % (self.live_server_url, '/incoming_invoices/'))
self.assertEqual(len(self.get_simple_list_text()), 1)
self.selenium.find_element_by_xpath('//span[@class="glyphicon glyphicon-plus"]/parent::a').click()
self.selenium.find_element_by_name("invoice_id").send_keys("inv-002")
self.selenium.find_element_by_name("date").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_name("date").send_keys("02.01.2019")
self.selenium.find_element_by_xpath("//select[@name='product/-1']/option[text()='Water']").click()
self.selenium.find_element_by_xpath("//input[@name='count/-1']").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='count/-1']").send_keys("10")
self.selenium.find_element_by_xpath("//input[@name='each_cents/-1']").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='each_cents/-1']").send_keys("30")
self.selenium.find_element_by_xpath("//select[@name='product/-2']/option[text()='Soda']").click()
self.selenium.find_element_by_xpath("//input[@name='count/-2']").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='count/-2']").send_keys("20")
self.selenium.find_element_by_xpath("//input[@name='each_cents/-2']").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='each_cents/-2']").send_keys("50")
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
created_url = self.selenium.current_url
inv = IncomingInvoice.objects.get(invoice_id="inv-002", date=datetime.datetime(2019, 1, 2))
orders = [(order.product.name, order.count, order.each_cents) for order in inv.order_set.all()]
self.assertEqual(len(orders), 2)
self.assertIn(("Soda", 20, 50), orders)
self.assertIn(("Water", 10, 30), orders)
self.selenium.get('%s%s' % (self.live_server_url, '/incoming_invoices/'))
invoices = [i.split("\n")[0] for i in self.get_simple_list_text()]
self.assertIn("inv-001", invoices)
self.assertIn("inv-002", invoices)
self.assertEqual(len(invoices), 2)
self.selenium.get(created_url)
order_id = inv.order_set.first().id
self.selenium.find_element_by_xpath("//select[@name='product/%d']/option[text()='Beer']" % order_id).click()
self.selenium.find_element_by_xpath("//input[@name='count/%d']" % order_id).send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='count/%d']" % order_id).send_keys("5")
self.selenium.find_element_by_xpath("//input[@name='each_cents/%d']" % order_id).send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='each_cents/%d']" % order_id).send_keys("5")
order_id = inv.order_set.last().id
self.selenium.find_element_by_xpath("//input[@name='count/%d']" % order_id).send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='count/%d']" % order_id).send_keys(Keys.DELETE)
self.selenium.find_element_by_xpath("//input[@name='each_cents/%d']" % order_id).send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_xpath("//input[@name='each_cents/%d']" % order_id).send_keys(Keys.DELETE)
self.selenium.find_element_by_xpath('//button[text()="Save"]').send_keys(Keys.NULL)
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
inv = IncomingInvoice.objects.get(invoice_id="inv-002", date=datetime.datetime(2019, 1, 2))
orders = [(order.product.name, order.count, order.each_cents) for order in inv.order_set.all()]
self.assertEqual(len(orders), 1)
self.assertIn(("Beer", 5, 5), orders)
self.selenium.get(created_url)
self.selenium.find_element_by_xpath('//button[@data-toggle="modal" and @data-target="#confirm-delete"]').click()
self.selenium.find_element_by_xpath('//button[@type="submit" and @form="delete_form"]').click()
self.wait_for_page()
self.assertFalse(IncomingInvoice.objects.filter(invoice_id="inv-002", date=datetime.datetime(2019, 1, 2)).exists())
def test_admin_consumption(self):
self.selenium.get('%s%s' % (self.live_server_url, '/create_consumtions/'))
consumption_elements = self.selenium\
.find_element_by_class_name("table-striped")\
.find_element_by_tag_name("tbody")\
.find_elements_by_tag_name("tr")
self.assertEqual(len(consumption_elements), 4)
replace_input_value(self.selenium.find_element_by_xpath("//input[@id='cons-product/-1']"), "Soda")
replace_input_value(self.selenium.find_element_by_xpath("//input[@name='cons-count/-1']"), "5")
replace_input_value(self.selenium.find_element_by_xpath("//input[@id='cons-user/-1']"), "User1")
self.selenium.find_element_by_xpath("//button[text()='Save']").click()
consumption_elements = self.selenium\
.find_element_by_class_name("table-striped")\
.find_element_by_tag_name("tbody")\
.find_elements_by_tag_name("tr")
self.assertEqual(len(consumption_elements), 5)
self.selenium.find_elements_by_xpath("//button[contains(@class, 'glyphicon-remove')]")[0].click()
self.selenium.find_element_by_xpath("//button[text()='Delete']").click()
consumption_elements = self.selenium\
.find_element_by_class_name("table-striped")\
.find_element_by_tag_name("tbody")\
.find_elements_by_tag_name("tr")
self.assertEqual(len(consumption_elements), 4)
def test_charts(self):
self.selenium.get('%s%s' % (self.live_server_url, '/consumptions/'))
self.wait_for_page()
self.selenium.get('%s%s' % (self.live_server_url, '/'))
self.selenium.find_element_by_class_name("product-button").find_element_by_class_name("product-button-one").click()
self.selenium.find_element_by_xpath('//button[text()="OK"]').click()
self.wait_for_page()
self.selenium.get('%s%s' % (self.live_server_url, '/consumptions/'))
consumption_elements = self.selenium\
.find_element_by_class_name("table-striped")\
.find_element_by_tag_name("tbody")\
.find_elements_by_tag_name("tr")
self.assertEqual(len(consumption_elements), 1)
self.selenium.get('%s%s' % (self.live_server_url, '/charts/'))
self.wait_for_page()
self.selenium.get('%s%s' % (self.live_server_url, '/user_invoices/'))
self.wait_for_page()
def test_inventory(self):
self.selenium.get('%s%s' % (self.live_server_url, '/inventories/'))
self.assertEqual(len(self.get_simple_list_text()), 1)
self.selenium.find_element_by_xpath('//span[@class="glyphicon glyphicon-plus"]/parent::a').click()
self.selenium.find_element_by_name("date").send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_name("date").send_keys("03.01.2019")
for p_id, amount in [
(Product.objects.get(name="Beer").pk, 24),
(Product.objects.get(name="Soda").pk, 14),
(Product.objects.get(name="Water").pk, 0)
]:
self.selenium.find_element_by_name("inv-%d" % p_id).send_keys(Keys.CONTROL + "a")
self.selenium.find_element_by_name("inv-%d" % p_id).send_keys(str(amount))
self.selenium.find_element_by_xpath('//button[text()="Save"]').click()
self.wait_for_page()
sleep(2) # time to recalculate
self.selenium.get('%s%s' % (self.live_server_url, '/inventories/'))
self.wait_for_page()
self.selenium.get('%s%s' % (self.live_server_url, '/invoices/'))
self.assertEqual(len(self.get_simple_list_text()), 2)
self.selenium.find_elements_by_class_name("list-group-item")[0].click()
csv_links = self.selenium.find_elements_by_xpath('//a[@target="_blank"]')
self.assertEqual(2, len(csv_links))
for csv in csv_links:
csv.click()
self.selenium.find_element_by_xpath("//button[text()='Submit Changes']").click()
self.selenium.find_element_by_xpath("//button[text()='Submit']").click()
self.assertEqual(len(self.selenium.find_elements_by_xpath("//button[text()='Submit Changes']")), 0)
self.selenium.get('%s%s' % (self.live_server_url, '/invoices/'))
self.assertEqual(len(self.get_simple_list_text()), 2)
self.selenium.find_elements_by_class_name("list-group-item")[1].click()
self.wait_for_page()
def test_login(self):
self.selenium.get('%s%s' % (self.live_server_url, '/logout/'))
self.wait_for_page()
self.assertEqual(parse.urlparse(self.selenium.current_url).path, "/admin/login/")
self.selenium.get('%s%s' % (self.live_server_url, '/accounts/login/?next=/'))
self.selenium.find_element_by_xpath('//button[text()="User1"]').click()
password_input = self.selenium.find_element_by_name("password")
password_input.send_keys('1234')
self.selenium.find_element_by_xpath('//button[text()="Add"]').click()
self.assertEqual(parse.urlparse(self.selenium.current_url).path, "/admin/login/")
self.selenium.get('%s%s' % (self.live_server_url, '/user_invoices/'))
self.wait_for_page()
self.selenium.get('%s%s' % (self.live_server_url, '/consumptions/'))
consumption_elements = self.selenium\
.find_element_by_class_name("table-striped")\
.find_element_by_tag_name("tbody")\
.find_elements_by_tag_name("tr")
self.assertEqual(len(consumption_elements), 2)
def test_trivial(self):
for O in [OutgoingInvoice, OutgoingInvoiceProductUserPosition, OutgoingInvoiceProductPosition, Consumption,
Order, ProductType, Product, UserExtension]:
str(O.objects.first()), repr(O.objects.first())