/
test_service.py
55 lines (47 loc) · 1.62 KB
/
test_service.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# _*_ coding: utf-8 _*_
from lib.store import insert, update, find, find_one, remove, add_file
from service.task import get_not_conver_file
from service.task import convert_media
from service.task import remove_tmp_file
from service.task import update_task_data_media
import unittest
import os
class Test_account(unittest.TestCase):
_id = ""
task_id = ""
file_id =""
def setUp(self):
remove_tmp_file()
self._id = str(add_file(open("./tmp.mov").read(),
content_type="video/quicktime",
filename="tmp.mov",
))
update("fs.files",{"_id": self._id}, {"uuid": self._id})
self.task_id = insert("TaskData", {
"converd": False,
"entryList":[{
"fieldType":"Video", "uuid": self._id, "converdid": "",
}]
})
def tearDown(self):
remove("fs.file", {"_id": self._id}, real=True)
remove('TaskData', {"_id": self.task_id}, real=True)
def test_get_not_conver_file(self):
self.assertTrue(get_not_conver_file() > 0)
def test_convert_media(self):
try:
result = convert_media()
except:
from lib.utils import print_err
print_err()
print "convert "
print result
self.assertTrue(result)
update_task_data_media()
result = find_one("TaskData", {
"_id": self.task_id
} )
self.assertTrue(result.get("converd", False))
self.assertTrue(result.get("entryList")[0].get("converdid", False))
if __name__ == '__main__':
unittest.main()