vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

file-api: migrate and fix tests

Browse source 
successfully executed all test cases

Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com>
This commit is contained in:
Chinmay Pai 2018-09-06 01:18:39 +05:30
parent 6eca292e1a
commit a80c23c9be
No known key found for this signature in database
GPG key ID: 75507BE256F40CED
1 changed files with 122 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

140
frappe/core/doctype/file/test_file.py
View file

@ -4,12 +4,105 @@
from __future__ import unicode_literals
import frappe
import os
import unittest
from frappe.utils.file_manager import get_files_path
from frappe import _
from frappe.core.doctype.file.file import move_file
from frappe.core.doctype.file.file import move_file, get_file, get_files_path
# test_records = frappe.get_test_records('File')
test_content1 = 'Hello'
test_content2 = 'Hello World'
def make_test_doc():
d = frappe.new_doc('ToDo')
d.description = 'Test'
d.save()
return d.doctype, d.name
class TestSimpleFile(unittest.TestCase):
def setUp(self):
self.attached_to_doctype, self.attached_to_docname = make_test_doc()
self.test_content = test_content1
_file = frappe.get_doc({"doctype": "File",
"file_name": "hello.txt",
"attached_to_doctype": self.attached_to_doctype,
"attached_to_name": self.attached_to_docname})
self.saved_file = _file.save_file(content=self.test_content)
self.saved_filename = get_files_path(self.saved_file.file_name)
def test_save(self):
_, content = get_file(self.saved_file.name)
self.assertEqual(content, self.test_content)
def tearDown(self):
# File gets deleted on rollback, so blank
pass
class TestSameFileName(unittest.TestCase):
def setUp(self):
self.attached_to_doctype, self.attached_to_docname = make_test_doc()
self.test_content1 = test_content1
self.test_content2 = test_content2
_file1 = frappe.get_doc({"doctype": "File",
"file_name": "hello.txt",
"attached_to_doctype": self.attached_to_doctype,
"attached_to_name": self.attached_to_docname})
_file2 = frappe.get_doc({"doctype": "File",
"file_name": "hello.txt",
"attached_to_doctype": self.attached_to_doctype,
"attached_to_name": self.attached_to_docname})
self.saved_file1 = _file1.save_file(content=self.test_content1)
self.saved_file2 = _file2.save_file(content=self.test_content2)
self.saved_filename1 = get_files_path(self.saved_file1.file_name)
self.saved_filename2 = get_files_path(self.saved_file2.file_name)
def test_saved_content(self):
_, content1 = get_file(self.saved_file1.name)
self.assertEqual(content1, self.test_content1)
_, content2 = get_file(self.saved_file2.name)
self.assertEqual(content2, self.test_content2)
def tearDown(self):
# File gets deleted on rollback, so blank
pass
class TestSameContent(unittest.TestCase):
def setUp(self):
self.attached_to_doctype1, self.attached_to_docname1 = make_test_doc()
self.attached_to_doctype2, self.attached_to_docname2 = make_test_doc()
self.test_content1 = test_content1
self.test_content2 = test_content1
self.orig_filename = 'hello.txt'
self.dup_filename = 'hello2.txt'
_file1 = frappe.get_doc({"doctype": "File",
"file_name": self.orig_filename,
"attached_to_doctype": self.attached_to_doctype1,
"attached_to_name": self.attached_to_docname1})
_file2 = frappe.get_doc({"doctype": "File",
"file_name": self.dup_filename,
"attached_to_doctype": self.attached_to_doctype2,
"attached_to_name": self.attached_to_docname2})
self.saved_file1 = _file1.save_file(content=self.test_content1)
self.saved_file2 = _file2.save_file(content=self.test_content2)
self.saved_filename1 = get_files_path(self.saved_file1.file_name)
self.saved_filename2 = get_files_path(self.saved_file2.file_name)
def test_saved_content(self):
filename1, content1 = get_file(self.saved_file1.name)
filename2, content2 = get_file(self.saved_file2.name)
self.assertEqual(filename1, filename2)
self.assertFalse(os.path.exists(get_files_path(self.dup_filename)))
def tearDown(self):
# File gets deleted on rollback, so blank
pass
class TestFile(unittest.TestCase):
def setUp(self):
self.delete_test_data()
@ -27,9 +120,12 @@ class TestFile(unittest.TestCase):
frappe.delete_doc("File", f[0])
def upload_file(self):
_file = frappe.get_doc("File", {"file_name": "file_copy.txt", "content": "Testing file copy example.",
"attached_to_name": "", "attached_to_doctype": ""})
self.saved_file = _file.save_file(folder=self.get_folder("Test Folder 1", "Home").name)
_file = frappe.get_doc({"doctype": "File",
"file_name": "file_copy.txt",
"attached_to_name": "",
"attached_to_doctype": "",
"folder": self.get_folder("Test Folder 1", "Home").name})
self.saved_file = _file.save_file(content="Testing file copy example.")
self.saved_filename = get_files_path(self.saved_file.file_name)
def get_folder(self, folder_name, parent_folder="Home"):
@ -51,9 +147,9 @@ class TestFile(unittest.TestCase):
def test_file_copy(self):
folder = self.get_folder("Test Folder 2", "Home")
file = frappe.get_doc("File", {"file_name":"file_copy.txt"})
file = frappe.get_doc("File", {"file_name": "file_copy.txt"})
move_file([{"name": file.name}], folder.name, file.folder)
file = frappe.get_doc("File", {"file_name":"file_copy.txt"})
file = frappe.get_doc("File", {"file_name": "file_copy.txt"})
self.assertEqual(_("Home/Test Folder 2"), file.folder)
self.assertEqual(frappe.db.get_value("File", _("Home/Test Folder 2"), "file_size"), file.file_size)
@ -62,9 +158,12 @@ class TestFile(unittest.TestCase):
def test_folder_copy(self):
folder = self.get_folder("Test Folder 2", "Home")
folder = self.get_folder("Test Folder 3", "Home/Test Folder 2")
_file = frappe.get_doc("File", {"file_name": "folder_copy.txt", "content": "Testing folder copy example.",
"attached_to_name": "", "attached_to_doctype": ""})
self.saved_file = _file.save_file(folder=folder.name)
_file = frappe.get_doc({"doctype": "File",
"file_name": "folder_copy.txt",
"attached_to_name": "",
"attached_to_doctype": "",
"folder": folder.name})
self.saved_file = _file.save_file(content="Testing folder copy example")
move_file([{"name": folder.name}], 'Home/Test Folder 1', folder.folder)
@ -87,15 +186,18 @@ class TestFile(unittest.TestCase):
self.assertRaises(frappe.ValidationError, d.save)
def test_on_delete(self):
file = frappe.get_doc("File", {"file_name":"file_copy.txt"})
file = frappe.get_doc("File", {"file_name": "file_copy.txt"})
file.delete()
self.assertEqual(frappe.db.get_value("File", _("Home/Test Folder 1"), "file_size"), 0)
folder = self.get_folder("Test Folder 3", "Home/Test Folder 1")
_file = frappe.get_doc("File", {"file_name": "folder_copy.txt", "content": "Testing folder copy example.",
"attached_to_name": "", "attached_to_doctype": ""})
self.saved_file = _file.save_file(folder=folder.name)
_file = frappe.get_doc({"doctype": "File",
"file_name": "folder_copy.txt",
"attached_to_name": "",
"attached_to_doctype": "",
"folder": folder.name})
self.saved_file = _file.save_file(content="Testing folder copy example")
folder = frappe.get_doc("File", "Home/Test Folder 1/Test Folder 3")
self.assertRaises(frappe.ValidationError, folder.delete)
@ -118,11 +220,13 @@ class TestFile(unittest.TestCase):
# Rebuild the frappe.local.conf to take up the changes from site_config
frappe.local.conf = _dict(frappe.get_site_config())
_file = frappe.get_doc("File", {"file_name": "_test_max_space.txt",
"content": "This file tests for max space usage",
"attached_to_name": "", "attached_to_doctype": ""})
_file = frappe.get_doc({"doctype": "File",
"file_name": "_test_max_space.txt",
"attached_to_name": "",
"attached_to_doctype": "",
"folder": self.get_folder("Test Folder 2", "Home").name})
self.assertRaises(MaxFileSizeReachedError,
_file.save_file(self.get_folder("Test Folder 2", "Home").name))
_file.save_file, content="This file tests for max space usage")
# Scrub the site_config and rebuild frappe.local.conf
clear_limit("space")