ValidatorTests

Complexity

Total Complexity

3

Size/Duplication

Total Lines	8
Duplicated Lines	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
A	test_notfound_get_validator_by_name()	0	2	1
A	setUp()	0	3	1
A	test_base_get_validator_by_name()	0	2	1

from django.test import TestCase

from django.core.exceptions import ValidationError

from sigma_core.models.validator import get_validator_by_name
from sigma_core.models.validator import Validator

def are_validator_fields_valid(validator, fields):
    try:
        v = get_validator_by_name(validator)
        return v['validate_fields'](fields)
    except ValidationError:
        return False

def is_validator_input_valid(validator, fields, value):
    try:
        v = get_validator_by_name(validator)
        v['validate_input'](fields, value)
        return True
    except ValidationError:
        return False

class ValidatorTests(TestCase):
    @classmethod
    def setUp(self):
        pass
    def test_notfound_get_validator_by_name(self):
        self.assertEqual(get_validator_by_name('notfound-validator'), None)
    def test_base_get_validator_by_name(self):
        self.assertNotEqual(get_validator_by_name(Validator.VALIDATOR_NONE), None)

class TextValidatorTests(TestCase):
    @classmethod
    def setUp(self):
        pass
    def test_get_validator_by_name(self):
        self.assertNotEqual(get_validator_by_name(Validator.VALIDATOR_TEXT), None)
    def test_regex1(self):
        fields = {'regex': "\s*(?P<header>[^:]+)\s*:(?P<value>.*?)\s*$", 'message': 'Err msg' }
        self.assertTrue(are_validator_fields_valid(Validator.VALIDATOR_TEXT, fields))
    def test_regex2(self):
        # The following pattern excludes filenames that end in either bat or exe:
        # https://docs.python.org/3/howto/regex.html
        fields = {'regex': ".*[.](?!bat$|exe$).*$", 'message': 'Err msg' }
        self.assertTrue(are_validator_fields_valid(Validator.VALIDATOR_TEXT, fields))
        self.assertTrue(is_validator_input_valid(Validator.VALIDATOR_TEXT, fields, 'file.png'))
        self.assertFalse(is_validator_input_valid(Validator.VALIDATOR_TEXT, fields, 'file.exe'))
    def test_evil_regex(self):
        fields = {'regex': "^(([a-z])+.)+[A-Z]([a-z])+$", 'message': 'Err msg' }
        value = 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa!'
        self.assertTrue(are_validator_fields_valid(Validator.VALIDATOR_TEXT, fields))
        self.assertFalse(is_validator_input_valid(Validator.VALIDATOR_TEXT, fields, value))