def test_min_max_length(self): v = StringField(min_length=5, max_length=10) generated = v.generate() self.assertTrue(5 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(min_length=0, max_length=0) generated = v.generate() self.assertEqual(generated, "") v = StringField(min_length=10, max_length=5) with self.assertRaises(ValueError): generated = v.generate()
def test_min_max_length(self): v = StringField(min_length=5, max_length=10) generated = v.generate() self.assertTrue(5 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(min_length=0, max_length=0) generated = v.generate() self.assertEqual(generated, '') v = StringField(min_length=10, max_length=5) with self.assertRaises(ValueError): generated = v.generate()
def test_blankable(self): v = StringField(blank=True, blank_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice mocked_random.randint = random.randint generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(blank=True, blank_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 generated = v.generate() self.assertEqual(generated, "")
def test_blankable(self): v = StringField(blank=True, blank_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice mocked_random.randint = random.randint generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(blank=True, blank_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 generated = v.generate() self.assertEqual(generated, '')
def test_nullable_blankable_choices(self): choices = ["a", "b", "c"] v = StringField(null=True, null_prob=0.3, blank=True, blank_prob=0.3, choices=choices) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 generated = v.generate() self.assertIsNone(generated) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0.5 mocked_random.choice = random.choice self.assertEqual(v.generate(), "") with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice self.assertIn(v.generate(), choices)
def test_nullable_blankable_choices(self): choices = ['a', 'b', 'c'] v = StringField(null=True, null_prob=0.3, blank=True, blank_prob=0.3, choices=choices) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 generated = v.generate() self.assertIsNone(generated) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0.5 mocked_random.choice = random.choice self.assertEqual(v.generate(), '') with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice self.assertIn(v.generate(), choices)
def test_not_required(self): v = StringField(required=False, not_present_prob=0) generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(required=False, not_present_prob=1) with self.assertRaises(NotGeneratedException): v.generate() v = StringField(required=False, not_present_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 with self.assertRaises(NotGeneratedException): v.generate() v = StringField(required=False, not_present_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice mocked_random.randint = random.randint generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated]))
def test_not_required(self): v = StringField(required=False, not_present_prob=0) generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated])) v = StringField(required=False, not_present_prob=1) with self.assertRaises(NotGeneratedException): v.generate() v = StringField(required=False, not_present_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 0 with self.assertRaises(NotGeneratedException): v.generate() v = StringField(required=False, not_present_prob=0.5) with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random: mocked_random.random.return_value = 1 mocked_random.choice = random.choice mocked_random.randint = random.randint generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated]))
def test_choices(self): choices = ['a', 'b', 'c'] v = StringField(choices=choices) self.assertIn(v.generate(), choices)
def test_charset(self): charset = '!@#$%^&*()_' v = StringField(charset=charset) generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in charset for s in generated]))
def test_default(self): v = StringField() generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated]))
def test_choices(self): choices = ["a", "b", "c"] v = StringField(choices=choices) self.assertIn(v.generate(), choices)
def test_charset(self): charset = "!@#$%^&*()_" v = StringField(charset=charset) generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in charset for s in generated]))
def test_default(self): v = StringField() generated = v.generate() self.assertTrue(1 <= len(generated) <= 10) self.assertTrue(all([s in (string.ascii_letters + string.digits) for s in generated]))