def test_blankable(self):
v = ObjectField(self.simple_schema, 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
self.assertEqualSimpleSchema(v.generate())
v = ObjectField(self.simple_schema, blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
mocked_random.choice = random.choice
mocked_random.randint = random.randint
self.assertEqual(v.generate(), {})
def test_blankable(self):
v = ObjectField(self.simple_schema, 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
self.assertEqualSimpleSchema(v.generate())
v = ObjectField(self.simple_schema, blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
mocked_random.choice = random.choice
mocked_random.randint = random.randint
self.assertEqual(v.generate(), {})
def test_not_required(self):
v = ObjectField(self.simple_schema, required=False, not_present_prob=0)
self.assertEqualSimpleSchema(v.generate())
v = ObjectField(self.simple_schema, required=False, not_present_prob=1)
with self.assertRaises(NotGeneratedException):
v.generate()
v = ObjectField(self.simple_schema, 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 = ObjectField(self.simple_schema, 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
self.assertEqualSimpleSchema(v.generate())
def test_not_required(self):
v = ObjectField(self.simple_schema, required=False, not_present_prob=0)
self.assertEqualSimpleSchema(v.generate())
v = ObjectField(self.simple_schema, required=False, not_present_prob=1)
with self.assertRaises(NotGeneratedException):
v.generate()
v = ObjectField(self.simple_schema, 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 = ObjectField(self.simple_schema, 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
self.assertEqualSimpleSchema(v.generate())
def test_default_nested_schema(self):
v = ObjectField(self.nested_schema)
generated = v.generate()
self.assertEqual(set(generated.keys()), set(self.nested_schema.keys()))
self.assertEqualSimpleSchema(generated['simple'])
self.assertEqualComplexSchema(generated['complex'])
def test_default_complex_schema(self):
v = ObjectField(self.complex_schema)
self.assertEqualComplexSchema(v.generate())
def test_default_simple_schema(self):
v = ObjectField(self.simple_schema)
self.assertEqualSimpleSchema(v.generate())
def test_default_nested_schema(self):
v = ObjectField(self.nested_schema)
generated = v.generate()
self.assertEqual(set(generated.keys()), set(self.nested_schema.keys()))
self.assertEqualSimpleSchema(generated["simple"])
self.assertEqualComplexSchema(generated["complex"])
def test_default_complex_schema(self):
v = ObjectField(self.complex_schema)
self.assertEqualComplexSchema(v.generate())
def test_default_simple_schema(self):
v = ObjectField(self.simple_schema)
self.assertEqualSimpleSchema(v.generate())
