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)
