def setUp(self):
studs = EntityRepository([(1, 'Ratiu Cosmina'), (2, 'Kiraly Alex'),
(3, 'Popa Andra'), (4, 'Ocian Eduard'),
(5, 'Groza Segiu')])
discs = EntityRepository([(1, 'Mate'), (2, 'Romana'), (3, 'Engleza')])
self.__studCtrl = EntityController(studs)
self.__discCtrl = EntityController(discs)
enrolled = [(1, 1), (1, 2), (2, 2), (2, 3), (3, 1), (3, 3), (4, 3),
(4, 2), (5, 1), (5, 2)]
grades = [
Grade(1, 1, 9.8),
Grade(1, 2, 7.5),
Grade(2, 2, 6.8),
Grade(2, 3, 8.8),
Grade(3, 1, 4.5),
Grade(3, 3, 7.3),
Grade(4, 3, 9.6),
Grade(4, 2, 6),
Grade(5, 1, 5.5),
Grade(5, 2, 4.5)
]
self.__grdRepo = GradeRepository(self.__studCtrl.repository,
self.__discCtrl.repository, enrolled,
grades)
self.__grdCtrl = GradeController(self.__grdRepo)
self.__undoRepo = UndoRepo()
self.__undoRepo.save(self.__studCtrl, self.__discCtrl, self.__grdCtrl)
unittest.TestCase.setUp(self)
def testSave(self):
studCtrl = EntityController(EntityRepository([]))
discCtrl = EntityController(EntityRepository([]))
repo = GradeRepository(studCtrl.repository, discCtrl.repository, [],
[])
grdCtrl = GradeController(repo)
self.assertEqual(self.__undoRepo.save(studCtrl, discCtrl, grdCtrl),
(studCtrl, discCtrl, grdCtrl),
'Function save doesn\'t work!')
def testRedo(self):
studCtrl = EntityController(EntityRepository([]))
discCtrl = EntityController(EntityRepository([]))
repo = GradeRepository(studCtrl.repository, discCtrl.repository, [],
[])
grdCtrl = GradeController(repo)
self.__undoRepo.save(studCtrl, discCtrl, grdCtrl)
self.__undoRepo.undo()
self.assertEqual(self.__undoRepo.redo(), (studCtrl, discCtrl, grdCtrl),
'Function redo doesn\'t work!')
self.assertRaises(UndoRepositoryException)
def setUp(self):
students = EntityRepository([(1, 'Tania Popescu'), (2, 'Anca Vlad'),
(3, 'Bianca Marculescu')])
disciplines = EntityRepository([(1, 'Matematica'), (2, 'Romana'),
(3, 'Engleza')])
self.grade_repo = GradeRepository(students, disciplines,
[(1, 1), (1, 2), (2, 2), (2, 3),
(3, 1), (3, 3)], [
Grade(1, 1, 9.2),
Grade(1, 1, 6.7),
Grade(1, 2, 10),
Grade(1, 2, 9.4),
Grade(2, 2, 7.8),
Grade(3, 1, 5.6),
Grade(3, 3, 4.5),
Grade(3, 3, 7.7)
])
unittest.TestCase.setUp(self)
def setUp(self):
repo = EntityRepository([(1, 'Tania Popescu'), (2, 'Anca Vlad'),
(3, 'Bianca Marculescu')])
self.entity_con = EntityController(repo)
unittest.TestCase.setUp(self)
def setUp(self):
self.entity_repo = EntityRepository([(1, 'Tania Popescu'),
(2, 'Anca Vlad'),
(3, 'Bianca Marculescu')])
unittest.TestCase.setUp(self)
class TestEntityRepository(unittest.TestCase):
def setUp(self):
self.entity_repo = EntityRepository([(1, 'Tania Popescu'),
(2, 'Anca Vlad'),
(3, 'Bianca Marculescu')])
unittest.TestCase.setUp(self)
def tearDown(self):
unittest.TestCase.tearDown(self)
def testAdd(self):
entity = Entity(4, 'Maria Ioana')
self.entity_repo.add(entity)
self.assertEqual(self.entity_repo.entities[4], 'Maria Ioana',
'Add function doesn\'t work!')
try:
self.entity_repo.add(Entity(1, 'Alex Pop'))
assert False
except RepositoryException:
pass
def testRemoveByID(self):
self.entity_repo.removeByID(3)
self.assertEqual(len(self.entity_repo.entities), 2,
'Function remove doesn\'t work!')
try:
self.entity_repo.removeByID(4)
assert False
except RepositoryException:
pass
def testFindByID(self):
self.assertEqual(self.entity_repo.findByID(1), 'Tania Popescu',
'Function findByID doesn\'t work!')
self.assertEqual(self.entity_repo.findByID(4), None,
'Function findByID doesn\'t work!')
def testSearchByID(self):
self.assertEqual(self.entity_repo.searchByID(1),
[(1, 'Tania Popescu')],
'Function searchByID doesn\'t work!')
self.assertEqual(self.entity_repo.searchByID(4), [],
'Function searchByID doesn\'t work!')
def testSearchByName(self):
self.assertEqual(self.entity_repo.searchByName('anc'),
[(2, 'Anca Vlad'), (3, 'Bianca Marculescu')],
'Function searchByID doesn\'t work!')
self.assertEqual(self.entity_repo.searchByID('cartof'), [],
'Function searchByID doesn\'t work!')
def testUpdate(self):
entity = Entity(1, 'Tania Popescu')
self.entity_repo.update(entity, 'Miruna Matei')
self.assertEqual(self.entity_repo.entities[1], 'Miruna Matei',
'Function update doesn\'t work!')
try:
self.entity_repo.update(Entity(4, 'Dragos Andrei'),
'Otilia Marculescu')
assert False
except RepositoryException:
pass
def testList(self):
self.assertEqual(self.entity_repo.list(), [(1, 'Tania Popescu'),
(2, 'Anca Vlad'),
(3, 'Bianca Marculescu')],
'Function list doesn\'t work!')
def update(self, entity, name):
EntityRepository.update(self, entity, name)
self.writeToFile()
def removeByID(self, entityID):
EntityRepository.removeByID(self, entityID)
self.writeToFile()
def add(self, entity):
EntityRepository.add(self, entity)
self.writeToFile()
'''Scrie fisierele de la inceput'''
# miau1 = EntityFileRepository('student_file')
# miau1.writeToFile(students)
# miau2 = EntityFileRepository('discipline_file')
# miau2.writeToFile(disciplines)
# miau3 = GradeFileRepository("enrolled_file","grade_file")
# miau3.writeToFile(enrolled, grades)
# miau1 = EntityBinaryFileRepository('students_binary')
# miau1.writeToFile(students)
# miau2 = EntityBinaryFileRepository('discipline_binary')
# miau2.writeToFile(disciplines)
# miau3 = GradeBinaryFileRepository("enrolled_binary","grade_binary")
# miau3.writeToFile(enrolled, grades)
''' in-memory '''
studentRepo = EntityRepository(students)
disciplineRepo = EntityRepository(disciplines)
gradeRepo = GradeRepository(studentRepo, disciplineRepo, enrolled, grades)
''' text file '''
# studentRepo = EntityFileRepository("student_file")
# studentRepo.setEntities(studentRepo.readFromFile())
# disciplineRepo = EntityFileRepository("discipline_file")
# disciplineRepo.setEntities(disciplineRepo.readFromFile())
# gradeRepo = GradeFileRepository("enrolled_file","grade_file")
# gradeRepo.students = EntityRepository(studentRepo.readFromFile())
# gradeRepo.disciplines = EntityRepository(disciplineRepo.readFromFile())
# gradeRepo.enrolled[:] = gradeRepo.readEnrolled()
# # print(gradeRepo.readEnrolled())
# gradeRepo.grades = gradeRepo.readGrades()
''' binary file '''
# studentRepo = EntityBinaryFileRepository('students_binary')