from repository.MedicineRepository import MedicineRepository from controller.MedicineController import MedicineController from ui.UI import * repository = MedicineRepository() repository.add(Medicine(1, "aspirin", "AX Aspirin", 100, 5)) repository.add(Medicine(5, "aspirin", "Final Aspirin", 100, 4)) repository.add(Medicine(6, "aspirin", "Easy Aspirin", 100, 8)) repository.add(Medicine(2, "vitamin C", "C Plus (Plus?)", 200, 4)) repository.add(Medicine(3, "vitamin B", "Micro B Vitamins", 154, 6)) repository.add(Medicine(4, "magnesium", "Magnesium +", 70, 60)) controller = MedicineController(repository) ui = UI(controller) ui.mainMenu()
def testRepository(): repo = MedicineRepository() m1 = Medicine(1, "Aspirin", "BX Aspirin", 100, 10) m2 = Medicine(1, "Paracetamol", "MM Para", 80, 25) assert len(repo) == 0 ''' Test adding and retrieving medicines to/from repository ''' repo.add(m1) assert len(repo) == 1 assert repo.findById(1) == m1 try: repo.add(m1) assert False except MedicineException: pass try: repo.add(m2) assert False except MedicineException: pass m2 = Medicine(2, "Paracetamol", "MM Para", 80, 25) repo.add(m2) assert len(repo) == 2 assert repo.findById(1) == m1 assert repo.findById(2) == m2 ''' Test removing medicines from repository ''' assert len(repo) == 2 repo.remove(1) assert len(repo) == 1 assert repo.findById(2) == m2 assert repo.findById(1) == None try: repo.remove(1) assert False except MedicineException: pass assert repo.remove(2) == m2 assert len(repo) == 0