18
18
from logging import WARNING
19
19
from unittest .mock import patch
20
20
21
- from opentelemetry .sdk import logs
22
- from opentelemetry .sdk .environment_variables import (
23
- OTEL_PYTHON_LOG_EMITTER_PROVIDER ,
24
- )
25
- from opentelemetry .sdk .logs import (
21
+ from opentelemetry .sdk import _logs
22
+ from opentelemetry .sdk ._logs import (
26
23
LogEmitterProvider ,
27
24
get_log_emitter_provider ,
28
25
set_log_emitter_provider ,
29
26
)
27
+ from opentelemetry .sdk .environment_variables import (
28
+ OTEL_PYTHON_LOG_EMITTER_PROVIDER ,
29
+ )
30
30
31
31
32
32
class TestGlobals (unittest .TestCase ):
33
33
def tearDown (self ):
34
- reload (logs )
34
+ reload (_logs )
35
35
36
36
def check_override_not_allowed (self ):
37
37
"""set_log_emitter_provider should throw a warning when overridden"""
@@ -42,34 +42,34 @@ def check_override_not_allowed(self):
42
42
test .output ,
43
43
[
44
44
(
45
- "WARNING:opentelemetry.sdk.logs :Overriding of current "
45
+ "WARNING:opentelemetry.sdk._logs :Overriding of current "
46
46
"LogEmitterProvider is not allowed"
47
47
)
48
48
],
49
49
)
50
50
self .assertIs (provider , get_log_emitter_provider ())
51
51
52
52
def test_set_tracer_provider (self ):
53
- reload (logs )
53
+ reload (_logs )
54
54
provider = LogEmitterProvider ()
55
55
set_log_emitter_provider (provider )
56
56
retrieved_provider = get_log_emitter_provider ()
57
57
self .assertEqual (provider , retrieved_provider )
58
58
59
59
def test_tracer_provider_override_warning (self ):
60
- reload (logs )
60
+ reload (_logs )
61
61
self .check_override_not_allowed ()
62
62
63
63
@patch .dict (
64
64
"os.environ" ,
65
65
{OTEL_PYTHON_LOG_EMITTER_PROVIDER : "sdk_log_emitter_provider" },
66
66
)
67
67
def test_sdk_log_emitter_provider (self ):
68
- reload (logs )
68
+ reload (_logs )
69
69
self .check_override_not_allowed ()
70
70
71
71
@patch .dict ("os.environ" , {OTEL_PYTHON_LOG_EMITTER_PROVIDER : "unknown" })
72
72
def test_unknown_log_emitter_provider (self ):
73
- reload (logs )
73
+ reload (_logs )
74
74
with self .assertRaises (Exception ):
75
75
get_log_emitter_provider ()
0 commit comments