コード例 #1
0
ファイル: tests.py プロジェクト: alexmerser/Atlas-LMS
	def testSaveFileToUploadDirectory(self):
		from django.conf import settings
		upload_dir = settings.UPLOAD_DIR
		
		self.assertTrue(os.access(upload_dir, os.W_OK)) #make sure upload dir can be written to
		
		#create a temporary file
		tempFilePath = os.getcwd() + "/assignments/tempFile.txt"
		with open(tempFilePath, "wr") as tmpFile:
			file = File(tmpFile)
			file.name = "tempFile.txt"
				
			self.assertEqual(upload_dir + file.__unicode__(), save_file_to_upload_dir(file))
		
			self.assertTrue(os.path.exists(upload_dir + file.__unicode__())) #file was written to upload directory
		
			self.assertEqual(upload_dir + file.__unicode__(), save_file_to_upload_dir(file)) #overwrite
			
			self.assertTrue(os.path.exists(upload_dir + file.__unicode__()))
			
			os.remove(upload_dir + file.__unicode__())
			self.assertFalse(os.path.exists(upload_dir + file.__unicode__()))
			
		os.remove(tempFilePath)
		self.assertFalse(os.path.exists(tempFilePath))
コード例 #2
0
    def testSaveFileToUploadDirectory(self):
        from django.conf import settings
        upload_dir = settings.UPLOAD_DIR

        self.assertTrue(os.access(
            upload_dir, os.W_OK))  #make sure upload dir can be written to

        #create a temporary file
        tempFilePath = os.getcwd() + "/assignments/tempFile.txt"
        with open(tempFilePath, "wr") as tmpFile:
            file = File(tmpFile)
            file.name = "tempFile.txt"

            self.assertEqual(upload_dir + file.__unicode__(),
                             save_file_to_upload_dir(file))

            self.assertTrue(os.path.exists(
                upload_dir +
                file.__unicode__()))  #file was written to upload directory

            self.assertEqual(upload_dir + file.__unicode__(),
                             save_file_to_upload_dir(file))  #overwrite

            self.assertTrue(os.path.exists(upload_dir + file.__unicode__()))

            os.remove(upload_dir + file.__unicode__())
            self.assertFalse(os.path.exists(upload_dir + file.__unicode__()))

        os.remove(tempFilePath)
        self.assertFalse(os.path.exists(tempFilePath))