92 lines
3.8 KiB
Python
92 lines
3.8 KiB
Python
import unittest
|
|
from core.tracker import Tracker
|
|
from storage.data_manager import DataManager
|
|
from core.user import User
|
|
from core.item import Item
|
|
import shutil
|
|
import os
|
|
|
|
class TestTracker(unittest.TestCase):
|
|
def setUp(self):
|
|
self.test_dir = "test_data"
|
|
self.dm = DataManager(self.test_dir)
|
|
self.tracker = Tracker(self.dm)
|
|
|
|
# Setup dummy user and item
|
|
self.tracker.users = {"test_user": User("test_user", "Test Alice")}
|
|
self.tracker.items = {"test_item": Item("test_item", "Test Coffee", 2.50)}
|
|
self.tracker.debts = {}
|
|
self.tracker.transactions = []
|
|
|
|
def tearDown(self):
|
|
if os.path.exists(self.test_dir):
|
|
shutil.rmtree(self.test_dir)
|
|
|
|
def test_adjust_debt_new_user(self):
|
|
res = self.tracker.adjust_debt("test_user", 2.0, "admin123")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.get_user_debt("test_user"), -2.0)
|
|
|
|
def test_add_purchase(self):
|
|
res = self.tracker.add_purchase("test_user", "test_item")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.get_user_debt("test_user"), 2.50)
|
|
self.assertEqual(len(self.tracker.transactions), 1)
|
|
|
|
def test_add_user(self):
|
|
res = self.tracker.add_user("test_user_2", "Test Bob")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.users["test_user_2"].name, "Test Bob")
|
|
self.assertEqual(self.tracker.users["test_user_2"].pin, "4242")
|
|
|
|
def test_remove_user(self):
|
|
res = self.tracker.remove_user("test_user")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertNotIn("test_user", self.tracker.users)
|
|
self.assertNotIn("test_user", self.tracker.debts)
|
|
|
|
def test_edit_user(self):
|
|
res = self.tracker.edit_user("test_user", "Updated Name", "9999")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.users["test_user"].name, "Updated Name")
|
|
self.assertEqual(self.tracker.users["test_user"].pin, "9999")
|
|
|
|
def test_set_user_pin_validation(self):
|
|
res = self.tracker.set_user_pin("test_user", "invalid_pin")
|
|
self.assertEqual(res["status"], "error")
|
|
self.assertEqual(self.tracker.users["test_user"].pin, "4242") # Unchanged
|
|
|
|
res2 = self.tracker.set_user_pin("test_user", "1234")
|
|
self.assertEqual(res2["status"], "success")
|
|
self.assertEqual(self.tracker.users["test_user"].pin, "1234")
|
|
|
|
def test_add_item(self):
|
|
res = self.tracker.add_item("new_item", "Donut", 1.80)
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.items["new_item"].name, "Donut")
|
|
self.assertEqual(self.tracker.items["new_item"].price, 1.80)
|
|
|
|
def test_add_item_invalid_price(self):
|
|
res = self.tracker.add_item("invalid_item", "Cake", "-2.50")
|
|
self.assertEqual(res["status"], "error")
|
|
self.assertNotIn("invalid_item", self.tracker.items)
|
|
|
|
def test_remove_item(self):
|
|
res = self.tracker.remove_item("test_item")
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertNotIn("test_item", self.tracker.items)
|
|
|
|
def test_edit_item(self):
|
|
res = self.tracker.edit_item("test_item", "Premium Coffee", 3.20)
|
|
self.assertEqual(res["status"], "success")
|
|
self.assertEqual(self.tracker.items["test_item"].name, "Premium Coffee")
|
|
self.assertEqual(self.tracker.items["test_item"].price, 3.20)
|
|
|
|
def test_edit_item_invalid_price(self):
|
|
res = self.tracker.edit_item("test_item", "Premium Coffee", "not_a_number")
|
|
self.assertEqual(res["status"], "error")
|
|
self.assertEqual(self.tracker.items["test_item"].price, 2.50) # Unchanged
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|