From a7dd41b675ca221da43948358ffc6973bc07b35f Mon Sep 17 00:00:00 2001 From: Tyler Gu Date: Sun, 21 Jan 2024 16:13:50 -0600 Subject: [PATCH] Create test generator directory Signed-off-by: Tyler Gu --- acto/input/{ => test_generators}/generator.py | 3 +++ test/integration_tests/test_testcase_generator_decorator.py | 6 +++++- 2 files changed, 8 insertions(+), 1 deletion(-) rename acto/input/{ => test_generators}/generator.py (98%) diff --git a/acto/input/generator.py b/acto/input/test_generators/generator.py similarity index 98% rename from acto/input/generator.py rename to acto/input/test_generators/generator.py index d26890b855..784b2d009c 100644 --- a/acto/input/generator.py +++ b/acto/input/test_generators/generator.py @@ -4,6 +4,8 @@ from dataclasses import dataclass from typing import Callable, Literal, Optional +import pydantic + from acto.input.k8s_schemas import KubernetesObjectSchema from acto.input.testcase import TestCase from acto.schema import ( @@ -84,6 +86,7 @@ def generator( ), "One of k8s_schema_name, schema_name, schema_type, paths must be specified" def wrapped_func(func: Callable[[BaseSchema], list[TestCase]]): + func = pydantic.validate_call(func) gen_obj = TestGenerator( k8s_schema_name, field_name, diff --git a/test/integration_tests/test_testcase_generator_decorator.py b/test/integration_tests/test_testcase_generator_decorator.py index e1d18056c2..8cbd3e6d52 100644 --- a/test/integration_tests/test_testcase_generator_decorator.py +++ b/test/integration_tests/test_testcase_generator_decorator.py @@ -6,8 +6,12 @@ import yaml -from acto.input.generator import TEST_GENERATORS, generator, get_testcases from acto.input.k8s_schemas import K8sSchemaMatcher +from acto.input.test_generators.generator import ( + TEST_GENERATORS, + generator, + get_testcases, +) from acto.input.testcase import TestCase from acto.schema import extract_schema