2012-10-22 313 views
0

我有2个模型,专辑和图片和一个函数album_reorder它接收图片ID列表并相应地改变权重。这工作得很好,但是我想写的功能测试,如:奇怪的字典分配问题

Class ReposTests(unittest.TestCase): 
    def test_album_reorder__fail__not_owned(self): 

     # Create a test user, album and pictures. 
     user = self.create_test_user() 
     # create_default_album() generates an Album along with 6 default Picture records.  
     create_default_album(user) 
     album = get_first_album(user.profile) 
     pictures = Picture.objects.filter(album=album) 

     # Get all picture IDs into a list and record each pictures existing weight. 
     picture_ids = [] 
     picture_weights = {} 
     for picture in pictures: 
      picture_ids.append(picture.id) 
      print "%s:%s:%s" % (album.id, picture.id, picture.weight) 
      picture_weights[picture.id] = picture.weight 

     # Create another test user, album and pictures. 
     user2 = self.create_test_user() 
     create_default_album(user2) 
     album2 = get_first_album(user2.profile) 
     pictures2 = Picture.objects.filter(album=album2) 

     # Add a Picture ID from a different user to test the restriction that an album cannot be re-ordered if any pictures do not belong to the specified user. 
     picture_ids.append(pictures2[0].id) 
     print "%s:%s:%s" % (album2.id, pictures2[0].id, pictures2[0].weight) 
     picture_weights[pictures[2].id] = pictures2[0].weight 

     # Shuffle all picture IDs 
     random = Random() 
     while pictures[0].id == picture_ids[0]: 
      random.shuffle(picture_ids) 

     # Reorder picture IDs according to shuffled ID list, however this should fail due to the ownership check. 
     album_reorder(user, picture_ids) 
     print picture_ids 
     print picture_weights 

     # Check each 
     for picture_id in picture_ids: 
      self.assertEqual(picture_weights[picture_id], Picture.objects.get(id=picture_id).weight) 

现在的问题似乎是在picture_weights字典的构造方式。由于以下两行输出:

# Loop output: (album.id, picture.id, picture.weight) 
# 1346:5699:0 
# 1346:5700:1 
# 1346:5701:2 
# 1346:5702:3 
# 1346:5703:4 
# 1346:5704:5 
# 1347:5705:0 

print picture_ids 
# Outputs: [5700L, 5703L, 5702L, 5699L, 5704L, 5705L, 5701L] 

print picture_weights 
# Outputs: {5699L: 0L, 5700L: 1L, 5701L: 0L, 5702L: 3L, 5703L: 4L, 5704L: 5L} 

注意picture_ids是7层的元件长,正如所料,在那里如picture_weights只有6的元件,并且权重不匹配的那些从环路输出。 picture_weights中缺少第二个用户图片5705,但picture_weights中的一个已被指定不正确的权重0L

我没有dict的经验,但是我只是试图运用我的PHP关联数组的知识,所以我假设我在那里有一个缺口。感谢您的任何建议!

+2

为什么你行'picture_weights [图片[2] .ID = pictures2 [0] .weight',而不是'picture_weights [pictures2 [0] .ID ] =图片2 [0] .weight'? –

回答

3

名单和字典发散这些行:

# Add a Picture ID from a different user to test the restriction that an album cannot be re-ordered if any pictures do not belong to the specified user. 
    picture_ids.append(pictures2[0].id) 
    print "%s:%s:%s" % (album2.id, pictures2[0].id, pictures2[0].weight) 
    picture_weights[pictures[2].id] = pictures2[0].weight 

注意如何使用pictures[2].id,这是5701L(这就是为什么在字典中特定键的体重变化)。你可能意味着该行是:

picture_weights[pictures2[0].id] = pictures2[0].weight 
+0

已修复!对不起*腮红* – DanH

+0

不用担心,它发生在我们所有人身上。 –