Skip to content

Instantly share code, notes, and snippets.

@katiayn
Created January 22, 2017 17:22
Show Gist options
  • Save katiayn/b0b2c8a05f09739dff9bb44fb3f956f1 to your computer and use it in GitHub Desktop.
Save katiayn/b0b2c8a05f09739dff9bb44fb3f956f1 to your computer and use it in GitHub Desktop.
from django.test import TestCase
import factory
from codekiwi.core.models import Employee
class EmployeeFactory(factory.DjangoModelFactory):
class Meta:
model = Employee
last_name = factory.Faker('last_name')
first_name = factory.Faker('first_name')
job = factory.Faker('job')
class EmployeeTestCase(TestCase):
def test_employee(self):
# Generate a fake employee instance
employee = EmployeeFactory()
# Test assertions
self.assertIsNotNone(employee.last_name)
self.assertIsNotNone(employee.first_name)
self.assertIsNotNone(employee.job)
def test_another_employee(self):
# Generate a fake employee with an explicit last_name
employee = EmployeeFactory(last_name='Nakamura')
# Test assertions
self.assertTrue(employee.last_name, 'Nakamura')
self.assertIsNotNone(employee.first_name)
self.assertIsNotNone(employee.job)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment