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