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]))
