def test_vk_manager_create_attachment_no_type(self):
test_attachment_raw = {
"type": "photo",
"photo": {
"id":
13,
"album_id":
13,
"owner_id":
1,
"sizes": [{
"type": "s",
"url": "url2",
"width": 0,
"height": 0
}, {
"type": "m",
"url": "url",
"width": 0,
"height": 0
}],
"text":
"",
"date":
0,
"post_id":
0,
"likes": {
"user_likes": 0,
"count": 577904
},
"comments": {
"count": 421
},
"can_comment":
1,
"can_repost":
1,
"tags": {
"count": 0
}
}
}
attachment = VKManager.create_attachment(test_attachment_raw)
self.assertEqual(attachment.id, 13)
self.assertEqual(attachment.type, "photo")
self.assertEqual(attachment.owner_id, 1)
self.assertEqual(attachment.access_key, None)
self.assertEqual(attachment.link, "url")
self.assertEqual(attachment.raw_attachment, test_attachment_raw)
def test_vk_manager_create_attachment_doc_no_link(self):
test_attachment_raw = {
"id": 13,
"owner_id": 1,
"title": "rrrrr.png",
"size": 119900,
"ext": "png",
"date": 0,
"type": 4,
}
attachment = VKManager.create_attachment(test_attachment_raw, "doc")
self.assertEqual(attachment.id, 13)
self.assertEqual(attachment.type, "doc")
self.assertEqual(attachment.owner_id, 1)
self.assertEqual(attachment.access_key, None)
self.assertEqual(attachment.link, None)
self.assertEqual(attachment.raw_attachment, test_attachment_raw)
