Exemplo n.º 1
0
 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()
Exemplo n.º 2
0
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")