Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ classifiers = [
"Intended Audience :: Developers",
"Topic :: Software Development :: Libraries",
]
dependencies = ["httpx>=0.24", "pydantic>=2"]
dependencies = ["httpx2>=2", "pydantic>=2"]

[project.urls]
Homepage = "https://github.com/crdbrd/python-brreg"
Expand Down Expand Up @@ -41,10 +41,10 @@ pyright = ["basedpyright==1.39.6"]
ruff = ["ruff==0.15.14"]
tests = [
"coverage>=7.6.12",
"pytest>=8.3.4",
"httpx2-pytest>=1",
"pytest-cov>=6.0.0",
"pytest-httpx>=0.35.0",
"pytest-watcher>=0.4.3",
"pytest>=8.3.4",
]
ty = ["ty==0.0.39"]
zizmor = ["zizmor==1.25.2"]
Expand Down
13 changes: 7 additions & 6 deletions src/brreg/enhetsregisteret/_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from types import TracebackType
from typing import Any

import httpx
import httpx2

import brreg
from brreg import BrregError, BrregRestError
Expand Down Expand Up @@ -38,7 +38,7 @@ class Client:
client.close()
"""

_client: httpx.Client
_client: httpx2.Client

def __init__(self) -> None:
self.open()
Expand All @@ -59,11 +59,12 @@ def open(self) -> None:

This is called automatically when the client is created.
"""
self._client = httpx.Client(
self._client = httpx2.Client(
base_url="https://data.brreg.no/enhetsregisteret/api",
headers={
"user-agent": (
f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
f"python-brreg/{brreg.__version__} "
f"python-httpx2/{httpx2.__version__}"
),
},
)
Expand Down Expand Up @@ -188,8 +189,8 @@ def search_underenhet(
def error_handler() -> Generator[None, Any, None]:
try:
yield
except httpx.HTTPError as exc:
response: httpx.Response | None = getattr(exc, "response", None)
except httpx2.HTTPError as exc:
response: httpx2.Response | None = getattr(exc, "response", None)
raise BrregRestError(
str(exc),
method=(exc.request.method if exc.request else None),
Expand Down
8 changes: 4 additions & 4 deletions tests/enhetsregisteret/test_get_enhet.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from datetime import date
from pathlib import Path

import httpx
import httpx2
import pytest
from pytest_httpx import HTTPXMock
from pytest_httpx2 import HTTPXMock

import brreg
from brreg import BrregError, BrregRestError, enhetsregisteret
Expand All @@ -26,7 +26,7 @@ def test_get_enhet(httpx_mock: HTTPXMock) -> None:
assert len(requests) == 1
assert (
requests[0].headers["user-agent"]
== f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
== f"python-brreg/{brreg.__version__} python-httpx2/{httpx2.__version__}"
)

assert org is not None
Expand Down Expand Up @@ -138,7 +138,7 @@ def test_get_enhet_when_bad_request(httpx_mock: HTTPXMock) -> None:

def test_get_enhet_when_http_timeout(httpx_mock: HTTPXMock) -> None:
httpx_mock.add_exception( # pyright: ignore[reportUnknownMemberType]
httpx.ConnectTimeout("Connection refused"),
httpx2.ConnectTimeout("Connection refused"),
)

with pytest.raises(BrregRestError) as exc_info:
Expand Down
6 changes: 3 additions & 3 deletions tests/enhetsregisteret/test_get_roller.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from datetime import date
from pathlib import Path

import httpx
import httpx2
import pytest
from pytest_httpx import HTTPXMock
from pytest_httpx2 import HTTPXMock

import brreg
from brreg import enhetsregisteret
Expand All @@ -26,7 +26,7 @@ def test_get_roller_with_person(httpx_mock: HTTPXMock) -> None:
assert len(requests) == 1
assert (
requests[0].headers["user-agent"]
== f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
== f"python-brreg/{brreg.__version__} python-httpx2/{httpx2.__version__}"
)

assert rollegrupper
Expand Down
6 changes: 3 additions & 3 deletions tests/enhetsregisteret/test_get_underenhet.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from datetime import date
from pathlib import Path

import httpx
import httpx2
import pytest
from pytest_httpx import HTTPXMock
from pytest_httpx2 import HTTPXMock

import brreg
from brreg import enhetsregisteret
Expand All @@ -26,7 +26,7 @@ def test_get_underenhet(httpx_mock: HTTPXMock) -> None:
assert len(requests) == 1
assert (
requests[0].headers["user-agent"]
== f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
== f"python-brreg/{brreg.__version__} python-httpx2/{httpx2.__version__}"
)

assert org is not None
Expand Down
6 changes: 3 additions & 3 deletions tests/enhetsregisteret/test_search_enhet.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from datetime import date
from pathlib import Path

import httpx
from pytest_httpx import HTTPXMock
import httpx2
from pytest_httpx2 import HTTPXMock

import brreg
from brreg import enhetsregisteret
Expand Down Expand Up @@ -59,7 +59,7 @@ def test_search_enhet(httpx_mock: HTTPXMock) -> None:
assert len(requests) == 1
assert (
requests[0].headers["user-agent"]
== f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
== f"python-brreg/{brreg.__version__} python-httpx2/{httpx2.__version__}"
)

page = next(cursor.pages)
Expand Down
6 changes: 3 additions & 3 deletions tests/enhetsregisteret/test_search_underenhet.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from datetime import date
from pathlib import Path

import httpx
from pytest_httpx import HTTPXMock
import httpx2
from pytest_httpx2 import HTTPXMock

import brreg
from brreg import enhetsregisteret
Expand Down Expand Up @@ -54,7 +54,7 @@ def test_search_underenhet(httpx_mock: HTTPXMock) -> None:
assert len(requests) == 1
assert (
requests[0].headers["user-agent"]
== f"python-brreg/{brreg.__version__} python-httpx/{httpx.__version__}"
== f"python-brreg/{brreg.__version__} python-httpx2/{httpx2.__version__}"
)

page = next(cursor.pages)
Expand Down
Loading