Skip to content

Commit 77eed75

Browse files
committed
tests: fix typing imports still looking for old symbols.
Signed-off-by: Romain Bezut <morian@xdec.net>
1 parent 9174635 commit 77eed75

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

tests/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919
Set as AbstractSet,
2020
)
2121

22+
from aiostem.command import Command
2223
from aiostem.controller import EventCallbackType
23-
from aiostem.utils import (
24-
Command,
25-
EventWord,
24+
from aiostem.event import EventWord
25+
from aiostem.reply import (
2626
ReplyAuthChallenge,
2727
ReplyProtocolInfo,
2828
ReplySetEvents,

tests/test_command.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
from aiostem.structures import (
5252
CircuitPurpose,
5353
DescriptorPurpose,
54+
HiddenServiceAddressV2,
5455
HsDescAuthCookie,
5556
HsDescAuthTypeInt,
5657
HsDescClientAuthV2,
@@ -294,11 +295,11 @@ def test_drop_guards(self):
294295
assert cmd.serialize() == 'DROPGUARDS\r\n'
295296

296297
def test_hs_fetch(self):
297-
cmd = CommandHsFetch(address='facebookcorewwwi')
298+
cmd = CommandHsFetch(address=HiddenServiceAddressV2('facebookcorewwwi'))
298299
assert cmd.serialize() == 'HSFETCH facebookcorewwwi\r\n'
299300

300301
def test_hs_fetch_with_servers(self):
301-
address = 'facebookcorewwwi'
302+
address = HiddenServiceAddressV2('facebookcorewwwi')
302303
server1 = '$b34a4ac3892e41c58709d9c51b3648620a7d5bfe~Test1'
303304
server2 = '$7b70bf914770f022e71a26cbf3d9519dc89f2a9a~Test2'
304305
cmd = CommandHsFetch(

0 commit comments

Comments
 (0)