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)