mirror of https://github.com/scrapy/scrapy.git
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from scrapy.link import Link
|
|
|
|
|
|
class TestLink:
|
|
def _assert_same_links(self, link1: Link, link2: Link) -> None:
|
|
assert link1 == link2
|
|
assert hash(link1) == hash(link2)
|
|
|
|
def _assert_different_links(self, link1: Link, link2: Link) -> None:
|
|
assert link1 != link2
|
|
assert hash(link1) != hash(link2)
|
|
|
|
def test_eq_and_hash(self) -> None:
|
|
l1 = Link("http://www.example.com")
|
|
l2 = Link("http://www.example.com/other")
|
|
l3 = Link("http://www.example.com")
|
|
|
|
self._assert_same_links(l1, l1)
|
|
self._assert_different_links(l1, l2)
|
|
self._assert_same_links(l1, l3)
|
|
|
|
l4 = Link("http://www.example.com", text="test")
|
|
l5 = Link("http://www.example.com", text="test2")
|
|
l6 = Link("http://www.example.com", text="test")
|
|
|
|
self._assert_same_links(l4, l4)
|
|
self._assert_different_links(l4, l5)
|
|
self._assert_same_links(l4, l6)
|
|
|
|
l7 = Link(
|
|
"http://www.example.com", text="test", fragment="something", nofollow=False
|
|
)
|
|
l8 = Link(
|
|
"http://www.example.com", text="test", fragment="something", nofollow=False
|
|
)
|
|
l9 = Link(
|
|
"http://www.example.com", text="test", fragment="something", nofollow=True
|
|
)
|
|
l10 = Link(
|
|
"http://www.example.com", text="test", fragment="other", nofollow=False
|
|
)
|
|
self._assert_same_links(l7, l8)
|
|
self._assert_different_links(l7, l9)
|
|
self._assert_different_links(l7, l10)
|
|
|
|
def test_repr(self) -> None:
|
|
l1 = Link(
|
|
"http://www.example.com", text="test", fragment="something", nofollow=True
|
|
)
|
|
l2 = eval(repr(l1))
|
|
self._assert_same_links(l1, l2)
|
|
|
|
def test_bytes_url(self) -> None:
|
|
with pytest.raises(TypeError):
|
|
Link(b"http://www.example.com/\xc2\xa3") # type: ignore[arg-type]
|
|
|
|
def test_eq_non_link(self) -> None:
|
|
url = "http://example.com"
|
|
assert Link(url) != url
|