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)