forked from douban/code
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbase.py
More file actions
73 lines (53 loc) · 1.96 KB
/
base.py
File metadata and controls
73 lines (53 loc) · 1.96 KB
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# encoding: UTF-8
import os
import unittest
import webtest
from vilya.libs.rdstore import rds
from vilya.libs.gyt import clear_memoizer
from vilya.libs.store import store
from vilya.models.user import User
from vilya.models.api_key import ApiKey
from vilya.models.gist import Gist
# Notify和Feed这两个引用是必要的的，这样才能使notification和feed的signal正常工作
from vilya.models.notification import Notification # noqa
from vilya.models.feed import Feed # noqa
os.environ['CODEDOUBAN_RUNNING_MODE'] = 'unittest'
def _flush_mc_server(mc):
pass
def clear_beansdb_for_test():
pass
class TestApp(webtest.TestApp):
"""
The same with `webtest.TestApp`_ except that the app parameter defaults to
"""
def __init__(self, app=None, **kwargs):
super(TestApp, self).__init__(app, **kwargs)
class TestCase(unittest.TestCase):
def setUp(self):
_flush_mc_server(None)
clear_beansdb_for_test()
def tearDown(self):
clear_memoizer()
rds.flushdb()
store.rollback_all(force=True)
def addUser(self, name='unittest_user'):
return User(name)
def _add_api_key(self):
name = 'test'
desc = ''
type = ApiKey.TYPE_WEB
url = 'http://www.douban.com'
redirect_uri = 'http://www.douban.com/callback'
owner_id = 'testuser'
return ApiKey.add(name, desc, type, url, redirect_uri, owner_id)
def _add_gist(self, description='', owner_id='testuser', is_public=1,
gist_names=[], gist_contents=[], fork_from=None):
return Gist.add(description, owner_id, is_public,
gist_names=gist_names, gist_contents=gist_contents,
fork_from=fork_from)
env_for_git = {
'GIT_AUTHOR_NAME': 'default_test',
'GIT_AUTHOR_EMAIL': '[email protected]',
'GIT_COMMITTER_NAME': 'default_test',
'GIT_COMMITTER_EMAIL': '[email protected]',
}