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()
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
