fix extension tests

spyoungtech · Aug 30, 2023 · aa6afec · aa6afec
1 parent 3eba8b8
commit aa6afec
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 10 deletions.
diff --git a/tests/_async/test_extensions.py b/tests/_async/test_extensions.py
@@ -1,4 +1,6 @@
 import asyncio
+import random
+import string
 import time
 import unittest
 
@@ -11,19 +13,22 @@
 
 sleep = time.sleep
 
-ext_text = '''\
-AHKDoSomething(ByRef command) {
+function_name = 'AHKDoSomething'
+function_name = 'AAHKDoSomething'  # unasync: remove
+
+ext_text = f'''\
+{function_name}(ByRef command) {{
     arg := command[2]
-    return FormatResponse("ahk.message.StringResponseMessage", Format("test{}", arg))
-}
+    return FormatResponse("ahk.message.StringResponseMessage", Format("test{{}}", arg))
+}}
 '''
 
 async_extension = Extension(script_text=ext_text)
 
 
 @async_extension.register
 async def do_something(ahk, arg: str) -> str:
-    res = await ahk.function_call('AHKDoSomething', [arg])
+    res = await ahk.function_call(function_name, [arg])
     return res
 
 

diff --git a/tests/_sync/test_extensions.py b/tests/_sync/test_extensions.py
@@ -1,4 +1,6 @@
 import asyncio
+import random
+import string
 import time
 import unittest
 
@@ -10,19 +12,21 @@
 
 sleep = time.sleep
 
-ext_text = '''\
-AHKDoSomething(ByRef command) {
+function_name = 'AHKDoSomething'
+
+ext_text = f'''\
+{function_name}(ByRef command) {{
     arg := command[2]
-    return FormatResponse("ahk.message.StringResponseMessage", Format("test{}", arg))
-}
+    return FormatResponse("ahk.message.StringResponseMessage", Format("test{{}}", arg))
+}}
 '''
 
 async_extension = Extension(script_text=ext_text)
 
 
 @async_extension.register
 def do_something(ahk, arg: str) -> str:
-    res = ahk.function_call('AHKDoSomething', [arg])
+    res = ahk.function_call(function_name, [arg])
     return res