class MyCheck(cb.Check):
title = 'My check'
description = 'A description'
parameters = [
cb.Parameter(name='my_param', data_type=cb.DataType.STRING),
cb.Parameter(name='other_param', data_type=cb.DataType.INT),
]
result_fields = [
cb.ResultField(name='my_result', data_type=cb.DataType.STRING),
]
def test_cast_none_date(self): p = cb.Parameter(name='', title='', data_type=cb.DataType.DATE) v = p.cast(None) self.assertEqual(type(v), date)
def test_cast_empty_str_date(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.DATE)
v = p.cast('')
self.assertEqual(type(v), date)
def test_cast_str_date(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.DATE)
v = p.cast('2010-01-01')
self.assertEqual(type(v), date)
self.assertEqual(v, date(2010, 1, 1))
def test_validate_none(self): p = cb.Parameter(name='', title='', data_type=cb.DataType.STRING, default='test') v = p.validate(None) self.assertEqual(type(v), str) self.assertEqual(v, 'test')
def test_cast_list_none(self): p = cb.Parameter(name='', title='', data_type=cb.DataType.LIST) v = p.cast(None) self.assertEqual(type(v), list) self.assertEqual(v, [])
def test_cast_list(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.LIST)
v = p.cast('1,2,3')
self.assertEqual(type(v), list)
self.assertEqual(v, ['1', '2', '3'])
def test_cast_string(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.STRING)
v = p.cast('test string')
self.assertEqual(type(v), str)
self.assertEqual(v, 'test string')
def test_cast_boolean_none(self): p = cb.Parameter(name='', title='', data_type=cb.DataType.BOOLEAN) v = p.cast(None) self.assertEqual(type(v), bool) self.assertEqual(v, False)
def test_cast_boolean_true(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.BOOLEAN)
v = p.cast('true')
self.assertEqual(type(v), bool)
self.assertEqual(v, True)
def test_cast_empty_str_datetime(self):
p = cb.Parameter(name='', title='', data_type=cb.DataType.DATETIME, default=None)
v = p.cast('')
self.assertEqual(type(v), datetime)
