@@ -94,99 +94,5 @@ def testRegisterSpiDifferentClasses(self):
9494 self .assertEqual ([config1 ], self .registry .all_api_configs ())
9595
9696
97- class BackedServiceImplTest (unittest .TestCase ):
98-
99- def setUp (self ):
100- self .service = api_backend_service .BackendServiceImpl (
101- api_backend_service .ApiConfigRegistry (), '1' )
102-
103- def testGetApiConfigsWithEmptyRequest (self ):
104- request = api_backend .GetApiConfigsRequest ()
105- self .assertEqual ([], self .service .getApiConfigs (request ).items )
106-
107- def testGetApiConfigsWithCorrectRevision (self ):
108- # TODO: there currently exists a bug in protorpc where non-unicode strings
109- # aren't validated correctly and so their values aren't set correctly.
110- # Remove 'u' this once that's fixed. This shouldn't affect production.
111- request = api_backend .GetApiConfigsRequest (appRevision = u'1' )
112- self .assertEqual ([], self .service .getApiConfigs (request ).items )
113-
114- def testGetApiConfigsWithIncorrectRevision (self ):
115- # TODO: there currently exists a bug in protorpc where non-unicode strings
116- # aren't validated correctly and so their values aren't set correctly.
117- # Remove 'u' this once that's fixed. This shouldn't affect production.
118- request = api_backend .GetApiConfigsRequest (appRevision = u'2' )
119- self .assertRaises (
120- api_exceptions .BadRequestException , self .service .getApiConfigs , request )
121-
122- # pylint: disable=g-bad-name
123- def verifyLogLevels (self , levels ):
124- Level = api_backend .LogMessagesRequest .LogMessage .Level
125- message = u'Test message.'
126- logger_name = api_backend_service .__name__
127-
128- with mock .patch ('logging.getLogger' ) as mock_getLogger :
129- log = mock_getLogger .return_value
130-
131- requestMessages = []
132- for level in levels :
133- if level :
134- requestMessage = api_backend .LogMessagesRequest .LogMessage (
135- level = getattr (Level , level ), message = message )
136- else :
137- requestMessage = api_backend .LogMessagesRequest .LogMessage (
138- message = message )
139- requestMessages .append (requestMessage )
140-
141- request = api_backend .LogMessagesRequest (messages = requestMessages )
142- self .service .logMessages (request )
143-
144- mock_getLogger .assert_called_once_with (logger_name )
145- mock_calls = []
146- for i , level in enumerate (levels ):
147- if level is None :
148- level = 'info'
149- levelno = getattr (logging , level .upper ())
150- # We can't assert equality of LogRecords because that's not
151- # supported. So instead we pull out the actual LogRecord
152- # objects and check values.
153- mock_call = log .handle .call_args_list [i ]
154- actual_record = mock_call [0 ][0 ] # first object in positional args
155- assert actual_record .name == logger_name
156- assert actual_record .levelno == levelno
157- assert actual_record .pathname == ''
158- assert actual_record .lineno == ''
159- assert actual_record .msg == message
160- assert actual_record .args is None
161- assert actual_record .exc_info is None
162-
163- def testLogMessagesUnspecifiedLevel (self ):
164- self .verifyLogLevels ([None ])
165-
166- def testLogMessagesDebug (self ):
167- self .verifyLogLevels (['debug' ])
168-
169- def testLogMessagesInfo (self ):
170- self .verifyLogLevels (['info' ])
171-
172- def testLogMessagesWarning (self ):
173- self .verifyLogLevels (['warning' ])
174-
175- def testLogMessagesError (self ):
176- self .verifyLogLevels (['error' ])
177-
178- def testLogMessagesCritical (self ):
179- self .verifyLogLevels (['critical' ])
180-
181- def testLogMessagesAll (self ):
182- self .verifyLogLevels ([None , 'debug' , 'info' , 'warning' , 'error' ,
183- 'critical' ])
184-
185- def testLogMessagesRandom (self ):
186- self .verifyLogLevels (['info' , 'debug' , 'info' , 'info' , 'warning' , 'info' ,
187- 'error' , 'error' , None , 'info' , None , None ,
188- 'critical' , 'critical' , 'info' , 'info' , None ])
189-
190-
19197if __name__ == '__main__' :
19298 unittest .main ()
0 commit comments