Skip to content

Commit a175a40

Browse files
committed
This should fix build error
1 parent 6701bfe commit a175a40

File tree

2 files changed

+25
-31
lines changed

2 files changed

+25
-31
lines changed

azure-eventhubs/conftest.py

+25-1
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import logging
1010
import sys
1111
import uuid
12+
from logging.handlers import RotatingFileHandler
1213

1314
# Ignore async tests for Python < 3.5
1415
collect_ignore = []
@@ -17,7 +18,8 @@
1718
collect_ignore.append("features")
1819
collect_ignore.append("examples/async_examples")
1920
else:
20-
from tests.asynctests import MockEventProcessor
21+
sys.path.append("tests")
22+
from asynctests import MockEventProcessor
2123
from azure.eventprocessorhost import EventProcessorHost
2224
from azure.eventprocessorhost import EventHubPartitionPump
2325
from azure.eventprocessorhost import AzureStorageCheckpointLeaseManager
@@ -32,6 +34,28 @@
3234
from azure.eventhub import EventHubClient, Receiver, Offset
3335

3436

37+
def get_logger(filename, level=logging.INFO):
38+
azure_logger = logging.getLogger("azure.eventhub")
39+
azure_logger.setLevel(level)
40+
uamqp_logger = logging.getLogger("uamqp")
41+
uamqp_logger.setLevel(logging.INFO)
42+
43+
formatter = logging.Formatter('%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
44+
console_handler = logging.StreamHandler(stream=sys.stdout)
45+
console_handler.setFormatter(formatter)
46+
if not azure_logger.handlers:
47+
azure_logger.addHandler(console_handler)
48+
if not uamqp_logger.handlers:
49+
uamqp_logger.addHandler(console_handler)
50+
51+
if filename:
52+
file_handler = RotatingFileHandler(filename, maxBytes=5*1024*1024, backupCount=2)
53+
file_handler.setFormatter(formatter)
54+
azure_logger.addHandler(file_handler)
55+
56+
return azure_logger
57+
58+
3559
log = get_logger(None, logging.DEBUG)
3660

3761

azure-eventhubs/tests/__init__.py

-30
This file was deleted.

0 commit comments

Comments
 (0)