def test_update_bill(self):
address_u = models.Address("street", "number", "postal_code", "city")
address_c = models.Address("street", "number", "postal_code", "city")
person_u = models.Person(address_u, "last_name_u", "first_name_u",
"gsm_u", "phone_u", "mail_u", "12345",
"remark_u")
person_c = models.Person(address_c, "last_name_c", "first_name_c",
"gsm_c", "phone_c", "mail_c", "12345",
"remark_c")
user = models.User(person_u, "password")
customer = models.Customer(person_c, "evaluation")
bill = models.Bill(customer, user, "num_ref", "billing_date",
"due_date", "tva_rate")
bill.customer.evaluation = "eval"
self.assertEqual(bill.customer.evaluation, "eval")
bill.num_ref = "N/R"
self.assertEqual(bill.num_ref, "N/R")
bill.billing_date = "BD"
self.assertEqual(bill.billing_date, "BD")
bill.due_date = "DD"
self.assertEqual(bill.due_date, "DD")
bill.tva_rate = "TR"
self.assertEqual(bill.tva_rate, "TR")
bill.paid = True
self.assertEqual(bill.paid, True)
bill.invoiced = True
self.assertEqual(bill.invoiced, True)
self.assertTrue(bill.user.compare_password("password"))
def customer_db_to_object(self, customer_id):
info = self.get_customer(condition="customer_id = (?)", values=(str(customer_id)))
tmp = None
if len(info) > 0:
tmp_p = self.person_db_to_object(info[0][1])
if tmp_p is not None:
tmp = models.Customer(tmp_p, info[0][2])
tmp.id = info[0][0]
return tmp
def test_db_bill(self):
address_u = models.Address("street", "number", "postal_code", "city")
address_c = models.Address("street", "number", "postal_code", "city")
person_u = models.Person(address_u, "last_name_u", "first_name_u", "gsm_u", "phone_u", "mail_u", "12345",
"remark_u")
person_c = models.Person(address_c, "last_name_c", "first_name_c", "gsm_c", "phone_c", "mail_c", "12345",
"remark_c")
user = models.User(person_u, "password")
customer = models.Customer(person_c, "evaluation")
bill = models.Bill(customer, user, "num_ref", "billing_date", "due_date", "tva_rate")
self.db.add_bill(bill)
for i in range(5):
p = models.Product(bill, "description_"+str(i), i+1, 50.00)
self.db.add_product(p)
bill.add_product(p)
sol = [(1, 1, 1, 'num_ref', 'billing_date', 'due_date', 'tva_rate', 'False', 'False')]
tmp = self.db.get_bill()
for i in range(len(sol[0])):
self.assertEqual(tmp[0][i], sol[0][i])
sol = [(1, 1, 'description_0', 1, 50.0),
(2, 1, 'description_1', 2, 50.0),
(3, 1, 'description_2', 3, 50.0),
(4, 1, 'description_3', 4, 50.0),
(5, 1, 'description_4', 5, 50.0)]
tmp = self.db.get_product()
for i in range(len(tmp)):
for j in range(len(tmp[i])):
self.assertEqual(tmp[i][j], sol[i][j])
bill.num_ref = "N/R"
bill.due_date = "DD"
bill.invoiced = True
bill.products[4].description = "TEST MOD DESCRIPTION"
self.db.update_bill(bill)
sol = [(1, 1, 1, 'N/R', 'billing_date', 'DD', 'tva_rate', 'False', 'True')]
tmp = self.db.get_bill()
print(tmp)
for i in range(len(sol[0])):
self.assertEqual(tmp[0][i], sol[0][i])
sol = [(1, 1, 'description_0', 1, 50.0),
(2, 1, 'description_1', 2, 50.0),
(3, 1, 'description_2', 3, 50.0),
(4, 1, "TEST MOD DESCRIPTION", 4, 50.0),
(5, 1, 'description_4', 5, 50.0)]
tmp = self.db.get_product()
for i in range(len(tmp)):
for j in range(len(tmp[i])):
self.assertEqual(tmp[i][j], sol[i][j])
self.db.remove_bill(bill)
self.assertEqual(self.db.get_bill(), [])
self.assertEqual(self.db.get_product(), [])
def generate_customer(self):
person = self.person.generate_person()
customer = models.Customer(person, self.evaluation.get())
if customer is not None and customer.is_sanitized():
if self.customer is None:
self.controller.db.add_customer(customer)
else:
customer = self.customer
self.controller.db.update_customer(customer)
self.controller.observer("customer")
self.controller.show_frame(CustomerList)
else:
self.error.set(
"Erreur, il y a une erreur dans le formulaire. Vérifiez que les champs sont bien remplis"
)
return customer
def test_db_customer(self):
address = models.Address("street", "number", "postal_code", "city")
person = models.Person(address, "last_name", "first_name", "gsm", "phone", "mail", "12345", "remark")
customer = models.Customer(person, "evaluation")
self.db.add_customer(customer)
self.assertEqual(self.db.get_address(), [(1, 'street', 'number', 'postal_code', 'city')])
self.assertEqual(self.db.get_person(), [(1, 1, 'last_name', 'first_name', 'gsm', 'mail', 'phone', "12345", 'remark')])
self.assertEqual(self.db.get_customer(), [(1, 1, 'evaluation')])
self.assertEqual(self.db.get_customer(condition="evaluation =(?)", values=("evaluation",)), [(1, 1, 'evaluation')])
self.assertEqual(self.db.get_customer(condition="evaluation =(?)", values=("evaluation' OR 1=1",)), [])
customer.evaluation = "EVAL"
self.db.update_customer(customer)
tmp = self.db.get_customer()
sol = [(1, 1, 'EVAL')]
self.assertEqual(self.db.get_customer(
row="customer.customer_id, person.last_name, person.first_name, person.address_id, customer.evaluation",
join="INNER JOIN person ON person.person_id = customer.person_id"), [(1, 'last_name', 'first_name', 1, 'EVAL')])
for i in range(len(tmp[0])):
self.assertEqual(tmp[0][i], sol[0][i])
self.db.remove_customer(customer)
self.assertEqual(self.db.get_address(), [])
self.assertEqual(self.db.get_person(), [])
self.assertEqual(self.db.get_customer(), [])
def test_sql_injection(self):
address = models.Address("\" OR 1=1", "number", "postal_code", "city")
self.db.add_address(address)
self.assertEqual(self.db.get_address(), [(1, '\" OR 1=1', 'number', 'postal_code', 'city')])
test_person = models.Person(address, "\" OR 1=1 \"", "first_name", "gsm", "phone", "mail", "12345", "remark")
test_person2 = models.Person(address, "\" OR 1=1 \"", "first_name", "gsm", "phone", "mail", "12345", "remark")
self.db.add_person(test_person)
self.assertEqual(self.db.get_person(),
[(1, 1, '\" OR 1=1 \"', 'first_name', 'gsm', 'mail', 'phone', '12345', "remark")])
user = models.User(test_person2, "'OR 1=1 ;'")
self.db.add_user(user)
self.assertEqual(self.db.get_user(), [(1, 2, 'f10ffa9edec8c0e49474ef2a142b4380bf2c67dfc13d1d6c622126a538fa9c2c')])
customer = models.Customer(test_person, "'OR 1=1 ;'")
self.db.add_customer(customer)
self.assertEqual(self.db.get_customer(), [(1, 1, "'OR 1=1 ;'")])
bill = models.Bill(customer, user, "' OR 1=1;'", "billing_date", "due_date", "tva_rate")
self.db.add_bill(bill)
for i in range(5):
p = models.Product(bill, "description_"+str(i) + "' OR 1=1;'", i+1, 50.00)
self.db.add_product(p)
bill.add_product(p)
def test_db_get_object(self):
address_u = models.Address("street", "number", "postal_code", "city")
self.db.add_address(address_u)
tmp = self.db.address_db_to_object(1)
self.assertEqual(tmp.street, address_u.street)
self.assertEqual(tmp.number, address_u.number)
self.assertEqual(tmp.postal_code, address_u.postal_code)
self.assertEqual(tmp.city, address_u.city)
self.assertEqual(1, address_u.id)
person_u = models.Person(address_u, "last_name_u", "first_name_u", "gsm_u", "phone_u", "mail_u", "12345",
"remark_u")
user = models.User(person_u, "password")
self.db.add_user(user)
tmp_p = self.db.person_db_to_object(1)
self.assertEqual(person_u.last_name, tmp_p.last_name)
self.assertEqual(person_u.first_name, tmp_p.first_name)
self.assertEqual(person_u.gsm, tmp_p.gsm)
self.assertEqual(person_u.phone, tmp_p.phone)
self.assertEqual(person_u.mail, tmp_p.mail)
self.assertEqual(person_u.timestamp, tmp_p.timestamp)
self.assertEqual(person_u.remark, tmp_p.remark)
self.assertEqual(tmp_p.id, 1)
tmp_u = self.db.user_db_to_object(1)
self.assertEqual(user.id, 1)
self.assertEqual(user.password, tmp_u.password)
address_c = models.Address("street", "number", "postal_code", "city")
person_c = models.Person(address_c, "last_name_c", "first_name_c", "gsm_c", "phone_c", "mail_c", "12345",
"remark_c")
customer = models.Customer(person_c, "evaluation")
self.db.add_customer(customer)
tmp_c = self.db.customer_db_to_object(1)
self.assertEqual(tmp_c.id, 1)
self.assertEqual(customer.evaluation, tmp_c.evaluation)
self.assertEqual(tmp_c.person.id, 2)
company = models.Company(address_c, user, "gsm", "phone", "mail", "tva_number", "iban", "bic", "name")
self.db.add_company(company)
tmp_company = self.db.company_db_to_object(1)
self.assertEqual(tmp_company.name, company.name)
self.assertEqual(tmp_company.address.street, company.address.street)
self.assertEqual(tmp_company.address.id, company.address.id)
self.assertEqual(tmp_company.user.password, company.user.password)
self.assertEqual(tmp_company.user.id, company.user.id)
self.assertEqual(tmp_company.bic, company.bic)
self.assertEqual(tmp_company.iban, company.iban)
self.assertEqual(tmp_company.tva_number, company.tva_number)
self.assertEqual(tmp_company.mail, company.mail)
self.assertEqual(tmp_company.phone, company.phone)
self.assertEqual(tmp_company.gsm, company.gsm)
bill = models.Bill(customer, user, "num_ref", "billing_date", "due_date", "tva_rate")
for i in range(5):
p = models.Product(bill, "description_" + str(i), i + 1, 50.00)
bill.add_product(p)
self.db.add_bill(bill)
tmp_b = self.db.bill_db_to_object(1)
self.assertEqual(bill.customer.id, tmp_b.customer.id)
self.assertEqual(bill.user.id, tmp_b.user.id)
self.assertEqual(bill.num_ref, tmp_b.num_ref)
self.assertEqual(bill.billing_date, tmp_b.billing_date)
self.assertEqual(bill.due_date, tmp_b.due_date)
self.assertEqual(bill.tva_rate, tmp_b.tva_rate)
for k in tmp_b.products.keys():
self.assertEqual(bill.products[tmp_b.products[k].description].description, tmp_b.products[k].description)
self.assertEqual(bill.products[tmp_b.products[k].description].amount, tmp_b.products[k].amount)
self.assertEqual(bill.products[tmp_b.products[k].description].price_ht, tmp_b.products[k].price_ht)