Skip to content
GitLab
Switch branch/tag
Find file BlameHistoryPermalink
test_link_reader.py 2.21 KiB
1 
from unittest import TestCase
2
3 
from pymemri import PodClient
4
5 
from link_reader.data.schema import LinkPreview, Photo, Edge, Link
6 
from link_reader.plugin import LinkReader
7 
from tests.loader import load_mock_data
8 
from linkpreview import link_preview
9 
import requests
10
11 
class LinkReaderTest(TestCase):
12 
    client = PodClient()
13 
    plugin = LinkReader(client=client)
14
15 
    def test_link_reader(self):
16 
        link_collectors = load_mock_data("link_collector")
17 
        for link in link_collectors:
18 
            edges = []
19 
            files = []
20 
            creates = []
21 
            link_item = Link(link=link["link"])
22 
            self.plugin.client.create(link_item)
23 
            preview = link_preview(link["link"])
24 
            link_reader_item = LinkPreview(
25 
                title=preview.title,
26 
                description=preview.description,
27 
                force_title=preview.force_title,
28 
                site_name=preview.site_name,
29 
            )
30 
            creates.append(link_reader_item)
31 
            try:
32 
                image = requests.get(preview.image)
33 
                image_byte = image.content
34 
                img = Photo.from_bytes(image_byte)
35 
                creates.extend([img, img.file[0]])
36 
                edges.extend(
37 
                    [
38 
                        Edge(img, img.file[0], "file"),
39 
                        Edge(link_reader_item, img, "image"),
40 
                    ]
41 
                )
42 
                files.append(img.data)
43 
            except:
44 
                pass
45
46 
            try:
47 
                absolute_image = requests.get(preview.absolute_image)
48 
                absolute_image_byte = absolute_image.content
49 
                absolute_img = Photo.from_bytes(absolute_image_byte)
50 
                creates.extend([absolute_img, absolute_img.file[0]])
51 
                edges.extend(
52 
                    [
53 
                        Edge(absolute_img, absolute_img.file[0], "file"),
54 
                        Edge(link_reader_item, absolute_img, "absolute_image"),
55 
                    ]
56 
                )
57 
                files.append(absolute_img.data)
58 
            except:
59 
                pass
60
61 
            self.client.bulk_action(create_items=creates, create_edges=edges)
62 
            for f in files:
63 
                self.client.upload_file(f)
64
65