update tests

spyoungtech · Aug 25, 2023 · be14d1c · be14d1c
1 parent 8327cb0
commit be14d1c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/tests/_async/test_extensions.py b/tests/_async/test_extensions.py
@@ -13,9 +13,8 @@
 
 ext_text = '''\
 AHKDoSomething(ByRef command) {
-    global STRINGRESPONSEMESSAGE
     arg := command[2]
-    return FormatResponse(STRINGRESPONSEMESSAGE, Format("test{}", arg))
+    return FormatResponse("ahk.message.StringResponseMessage", Format("test{}", arg))
 }
 '''
 
@@ -24,7 +23,7 @@
 
 @async_extension.register
 async def do_something(ahk, arg: str) -> str:
-    res = await ahk._transport.function_call('AHKDoSomething', [arg])
+    res = await ahk.function_call('AHKDoSomething', [arg])
     return res
 
 

diff --git a/tests/_sync/test_extensions.py b/tests/_sync/test_extensions.py
@@ -12,9 +12,8 @@
 
 ext_text = '''\
 AHKDoSomething(ByRef command) {
-    global STRINGRESPONSEMESSAGE
     arg := command[2]
-    return FormatResponse(STRINGRESPONSEMESSAGE, Format("test{}", arg))
+    return FormatResponse("ahk.message.StringResponseMessage", Format("test{}", arg))
 }
 '''
 
@@ -23,7 +22,7 @@
 
 @async_extension.register
 def do_something(ahk, arg: str) -> str:
-    res = ahk._transport.function_call('AHKDoSomething', [arg])
+    res = ahk.function_call('AHKDoSomething', [arg])
     return res