bbfdm/test/python-test-cases/python/validate_ubus_schema_method.py
2024-02-25 15:17:40 +01:00

80 lines
2.1 KiB
Python
Executable file

#!/usr/bin/python3
import sys
import ubus
import json
class TestArguments:
"""
Class to hold test arguments.
"""
def __init__(self, description, service, input, expected_error, output):
self.description = description
self.service = service
self.input = input
self.expected_error = expected_error
self.output = output
def run_test(args):
"""
Run a single test.
"""
print("Running: " + args.description)
out = ubus.call(args.service, "schema", args.input)
if isinstance(out, list) and out:
out = out[0]
else:
print("FAIL: " + args.description)
return
res_len = len(out["results"])
fault = out["results"][0]["fault"] if res_len and "fault" in out["results"][0] else 0
if fault != args.expected_error:
print("FAIL: " + args.description)
return
if args.expected_error == 0 and res_len == 0:
print("FAIL: " + args.description)
return
# Check if output matches expected output
if args.output != {}:
print(out)
if out == args.output:
print("PASS: " + args.description)
else:
print("FAIL: " + args.description)
else:
print("PASS: " + args.description)
if __name__ == "__main__":
# Check for correct command line arguments
if len(sys.argv) != 2:
print("Usage: {} <test_arguments_json>".format(sys.argv[0]))
sys.exit(1)
test_arguments_file = sys.argv[1]
try:
with open(test_arguments_file, 'r') as f:
test_arguments_data = json.load(f)
service_name = test_arguments_data.get("object", "bbfdm")
args_list = [TestArguments(**test_case, service=service_name) for test_case in test_arguments_data["schema"]]
except FileNotFoundError:
print("File not found:", test_arguments_file)
sys.exit(1)
except json.JSONDecodeError as e:
print("Error parsing JSON:", e)
sys.exit(1)
ubus.connect()
# Run tests for each set of arguments
for args in args_list:
run_test(args)
ubus.disconnect()