87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
import os
|
|
import infra.basetest
|
|
|
|
|
|
class TestPolkitInfra(infra.basetest.BRTest):
|
|
br2_external = [infra.filepath("tests/package/br2-external/polkit")]
|
|
config = \
|
|
"""
|
|
BR2_arm=y
|
|
BR2_cortex_a9=y
|
|
BR2_ARM_ENABLE_VFP=y
|
|
BR2_TOOLCHAIN_EXTERNAL=y
|
|
BR2_TOOLCHAIN_EXTERNAL_BOOTLIN=y
|
|
BR2_TARGET_ROOTFS_CPIO=y
|
|
BR2_PACKAGE_POLKIT=y
|
|
BR2_PACKAGE_POLKIT_RULES_TEST=y
|
|
"""
|
|
rule_paths = [
|
|
"/etc/polkit-1/rules.d",
|
|
"/usr/share/polkit-1/rules.d"
|
|
]
|
|
|
|
def base_test_run(self):
|
|
cpio_file = os.path.join(self.builddir, "images", "rootfs.cpio")
|
|
self.emulator.boot(arch="armv7", kernel="builtin",
|
|
options=["-initrd", cpio_file])
|
|
self.emulator.login()
|
|
|
|
|
|
class TestPolkitSystemd(TestPolkitInfra):
|
|
config = \
|
|
"""
|
|
{}
|
|
BR2_INIT_SYSTEMD=y
|
|
BR2_PACKAGE_SYSTEMD_POLKIT=y
|
|
BR2_TARGET_GENERIC_GETTY_PORT="ttyAMA0"
|
|
# BR2_TARGET_ROOTFS_TAR is not set
|
|
""".format(TestPolkitInfra.config)
|
|
|
|
def test_run(self):
|
|
TestPolkitInfra.base_test_run(self)
|
|
|
|
rule_file = "systemd-timesyncd-restart.rules"
|
|
for rule_path in TestPolkitInfra.rule_paths:
|
|
cmd = "su brtest -c '/bin/systemctl restart systemd-timesyncd.service'"
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertNotEqual(exit_code, 0)
|
|
|
|
cmd = "cp /root/{file} {path}".format(file=rule_file, path=rule_path)
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|
|
|
|
cmd = "su brtest -c '/bin/systemctl restart systemd-timesyncd.service'"
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|
|
|
|
cmd = "rm {path}/{file}".format(file=rule_file, path=rule_path)
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|
|
|
|
|
|
class TestPolkitInitd(TestPolkitInfra):
|
|
config = TestPolkitInfra.config
|
|
|
|
def test_run(self):
|
|
TestPolkitInfra.base_test_run(self)
|
|
|
|
rule_file = "hello-polkit.rules"
|
|
for rule_path in TestPolkitInfra.rule_paths:
|
|
cmd = "su brtest -c 'pkexec hello-polkit'"
|
|
output, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 127)
|
|
self.assertEqual(output[0], "Error executing command as another user: Not authorized")
|
|
|
|
cmd = "cp /root/{file} {path}/{file}".format(file=rule_file, path=rule_path)
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|
|
|
|
cmd = "su brtest -c 'pkexec hello-polkit'"
|
|
output, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|
|
self.assertEqual(output[0], "Hello polkit!")
|
|
|
|
cmd = "rm {path}/{file}".format(file=rule_file, path=rule_path)
|
|
_, exit_code = self.emulator.run(cmd, 10)
|
|
self.assertEqual(exit_code, 0)
|