def test_integer_validator(self):
v = IntegerValidator()
matrix = [
("1", True),
("1,2,3", False),
("a,b,c", False),
("a", False),
("1,2,c", False),
]
self.run_matrix_test(matrix, v)
def test_integer_validator(self):
v = IntegerValidator()
matrix = [
('1', True),
('1,2,3', False),
('a,b,c', False),
('a', False),
('1,2,c', False)
]
self.run_matrix_test(matrix, v)
def test_linked(self):
p1 = params.EnumStringParameter(
name="attribute",
description="Testing123",
enum_values=["a", "b"],
is_required=True,
)
p2 = params.SimpleStringParameter(
name="downstream",
validator=LinkedValidator(
discriminator_parameter="attribute",
default_validator=TypeValidator("string"),
value_map={
"a": BooleanStringValidator(),
"b": IntegerValidator()
},
),
description="test123",
)
print(p2.validator.validation_criteria())
# p1.set_value('a')
p2.validator.inject_discriminator(None)
test_matrix = [
("true", "true"),
("blah", "blah"), # p1 not set, so uses default
]
self.run_matrix_test(test_matrix, p2)
p1._param_value = None
p2._param_value = None
p2.validator.inject_discriminator("a")
p1.set_value("a")
test_matrix = [
("true", "true"),
("blah", False), # should fail now that p1 has a value
]
self.run_matrix_test(test_matrix, p2)
p1._param_value = None
p2._param_value = None
p1.set_value("b")
p2.validator.inject_discriminator("b")
test_matrix = [("true", False), ("blah", False), ("123", "123")]
self.run_matrix_test(test_matrix, p2)
class ImageMetadataAttributeCheckTrigger(BaseTrigger):
__trigger_name__ = "attribute"
__description__ = (
"Triggers if a named image metadata value matches the given condition."
)
__ops__ = {
"=":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: x == y),
"!=":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: x != y),
">":
CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x > y),
"<":
CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x < y),
">=":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: x >= y),
"<=":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: x <= y),
"exists":
CheckOperation(requires_rvalue=False,
eval_function=lambda x, y: bool(x)),
"not_exists":
CheckOperation(requires_rvalue=False,
eval_function=lambda x, y: not bool(x)),
"like":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: bool(re.match(y, x))),
"not_like":
CheckOperation(requires_rvalue=True,
eval_function=lambda x, y: not bool(re.match(y, x))),
"in":
CheckOperation(
requires_rvalue=True,
eval_function=lambda x, y: x in [z.strip() for z in y.split(",")],
),
"not_in":
CheckOperation(
requires_rvalue=True,
eval_function=lambda x, y: x not in
[z.strip() for z in y.split(",")],
),
}
__valid_attributes__ = {
"size":
(lambda x: x.size, RegexParamValidator(regex=BYTES_REGEX.pattern)),
"architecture": (
lambda x: x.docker_data_json.get("Architecture")
if x.docker_data_json else None,
TypeValidator("string"),
),
"os_type": (
lambda x: x.docker_data_json.get("Os", x.docker_data_json.get("os")
) if x.docker_data_json else None,
TypeValidator("string"),
),
"distro": (lambda x: x.distro_name, TypeValidator("string")),
"distro_version":
(lambda x: x.distro_version, TypeValidator("string")),
"like_distro": (lambda x: x.like_distro, TypeValidator("string")),
"layer_count": (
lambda x: len(x.layers_json) if x.layers_json else 0,
IntegerValidator(),
),
}
attribute = EnumStringParameter(
name="attribute",
example_str="size",
description="Attribute name to be checked.",
enum_values=list(__valid_attributes__.keys()),
is_required=True,
sort_order=1,
)
check = EnumStringParameter(
name="check",
example_str=">",
description="The operation to perform the evaluation.",
enum_values=list(__ops__.keys()),
is_required=True,
sort_order=2,
)
_check_value_validator = LinkedValidator(
discriminator_parameter="attribute",
default_validator=TypeValidator("string"),
value_map={k: v[1]
for k, v in __valid_attributes__.items()},
)
check_value = TriggerParameter(
name="value",
example_str="1073741824",
description="Value used in comparison.",
validator=_check_value_validator,
is_required=False,
sort_order=3,
)
def evaluate(self, image_obj, context):
attr = self.attribute.value()
check = self.check.value()
rval = self.check_value.value()
if not attr or not check:
return
op = self.__ops__.get(check)
if op is None or op.requires_rvalue and not rval:
# Raise exception or fall thru
return
img_val = self.__valid_attributes__[attr][0](image_obj)
# Make consistent types (specifically for int/float/str)
if type(img_val) in [int, float, str]:
if attr == "size":
rval = convert_bytes_size(rval)
else:
rval = type(img_val)(rval)
if op.eval_function(img_val, rval):
self._fire(
msg=
"Attribute check for attribute: '{}' check: '{}' check_value: '{}' matched image value: '{}'"
.format(attr, check, (
str(rval) if rval is not None else ""), img_val))
class ImageMetadataAttributeCheckTrigger(BaseTrigger):
__trigger_name__ = 'attribute'
__description__ = 'Triggers if a named image metadata value matches the given condition.'
__ops__ = {
'=': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x == y),
'!=': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x != y),
'>': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x > y),
'<': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x < y),
'>=': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x >= y),
'<=': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x <= y),
'exists': CheckOperation(requires_rvalue=False, eval_function=lambda x, y: bool(x)),
'not_exists': CheckOperation(requires_rvalue=False, eval_function=lambda x, y: not bool(x)),
'like': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: bool(re.match(y, x))),
'not_like': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: not bool(re.match(y, x))),
'in': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x in [z.strip() for z in y.split(',')]),
'not_in': CheckOperation(requires_rvalue=True, eval_function=lambda x, y: x not in [z.strip() for z in y.split(',')])
}
__valid_attributes__ = {
'size': (lambda x: x.size, RegexParamValidator(regex=BYTES_REGEX.pattern)),
'architecture': (lambda x: x.docker_data_json.get('Architecture') if x.docker_data_json else None, TypeValidator('string')),
'os_type': (lambda x: x.docker_data_json.get('Os', x.docker_data_json.get('os')) if x.docker_data_json else None, TypeValidator('string')),
'distro': (lambda x: x.distro_name, TypeValidator('string')),
'distro_version': (lambda x: x.distro_version, TypeValidator('string')),
'like_distro': (lambda x: x.like_distro, TypeValidator('string')),
'layer_count': (lambda x: len(x.layers_json) if x.layers_json else 0, IntegerValidator())
}
attribute = EnumStringParameter(name='attribute', example_str='size', description='Attribute name to be checked.', enum_values=list(__valid_attributes__.keys()), is_required=True, sort_order=1)
check = EnumStringParameter(name='check', example_str='>', description='The operation to perform the evaluation.', enum_values=list(__ops__.keys()), is_required=True, sort_order=2)
_check_value_validator = LinkedValidator(discriminator_parameter='attribute', default_validator=TypeValidator('string'), value_map={k: v[1] for k, v in __valid_attributes__.items()})
check_value = TriggerParameter(name='value',
example_str='1073741824',
description='Value used in comparison.',
validator=_check_value_validator, is_required=False, sort_order=3)
def evaluate(self, image_obj, context):
attr = self.attribute.value()
check = self.check.value()
rval = self.check_value.value()
if not attr or not check:
return
op = self.__ops__.get(check)
if op is None or op.requires_rvalue and not rval:
# Raise exception or fall thru
return
img_val = self.__valid_attributes__[attr][0](image_obj)
# Make consistent types (specifically for int/float/str)
if type(img_val) in [int, float, str]:
if attr == 'size':
rval = convert_bytes_size(rval)
else:
rval = type(img_val)(rval)
if op.eval_function(img_val, rval):
self._fire(msg="Attribute check for attribute: '{}' check: '{}' check_value: '{}' matched image value: '{}'".format(attr, check, (str(rval) if rval is not None else ''), img_val))