def test_get_unowned_files(self): user = factory.make_User() files = [ factory.make_FileStorage(owner=None), factory.make_FileStorage(owner=user), factory.make_FileStorage(owner=None), ] self.assertSetEqual({files[0], files[2]}, set(get_unowned_files()))
def test_get_unowned_files_no_files(self): self.assertEqual([], list(get_unowned_files()))