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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
import threading
import time
import os
import sys
import logging
import traceback
import pytest
from multiprocessing import Process, Queue
logger = logging.getLogger('pywebview')
def run_test(webview, main_func, thread_func=None, param=None, no_destroy=False, destroy_delay=0):
__tracebackhide__ = True
queue = Queue()
try:
time.sleep(2)
_create_window(webview, main_func, thread_func, queue, param, no_destroy, destroy_delay)
except Exception as e:
pytest.fail(e)
def assert_js(webview, func_name, expected_result, uid='master'):
execute_func = 'window.pywebview.api.{0}()'.format(func_name)
check_func = 'window.pywebview._returnValues["{0}"].value'.format(func_name)
webview.evaluate_js(execute_func, uid)
result = webview.evaluate_js(check_func, uid)
counter, MAX = 0, 50
while result is None:
if counter == MAX:
raise AssertionError('assert_js timeout')
else:
counter += 1
time.sleep(0.1)
result = webview.evaluate_js(check_func, uid)
assert expected_result == result
def _create_window(webview, main_func, thread_func, queue, param, no_destroy, destroy_delay):
def thread(destroy_event, param):
try:
if thread_func:
thread_func()
destroy_event.set()
except Exception as e:
logger.exception(e, exc_info=True)
queue.put(traceback.format_exc())
destroy_event.set()
if not no_destroy:
args = (param,) if param else ()
destroy_event = _destroy_window(webview, destroy_delay)
t = threading.Thread(target=thread, args=(destroy_event, args))
t.start()
main_func()
def get_test_name():
return os.environ.get('PYTEST_CURRENT_TEST').split(':')[-1].split(' ')[0]
def _destroy_window(webview, delay):
def stop():
event.wait()
time.sleep(delay)
webview.destroy_window()
if sys.platform == 'darwin':
from .util_cocoa import mouseMoveRelative
time.sleep(1)
mouseMoveRelative(1, 1)
event = threading.Event()
event.clear()
t = threading.Thread(target=stop)
t.start()
return event
|
