Extend tests for invalid data

This commit is contained in:
Jeremy Stretch 2024-04-29 12:33:02 -04:00
parent 0bf8cd7690
commit 22c2e7e6e4
2 changed files with 14 additions and 2 deletions

View File

@ -287,11 +287,11 @@ An IPv4 or IPv6 network with a mask. Returns a `netaddr.IPNetwork` object. Two a
### DateVar
A date. Returns a `datetime.date` object.
A calendar date. Returns a `datetime.date` object.
### DateTimeVar
A datetime. Returns a `datetime.datetime` object.
A complete date & time. Returns a `datetime.datetime` object.
## Running Custom Scripts

View File

@ -331,6 +331,12 @@ class ScriptVariablesTest(TestCase):
var1 = DateVar()
var2 = DateVar(required=False)
# Test date validation
data = {'var1': 'not a date'}
form = TestScript().as_form(data, None)
self.assertFalse(form.is_valid())
self.assertIn('var1', form.errors)
# Validate valid data
input_date = date(2024, 4, 1)
data = {'var1': input_date}
@ -347,6 +353,12 @@ class ScriptVariablesTest(TestCase):
var1 = DateTimeVar()
var2 = DateTimeVar(required=False)
# Test datetime validation
data = {'var1': 'not a datetime'}
form = TestScript().as_form(data, None)
self.assertFalse(form.is_valid())
self.assertIn('var1', form.errors)
# Validate valid data
input_datetime = datetime(2024, 4, 1, 8, 0, 0, 0, timezone.utc)
data = {'var1': input_datetime}