def test_add_service(self):
"""Tests that adding services into database occurs correctly"""
database = Database()
count = 0
self.assertEqual(len(database.services), count)
tags = set()
self.assertEqual(database.get_tags(), tags)
for key, value in self.services.items():
database.add_service(key, value[0])
count += 1
self.assertEqual(len(database.services), count)
tags = tags | set(value[0].split(','))
self.assertEqual(database.get_tags(), tags)
self.assertEqual(count, len(self.services))
def tast_remove_service(self):
"""Tests that removing services from database is done correctly"""
database = Database()
for key, value in self.services.items():
database.add_service(key, value[0])
count = len(self.services)
for key in self.services.keys():
database.remove_service(key)
count -= 1
self.assertEqual(len(database.services), count)
self.assertEqual(len(database.services), 0)
self.assertEqual(database.get_tags(), set())
def test_add_existing_service(self):
"""Tests that adding services into database fails if service is already present"""
database = Database()
for key, value in self.services.items():
database.add_service(key, value[0])
self.assertRaises(ValueError, database.add_service, key, value[0])
