class FriendShipsStore:
def __init__(self):
self.friendShips = FriendShips()
def makeFriends(self, person1, person2):
self.friendShips.makeFriends(person1, person2)
def getFriendsList(self, person):
self.friendShips.getFriendsList(person)
def areFriends(self, person1, person2):
self.friendShips.areFriends(person1, person2)
def addFriend(self, person, friend):
self.friendShips.addFriend(person, friend)
def test_get_friends_list_error(self):
temp = FriendShips()
temp.makeFriends("Kowalski", "Nowak")
with self.assertRaises(KeyError):
temp.getFriendsList("Kwiatkowski")
def test_get_friends_list(self):
temp = FriendShips()
temp.makeFriends("Kowalski", "Nowak")
self.assertEqual(temp.getFriendsList("Kowalski"), ["Nowak"])