tests: add common functions useful for testing

Signed-off-by: Yuxuan Shui <yshuiv7@gmail.com>
This commit is contained in:
Yuxuan Shui 2020-03-20 19:14:42 +00:00
parent 4c22d564b2
commit 323a59b933
No known key found for this signature in database
GPG Key ID: 37C999F617EA1A47
1 changed files with 43 additions and 0 deletions

43
tests/testcases/common.py Normal file
View File

@ -0,0 +1,43 @@
import xcffib.xproto as xproto
import xcffib.randr as randr
import time
import random
import string
def set_window_name(conn, wid, name):
prop_name = "_NET_WM_NAME"
prop_name = conn.core.InternAtom(True, len(prop_name), prop_name).reply().atom
str_type = "STRING"
str_type = conn.core.InternAtom(True, len(str_type), str_type).reply().atom
conn.core.ChangePropertyChecked(xproto.PropMode.Replace, wid, prop_name, str_type, 8, len(name), name).check()
def find_picom_window(conn):
prop_name = "WM_NAME"
prop_name = conn.core.InternAtom(True, len(prop_name), prop_name).reply().atom
setup = conn.get_setup()
root = setup.roots[0].root
windows = conn.core.QueryTree(root).reply()
ext = xproto.xprotoExtension(conn)
for w in windows.children:
name = ext.GetProperty(False, w, prop_name, xproto.GetPropertyType.Any, 0, (2 ** 32) - 1).reply()
if name.value.buf() == b"picom":
return w
def trigger_root_configure(conn):
setup = conn.get_setup()
root = setup.roots[0].root
# Xorg sends root ConfigureNotify when we add a new mode to an output
rr = conn(randr.key)
name = ''.join([random.choice(string.ascii_letters + string.digits) for n in range(0, 32)])
mode_info = randr.ModeInfo.synthetic(id = 0, width = 1000, height = 1000, dot_clock = 0,
hsync_start = 0, hsync_end = 0, htotal = 0, hskew = 0, vsync_start = 0, vsync_end = 0,
vtotal = 0, name_len = len(name), mode_flags = 0)
reply = rr.CreateMode(root, mode_info, len(name), name).reply()
mode = reply.mode
reply = rr.GetScreenResourcesCurrent(root).reply()
# our xvfb is setup to only have 1 output
output = reply.outputs[0]
rr.AddOutputModeChecked(output, mode).check()
rr.SetCrtcConfig(reply.crtcs[0], reply.timestamp, reply.config_timestamp, 0, 0, mode, randr.Rotation.Rotate_0, 1, [output]).reply()