def test_get_all(self): io = IO('test.json') io.set([]) repo = EventRepository('test.json') content = repo.get_all() self.assertEqual(content, []) event1 = Event('14/01/2015', '14:00', 'Colectiv') event2 = Event('14/01/2015', '14:00', 'Colectiv') repo.insert(event1) repo.insert(event2) content = repo.get_all() self.assertIs(type(content), list) self.assertEqual(len(content), 2) stored_event1 = content[0] stored_event2 = content[1] self.assertEqual(stored_event1.get_date(), '14/01/2015') self.assertEqual(stored_event1.get_time(), '14:00') self.assertEqual(stored_event1.get_description(), 'Colectiv') self.assertEqual(stored_event2.get_date(), '14/01/2015') self.assertEqual(stored_event2.get_time(), '14:00') self.assertEqual(stored_event2.get_description(), 'Colectiv')
def insert(self, person_event): operation = IO(self.__defaultfile) try: person_events = operation.get() except ValueError: person_events = [] person_events.append(person_event.get_serialization()) operation.set(person_events)
def test_set(self): file = open(IO.filepath('test.json'), 'w') file.close() io = IO('test.json') io.set('{"data":[{"a":1}]}') self.assertEqual(io.get(), '{"data":[{"a":1}]}') io.set(1) self.assertEqual(io.get(), 1)
def test_delete(self): io = IO('test.json') io.set([]) Person.set_class_id(10) person = Person('Vasile', 'Pop', 'Str. Calea Floresti, nr. 24') repo = PersonRepository('test.json') repo.insert(person) repo.delete(person) with self.assertRaisesRegex(ValueError, 'Person not found!'): person = repo.get(10)
def delete(self, person): ''' Deletes a person from disk storage :param deleted_event: :return: ''' operation = IO(self.__defaultfile) people = operation.get() for per in people: if per['id'] == person.get_id(): people.remove(per) operation.set(people)
def delete(self, deleted_event): ''' Deletes an event from disk storage :param deleted_event: :return: ''' operation = IO(self.__datafile) events = operation.get() for event in events: if event['id'] == deleted_event.get_id(): events.remove(event) break operation.set(events)
def update(self, updated_event): ''' Updates an event into disk storage :param updated_event: :return: ''' operation = IO(self.__datafile) events = operation.get() for event in events: if event['id'] == updated_event.get_id(): events[events.index(event)] = updated_event.get_serialization() break operation.set(events)
def test_insert(self): repo = PersonEventRepository() person_event = PersonEvent(25, 61) repo.insert(person_event) person_events = repo.get_by_id(25, 61) self.assertEqual(len(person_events), 1) # Remove last appended object io = IO('person_event.json') updated_person_events = io.get() updated_person_events.pop() io.set(updated_person_events)
def update(self, person): ''' Updates a person into disk storage :param updated_event: :return: ''' operation = IO(self.__defaultfile) people = operation.get() for per in people: if per['id'] == person.get_id(): people[people.index(per)] = person.get_serialization() operation.set(people)
def insert(self, person): ''' Add a person object into disk storage :param event: :return: None ''' operation = IO(self.__defaultfile) try: people = operation.get() except ValueError: people = [] people.append(person.get_serialization()) operation.set(people)
def test_insert(self): io = IO('test.json') io.set([]) event = Event('14/01/2015', '14:00', 'Colectiv') EventRepository('test.json').insert(event) content = io.get() self.assertIs(type(content), list) stored_event = content[0] self.assertEqual(stored_event['date'], '14/01/2015') self.assertEqual(stored_event['time'], '14:00') self.assertEqual(stored_event['description'], 'Colectiv')
def test_get_max_id(self): Event.set_class_id(0) io = IO('test.json') io.set([]) repo = EventRepository('test.json') self.assertEqual(repo.get_max_id(), 0) content = repo.get_all() self.assertEqual(content, []) event1 = Event('14/01/2015', '14:00', 'Colectiv') event2 = Event('14/01/2015', '14:00', 'Colectiv') repo.insert(event1) repo.insert(event2) content = repo.get_all() self.assertEqual(repo.get_max_id(), 2)
def insert(self, event): ''' Add an event object into disk storage :param event: :return: None ''' operation = IO(self.__datafile) try: data = operation.get() except ValueError: data = [] data.append(event.get_serialization()) operation.set(data)
def test_get(self): io = IO('test.json') io.set([]) Person.set_class_id(10) person = Person('Vasile', 'Pop', 'Str. Calea Floresti, nr. 24') repo = PersonRepository('test.json') repo.insert(person) person = repo.get(10) self.assertEqual(person.get_id(), 10) self.assertEqual(person.get_name(), 'Vasile Pop') self.assertEqual(person.get_address(), 'Str. Calea Floresti, nr. 24') with self.assertRaisesRegex(ValueError, 'Person not found!'): person = repo.get(0)
def test_insert(self): io = IO('test.json') io.set([]) Person.set_class_id(0) person = Person('Vasile', 'Pop', 'Str. Calea Floresti, nr. 24') repo = PersonRepository('test.json') repo.insert(person) people = io.get() person = people[0] self.assertEqual(len(people), 1) self.assertEqual(person['id'], 0) self.assertEqual(person['first_name'], 'Vasile') self.assertEqual(person['last_name'], 'Pop') self.assertEqual(person['address'], 'Str. Calea Floresti, nr. 24')
def test_delete(self): Event.set_class_id(0) io = IO('test.json') io.set([]) repo = EventRepository('test.json') event1 = Event('14/01/2015', '14:00', 'Colectiv') event2 = Event('14/01/2015', '14:00', 'Colectiv') repo.insert(event1) repo.insert(event2) event = repo.get(0) self.assertIs(type(event), Event) repo.delete(event) with self.assertRaisesRegex(ValueError, 'Event not found!'): updated_event = repo.get(0)
def test_update(self): io = IO('test.json') io.set([]) Person.set_class_id(10) person = Person('Vasile', 'Pop', 'Str. Calea Floresti, nr. 24') repo = PersonRepository('test.json') repo.insert(person) person = repo.get(10) person.update('Dan', 'Popescu', 'Calea Dorobantilor') repo.update(person) updated_person = repo.get(10) self.assertEqual(person.get_id(), 10) self.assertEqual(person.get_name(), 'Dan Popescu') self.assertEqual(person.get_address(), 'Calea Dorobantilor')
def test_get_all(self): io = IO('test.json') io.set([]) Person.set_class_id(0) person = Person('Vasile', 'Pop', 'Str. Calea Floresti, nr. 24') repo = PersonRepository('test.json') repo.insert(person) people = repo.get_all() self.assertEqual(len(people), 1) person = people[0] self.assertEqual(person.get_id(), 0) self.assertEqual(person.get_name(), 'Vasile Pop') self.assertEqual(person.get_address(), 'Str. Calea Floresti, nr. 24')
def test_get(self): Event.set_class_id(0) io = IO('test.json') io.set([]) repo = EventRepository('test.json') event1 = Event('14/01/2015', '14:00', 'Colectiv') event2 = Event('14/01/2015', '14:00', 'Colectiv') repo.insert(event1) repo.insert(event2) content = repo.get(0) self.assertIs(type(content), Event) self.assertEqual(content.get_date(), '14/01/2015') self.assertEqual(content.get_time(), '14:00') self.assertEqual(content.get_description(), 'Colectiv') with self.assertRaisesRegex(ValueError, 'Event not found!'): content = repo.get(10)
def test_update(self): Event.set_class_id(0) io = IO('test.json') io.set([]) repo = EventRepository('test.json') event1 = Event('14/01/2015', '14:00', 'Colectiv') event2 = Event('14/01/2015', '14:00', 'Colectiv') repo.insert(event1) repo.insert(event2) event = repo.get(0) self.assertIs(type(event), Event) event.update('15/01/2015', '15:00', 'Untold festival') repo.update(event) updated_event = repo.get(0) self.assertEqual(updated_event.get_date(), '15/01/2015') self.assertEqual(updated_event.get_time(), '15:00') self.assertEqual(updated_event.get_description(), 'Untold festival')
def test_initialization(self): io = IO('test.json') io.set([]) repository = PersonRepository('test.json') self.assertIsInstance(repository, PersonRepository)