7
7
from django .test import TestCase
8
8
from django .test .utils import override_settings
9
9
10
- from django_comments .forms import CommentForm
11
- from django_comments .models import Comment
10
+ from django_comments import get_model , get_form
12
11
13
12
from testapp .models import Article , Author
14
13
@@ -42,7 +41,7 @@ def tearDown(self):
42
41
43
42
def createSomeComments (self ):
44
43
# Two anonymous comments on two different objects
45
- c1 = Comment .objects .create (
44
+ c1 = get_model () .objects .create (
46
45
content_type = CT (Article ),
47
46
object_pk = "1" ,
48
47
user_name = "Joe Somebody" ,
@@ -51,7 +50,7 @@ def createSomeComments(self):
51
50
comment = "First!" ,
52
51
site = Site .objects .get_current (),
53
52
)
54
- c2 = Comment .objects .create (
53
+ c2 = get_model () .objects .create (
55
54
content_type = CT (Author ),
56
55
object_pk = "1" ,
57
56
user_name = "Joe Somebody" ,
@@ -73,15 +72,15 @@ def createSomeComments(self):
73
72
is_active = True ,
74
73
is_superuser = False ,
75
74
)
76
- c3 = Comment .objects .create (
75
+ c3 = get_model () .objects .create (
77
76
content_type = CT (Article ),
78
77
object_pk = "1" ,
79
78
user = user ,
80
79
user_url = "http://example.com/~frank/" ,
81
80
comment = "Damn, I wanted to be first." ,
82
81
site = Site .objects .get_current (),
83
82
)
84
- c4 = Comment .objects .create (
83
+ c4 = get_model () .objects .create (
85
84
content_type = CT (Author ),
86
85
object_pk = "2" ,
87
86
user = user ,
@@ -101,7 +100,7 @@ def getData(self):
101
100
}
102
101
103
102
def getValidData (self , obj ):
104
- f = CommentForm (obj )
103
+ f = get_form () (obj )
105
104
d = self .getData ()
106
105
d .update (f .initial )
107
106
return d
0 commit comments