model_bakery_django_test_fixture_creation_quickstart.py

1from baker import baker
2from django.test import TestCase
3from myapp.models import Customer
4
5class CustomerTestCase(TestCase):
6    def test_customer_creation(self):
7        # This will create and save a Customer instance with random data
8        customer = baker.make(Customer)
9        
10        # This will create and save a Customer instance with specific data
11        customer_named = baker.make(Customer, name='John Doe')
12        
13        # This will create a Customer instance without saving it to the database
14        customer_unsaved = baker.prepare(Customer)
15        
16        self.assertIsInstance(customer, Customer)
17        self.assertEqual(customer_named.name, 'John Doe')
18        self.assertIsNone(customer_unsaved.pk)