FilesExpand file tree

 main

/

test_client.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

33 lines (27 loc) · 775 Bytes

 main

/

test_client.py

Copy path

Top

File metadata and controls

• Code
• Blame

33 lines (27 loc) · 775 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import os

from unittest import mock

import httpx

import pytest

import respx

@pytest.mark.asyncio

async def test_authorization_when_setting_environ_after_import():

import replicate

router = respx.Router()

router.route(

method="GET",

url="https://api.replicate.com/",

headers={"Authorization": "Token test-set-after-import"},

).mock(

return_value=httpx.Response(

200,

json={},

)

)

token = "test-set-after-import" # noqa: S105

with mock.patch.dict(

os.environ,

{"REPLICATE_API_TOKEN": token},

):

client = replicate.Client(transport=httpx.MockTransport(router.handler))

resp = client._request("GET", "/")

assert resp.status_code == 200