Python ModelDataHelper.fill_model Examples

Example #1

File: tests.py Project: niwinz/django-sampledatahelper

    def test_db_object_from_queryset(self):
        mdh = ModelDataHelper()

        with self.assertRaises(NotChoicesFound):
            self.sd.db_object_from_queryset(TestRelatedModel.objects.all())

        mdh.fill_model(TestRelatedModel, 10)

        self.assertTrue(isinstance(
            self.sd.db_object_from_queryset(TestRelatedModel.objects.all()),
            TestRelatedModel
        ))
        TestRelatedModel.objects.all().delete()

Example #2

File: tests.py Project: kaleidos/django-sampledatahelper

    def test_db_object_string(self):
        mdh = ModelDataHelper()

        with self.assertRaises(NotChoicesFound):
            self.sd.db_object('tests.TestRelatedModel')

        mdh.fill_model(models.TestRelatedModel, 1)

        self.assertTrue(isinstance(
            self.sd.db_object('tests.TestRelatedModel'),
            models.TestRelatedModel
        ))
        models.TestRelatedModel.objects.all().delete()