1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
from typing import ClassVar, Tuple
from urllib.parse import urljoin
from stac_pydantic.api.utils.link_factory import BaseLinks, CollectionLinks, ItemLinks
from stac_pydantic.links import Link
from stac_pydantic.shared import MimeTypes
def test_collection_links() -> None:
links = CollectionLinks(
collection_id="collection", base_url="http://stac.com"
).create_links()
for link in links.link_iterator():
assert isinstance(link, Link)
assert link.rel in CollectionLinks._link_members
def test_item_links() -> None:
links = ItemLinks(
collection_id="collection", item_id="item", base_url="http://stac.com"
).create_links()
for link in links.link_iterator():
assert isinstance(link, Link)
assert link.rel in ItemLinks._link_members
def test_custom_links() -> None:
class CustomLinks(BaseLinks):
_link_members: ClassVar[Tuple[str]] = ("another_link",)
def another_link(self) -> Link:
return Link(
rel="another-link",
type=MimeTypes.json,
href=urljoin(self.base_url, "/another-link"),
)
links = CustomLinks(base_url="http://stac.com").create_links()
assert len(links) == 1
assert links[0].rel == "another-link"
assert links[0].type == "application/json"
assert links[0].href == "http://stac.com/another-link"
|
