17
17
18
18
19
19
class TestLoggingShim (unittest .TestCase ):
20
- def test_shim_matches_logging_v2 (self ):
20
+ def test_root_shim_matches_logging_v2 (self ):
21
21
from google .cloud import logging
22
22
from google .cloud import logging_v2
23
23
@@ -27,3 +27,36 @@ def test_shim_matches_logging_v2(self):
27
27
found = getattr (logging , name )
28
28
expected = getattr (logging_v2 , name )
29
29
self .assertIs (found , expected )
30
+
31
+ def test_handler_shim_matches_logging_v2 (self ):
32
+ from google .cloud .logging import handlers
33
+ from google .cloud .logging_v2 import handlers as handler_2
34
+
35
+ self .assertEqual (handlers .__all__ , handlersv2 .__all__ )
36
+
37
+ for name in handlers .__all__ :
38
+ found = getattr (handlers , name )
39
+ expected = getattr (handlers_2 , name )
40
+ self .assertIs (found , expected )
41
+
42
+ def test_middleware_shim_matches_logging_v2 (self ):
43
+ from google .cloud .logging .handlers import middleware
44
+ from google .cloud .logging_v2 .handlers import middleware as middleware_2
45
+
46
+ self .assertEqual (middleware .__all__ , middleware_2 .__all__ )
47
+
48
+ for name in middleware .__all__ :
49
+ found = getattr (middleware , name )
50
+ expected = getattr (middleware_2 , name )
51
+ self .assertIs (found , expected )
52
+
53
+ def test_transports_shim_matches_logging_v2 (self ):
54
+ from google .cloud .logging .handlers import transports
55
+ from google .cloud .logging_v2 .handlers import transports as transports_2
56
+
57
+ self .assertEqual (middleware .__all__ , transports_2 .__all__ )
58
+
59
+ for name in transports .__all__ :
60
+ found = getattr (transports , name )
61
+ expected = getattr (transports_2 , name )
62
+ self .assertIs (found , expected )
0 commit comments