def test_are_friend(self):
objectFriend = FriendShips()
objectFriend.dict = {"Przemek": ["Ala", "Basia", "Piotrek"]}
objectFriend.areFriends = MagicMock()
objectFriend.areFriends.return_value = "Basia is friend Przemek"
objectStorage = FriendStorage()
objectStorage.storage = objectFriend
result = objectStorage.areFriends("Basia", "Przemek")
self.assertEqual(result, "Basia is friend Przemek")
def test_get_friends_list(self):
objectFriend = FriendShips()
objectFriend.dict = {"Przemek": ["Ala", "Basia", "Piotrek"]}
objectFriend.getFriendsList = MagicMock()
objectFriend.getFriendsList.return_value = ["Ala", "Basia", "Piotrek"]
objectStorage = FriendStorage()
objectStorage.storage = objectFriend
result = objectStorage.getFriendsList("Przemek")
self.assertEqual(result, ["Ala", "Basia", "Piotrek"])
def test_get_friends_list_lack_person(self):
objectFriend = FriendShips()
objectFriend.dict = {"Przemek": ["Ala", "Basia", "Piotrek"]}
objectFriend.areFriends = MagicMock()
objectFriend.areFriends.side_effect = Exception(
"This person not exist")
objectStorage = FriendStorage()
objectStorage.storage = objectFriend
result = objectStorage.getFriendsList
self.assertRaisesRegex(Exception, "This person not exist", result,
"Adam")
def test_make_friends_add_friend(self):
objectFriend = FriendShips()
objectFriend.dict = {"Przemek": ["Ala"]}
objectFriend.makeFriends = MagicMock()
objectFriend.makeFriends.return_value = {
"Przemek": ["Ala", "Bartek"],
"Bartek": ["Przemek"]
}
objectStorage = FriendStorage()
objectStorage.storage = objectFriend
result = objectStorage.makeFriends("Przemek", "Bartek")
self.assertEqual(result, {
"Przemek": ["Ala", "Bartek"],
"Bartek": ["Przemek"]
})
objectStorage.storage.makeFriends.assert_called_with(
"Przemek", "Bartek")