1
1
import sys
2
2
3
3
from common .models import Arch
4
- import middle
5
- from middle import add_changeset_entry
4
+ import middleware
5
+ from middleware import add_changeset_entry
6
6
from models import Logging , ChangeSet , ChangeSetEntry
7
7
8
+ from django .core import management
9
+ from django .core .management import sql , color
10
+ from django .db import connection , models
8
11
from django .test import TestCase
9
12
from django .test .client import RequestFactory
10
13
from django .core .handlers .base import BaseHandler
@@ -35,23 +38,48 @@ def user_login(test_case, name, pswd, **request):
35
38
36
39
37
40
class LoggingTest (TestCase ):
41
+ def make_model (self ):
42
+ self .model = type ('DummyModel' , (models .Model ,), {
43
+ 'dummy_attr' : models .CharField (max_length = 255 ),
44
+ "__module__" : "mlogging.tests" })
45
+
46
+ self ._style = color .no_style ()
47
+
48
+ self ._cursor = connection .cursor ()
49
+ statements , pending = connection .creation .sql_create_model (self .model ,
50
+ self ._style )
51
+ for statement in statements :
52
+ self ._cursor .execute (statement )
53
+ ContentType .objects .get_for_model (self .model ).save ()
54
+
38
55
def setUp (self ):
39
56
test_user = User .
objects .
create_user (
"test_user" ,
"[email protected] " )
40
57
test_user .set_password ("test_pass" )
41
58
test_user .save ()
42
59
self .user = authenticate (username = "test_user" , password = "test_pass" )
43
60
self .rf = RequestMock ()
44
- arch = Arch .objects .create (name = "dummy arch" )
45
- arch .save ()
46
- self .l_instance = middle .LoggingMiddleware ()
61
+ self .make_model ()
62
+ #self.model.save(self.model)
63
+ dm = self .model .objects .create (dummy_attr = "some value" )
64
+
65
+ dm .save ()
66
+
67
+ self .l_instance = middleware .LoggingMiddleware ()
68
+
69
+ def tearDown (self ):
70
+ statements = connection .creation .sql_destroy_model (self .model , (),
71
+ self ._style )
72
+ for statement in statements :
73
+ self ._cursor .execute (statement )
47
74
48
- def test_middle (self ):
75
+ def test_middle_x (self ):
49
76
request = self .rf .request (REQUEST_METHOD = 'POST' )
50
77
login (request , self .user )
51
78
self .l_instance .process_request (request )
52
- arch = Arch .objects .all ()[0 ]
53
- with middle .Log (request , {"arch" : arch }):
54
- arch .name = "silly arch"
79
+ dummy_o = self .model .objects .all ()[0 ]
80
+ #print >> sys.stderr, dummy_o
81
+ with middleware .Log (request , {"dummy_o" : dummy_o }):
82
+ dummy_o .dummy_attr = "silly value"
55
83
56
84
response = lambda x : x
57
85
response .status_code = 200
@@ -69,20 +97,20 @@ def test_middle(self):
69
97
self .assertEqual (entries [0 ].changeset , changesets [0 ])
70
98
self .assertEqual (changesets [0 ].user , self .user )
71
99
self .assertEqual (entries [0 ].model ,
72
- ContentType .objects .get_for_model (Arch ))
100
+ ContentType .objects .get_for_model (self . model ))
73
101
74
102
def test_middle_manual (self ):
75
103
request = self .rf .request (REQUEST_METHOD = 'POST' )
76
104
login (request , self .user )
77
105
self .l_instance .process_request (request )
78
- arch = Arch .objects .all ()[0 ]
106
+ dummy_o = self . model .objects .all ()[0 ]
79
107
80
- old_arch = {"id" : arch .id , "name " : arch . name }
81
- arch . name = "silly arch "
82
- new_arch = {"id" : arch .id , "name " : arch . name }
108
+ old_dummy = {"id" : dummy_o .id , "dummy_attr " : dummy_o . dummy_attr }
109
+ dummy_o . dummy_attr = "new dummy value "
110
+ new_dummy = {"id" : dummy_o .id , "dummy_attr " : dummy_o . dummy_attr }
83
111
add_changeset_entry (request , "test_middle_manual" ,
84
- [ContentType .objects .get_for_model (Arch )],
85
- {"arch " : old_arch }, {"arch " : new_arch })
112
+ [ContentType .objects .get_for_model (dummy_o . _meta . model )],
113
+ {"dummy_o " : old_dummy }, {"dummy_o " : new_dummy })
86
114
87
115
response = lambda x : x
88
116
response .status_code = 200
@@ -100,4 +128,4 @@ def test_middle_manual(self):
100
128
self .assertEqual (entries [0 ].changeset , changesets [0 ])
101
129
self .assertEqual (changesets [0 ].user , self .user )
102
130
self .assertEqual (entries [0 ].model ,
103
- ContentType .objects .get_for_model (Arch ))
131
+ ContentType .objects .get_for_model (self . model ))
0 commit comments