def test_file_saved(self):
user = factory.make_user()
file = factory.make_file_storage(owner=None)
save = self.patch(file, "save")
give_file_to_user(file, user)
save.assert_called_once()
def test_give_unowned_file(self):
user = factory.make_user()
file = factory.make_file_storage(owner=None)
give_file_to_user(file, user)
self.assertEqual(user, file.owner)
def test_give_owned_file(self):
user1 = factory.make_user()
user2 = factory.make_user()
file = factory.make_file_storage(owner=user1)
give_file_to_user(file, user2)
self.assertEqual(user2, file.owner)
def test_file_saved(self):
user = factory.make_User()
file = factory.make_FileStorage(owner=None)
save = self.patch(file, "save")
give_file_to_user(file, user)
self.assertThat(save, MockCalledOnceWith())
