def test_updateElement_with1emptyField_elementUpdated1fieldUnchanged(self): repository = Repository() movie = Movie(1, 'Frozen', 'Ice queen with magical powers', 'Animation') repository.add(movie) repository.update(movie, Movie(1, '', 'a', 'b')) expected_result = Movie(1, 'Frozen', 'a', 'b') self.assertTrue(repository.get_all()[0].identical(expected_result))
def test_updateElement_withAllFields_elementUpdated(self): repository = Repository() movie = Movie(1, 'Frozen', 'Ice queen with magical powers', 'Animation') repository.add(movie) movieUpdate = Movie(1, '-', '-', '-') repository.update(movie, movieUpdate) self.assertTrue(repository.get_all()[0].identical(movieUpdate))
def _saveFile(self): ''' 1. Open text file for writing 'w' 2. for each element in the repository: a. transform it into one-line string b. write it to the file 3. close file ''' filepath = self._fileName file = open(filepath, 'wb') for element in Repository.get_all(self): pickle.dump(element, file) file.close()
def all_students(self, discipline_id): """ This function returns a repository that contains all the students that have a grade at the given discipline :param discipline_id: The id of the discipline :return: A Repository containing the students sorted by student name """ full = self.__repository.get_all() new = Repository() for i in full: if i.discipline_id == discipline_id: ret = new.get_all() s = self.__studentRepository.find(i.student_id) if s not in ret: new.store(s) return new.sort_()
def get(self, training=None): training = training or Training() parts = Participants() org = Organization() sched = Schedules() when = When() costs = Costs() list_catg = [(item.value, item.name) for item in CategoryType] with Session(database='training') as session: repo = Repository(session) operator = repo.get_all(coll_name='organization', conditions={}, projection={'org_identity': 1, 'org_name': 1, '_id': 0}) list_op = [(item['org_identity'], item['org_name']) for item in operator] return self.render_response("training/add.html", training=training, list_catg=list_catg, list_op=list_op, parts=parts, org=org, sched=sched, when=when, costs=costs)
def test_add_newMovie_movieIsAdded(self): movie = Movie(1, 'a', 'a', 'a') repo = Repository() repo.add(movie) self.assertEqual(repo.size(), 1) self.assertEqual(repo.get_all()[repo.size() - 1], movie)
def get_all(self): self.__read_all_from_file() return Repository.get_all(self)