def setUp(self):
self.dir_name = os.path.dirname(__file__)
apps = list(settings.INSTALLED_APPS)
apps.append('nomnom.tests')
settings.INSTALLED_APPS = tuple(apps)
loading.cache.loaded = False
call_command('syncdb', verbosity=0)
from nomnom.tests.models import Department, Instructor, Course
self.department1 = Department(name="Intergalactic Exploration", code="INX")
self.department1.save()
self.instructor1 = Instructor(name="James Kirk", title="Capt.", department=self.department1)
self.instructor1.save()
self.course1 = Course(name="Photon-based Ballistics", ext_id="INX-324")
self.course1.save()
self.department2 = Department(name="Paranormal Investigation", code="PNV")
self.department2.save()
self.instructor2 = Instructor(name="Peter Venkman", title="Dr.", department=self.department2)
self.instructor2.save()
self.course2 = Course(name="Proton Pack Safety", ext_id="PNV-252")
self.course2.save()
self.course3 = Course(name="Spores, Molds, and Fungus", ext_id="PNV-112")
self.course3.save()
class ModelsTest(TestCase):
"""
These tests require the existence of the models in nomnom/tests/models.py,
hence they are separated from the others. Technique described here:
http://stackoverflow.com/questions/502916/django-how-to-create-a-model-dynamically-just-for-testing
"""
def setUp(self):
self.dir_name = os.path.dirname(__file__)
apps = list(settings.INSTALLED_APPS)
apps.append('nomnom.tests')
settings.INSTALLED_APPS = tuple(apps)
loading.cache.loaded = False
call_command('syncdb', verbosity=0)
from nomnom.tests.models import Department, Instructor, Course
self.department1 = Department(name="Intergalactic Exploration", code="INX")
self.department1.save()
self.instructor1 = Instructor(name="James Kirk", title="Capt.", department=self.department1)
self.instructor1.save()
self.course1 = Course(name="Photon-based Ballistics", ext_id="INX-324")
self.course1.save()
self.department2 = Department(name="Paranormal Investigation", code="PNV")
self.department2.save()
self.instructor2 = Instructor(name="Peter Venkman", title="Dr.", department=self.department2)
self.instructor2.save()
self.course2 = Course(name="Proton Pack Safety", ext_id="PNV-252")
self.course2.save()
self.course3 = Course(name="Spores, Molds, and Fungus", ext_id="PNV-112")
self.course3.save()
def test_export_fk_as_id(self):
"""
CSVs should (by default) export FKs as IDs, not titles
"""
from nomnom.tests.models import Instructor
request_factory = RequestFactory()
req = request_factory.get('/admin/test/instructor/')
response = export_as_csv(modeladmin=self.instructor1.__class__,
request=req,
queryset=Instructor.objects.all(),
export_type="D")
expected_response = 'department,title,id,name,courses\r\n1,Capt.,1,James Kirk,\r\n2,Dr.,2,Peter Venkman,'
self.assertContains(response, expected_response)
def test_upload_fk_as_id(self):
"""
Allow unique-field values in M2M columns in the CSV upload
"""
f = open(os.path.join(self.dir_name, 'files/instructors_fk.csv'), "r")
uploaded_file = SimpleUploadedFile('instructors_fk.csv', f.read())
output = handle_uploaded_file(uploaded_file, 'tests', 'instructor')
from nomnom.tests.models import Instructor
# 4 Instructors total
self.assertEquals(Instructor.objects.all().count(), 4)
def test_upload_fk_as_unique(self):
"""
Allow unique-field values in M2M columns in the CSV upload
"""
f = open(os.path.join(self.dir_name, 'files/instructors_fk_unique.csv'), "r")
uploaded_file = SimpleUploadedFile('instructors_fk_unique.csv', f.read())
output = handle_uploaded_file(uploaded_file, 'tests', 'instructor')
from nomnom.tests.models import Instructor
# 4 Instructors total
self.assertEquals(Instructor.objects.all().count(), 4)
def test_upload_m2m_as_unique(self):
"""
Allow unique-field values in M2M columns in the CSV upload
"""
f = open(os.path.join(self.dir_name, 'files/instructors_m2m.csv'), "r")
uploaded_file = SimpleUploadedFile('instructors.csv', f.read())
output = handle_uploaded_file(uploaded_file, 'tests', 'instructor')
from nomnom.tests.models import Instructor, Course
# 4 Instructors total
self.assertEquals(Instructor.objects.all().count(), 4)
# Courses were properly attributed to Instructors
courses = Course.objects.filter(id__in=[2,3])
spengler = Instructor.objects.get(id=4)
self.assertQuerysetEqual(spengler.courses.all(), [repr(c) for c in courses], ordered=False)
def test_upload_blank_num_field(self):
"""
Allow null values for nullable number fields
"""
f = open(os.path.join(self.dir_name, 'files/courses_blank_creds.csv'), "r")
uploaded_file = SimpleUploadedFile('courses_blank_creds.csv', f.read())
output = handle_uploaded_file(uploaded_file, 'tests', 'course')
from nomnom.tests.models import Course
self.assertEquals(Course.objects.all().count(), 5)
def test_handle_float_field(self):
"""
Make sure instantiate_from_row() can handle floats (and blank floats)
"""
from nomnom.tests.models import Department
row = {'name' : 'Magical Creatures', 'code' : 'WIZ-203', 'avg_gpa' : '3.25'}
instantiate_from_row(Department, row).save()
self.assertEquals(Department.objects.all().count(), 3)
self.assertEquals(Department.objects.get(id=3).avg_gpa, 3.25)
def test_get_unique_fields(self):
"""
Confirm that we get back a list of unique=True fields names
"""
from nomnom.tests.models import Department
self.assertListEqual(['id','code'], get_unique_field_names(Department))
def test_lookup_by_unique(self):
"""
If no ID is provided, but a value for a unique field is, attempt to update instead of insert
"""
from nomnom.tests.models import Course
initial_count = Course.objects.all().count()
row = {"name" : "Photon Torpedo Strategy", "ext_id" : "INX-324"}
instantiate_from_row(Course, row).save()
self.assertEqual(initial_count, Course.objects.all().count())
self.assertEqual(Course.objects.get(id=1).name, "Photon Torpedo Strategy")
