SIENTIAPDE-988
Refactor OpcManager initialization and unsubscribe method; enhance logging - Removed the `initialize_from_config` method from `OpcManager` class and adjusted the constructor to handle configuration directly. - Improved the `unsubscribe` method to include detailed logging for non-existent subscriptions. - Updated tests in `test_opc_manager.py` to reflect changes in the `OpcManager` class. - Commented out Docker-related fixtures in `conftest.py` for potential future use. - Enhanced test coverage in `test_single_node.py` and `test_data_manager.py` with additional scenarios and assertions. - Introduced new methods in `Ingestor` and `IngestorManager` classes to manage server subscriptions and leases more effectively. - Added new tests for `Ingestor` class to validate initialization and slot management logic. - Implemented logging improvements across various classes to ensure better traceability of actions and errors.
This commit is contained in:
@@ -1,47 +1,47 @@
|
||||
import subprocess
|
||||
from time import sleep
|
||||
from typing import Generator
|
||||
import uuid
|
||||
from kafka import KafkaConsumer
|
||||
import pytest
|
||||
from redis import Redis
|
||||
# import subprocess
|
||||
# from time import sleep
|
||||
# from typing import Generator
|
||||
# import uuid
|
||||
# from kafka import KafkaConsumer
|
||||
# import pytest
|
||||
# from redis import Redis
|
||||
|
||||
|
||||
@pytest.fixture(scope="session", autouse=True)
|
||||
def docker_compose():
|
||||
"""Sobe os containers antes dos testes e derruba depois."""
|
||||
print("\n🚀 Subindo Docker Compose...")
|
||||
subprocess.run(["docker", "compose", "up", "-d"], check=True)
|
||||
# @pytest.fixture(scope="session", autouse=True)
|
||||
# def docker_compose():
|
||||
# """Sobe os containers antes dos testes e derruba depois."""
|
||||
# print("\n🚀 Subindo Docker Compose...")
|
||||
# subprocess.run(["docker", "compose", "up", "-d"], check=True)
|
||||
|
||||
print("⏳ Aguardando containers ficarem prontos...")
|
||||
sleep(15) # ajuste conforme necessário
|
||||
# print("⏳ Aguardando containers ficarem prontos...")
|
||||
# sleep(15) # ajuste conforme necessário
|
||||
|
||||
yield # os testes rodam aqui
|
||||
# yield # os testes rodam aqui
|
||||
|
||||
print("\n🧹 Derrubando Docker Compose...")
|
||||
subprocess.run(["docker", "compose", "down"], check=True)
|
||||
# print("\n🧹 Derrubando Docker Compose...")
|
||||
# subprocess.run(["docker", "compose", "down"], check=True)
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def redis_client():
|
||||
redis = Redis(host="localhost", port=6379, decode_responses=True)
|
||||
redis.flushdb()
|
||||
# @pytest.fixture()
|
||||
# def redis_client():
|
||||
# redis = Redis(host="localhost", port=6379, decode_responses=True)
|
||||
# redis.flushdb()
|
||||
|
||||
yield redis
|
||||
# yield redis
|
||||
|
||||
# Limpa o banco de dados após os testes
|
||||
redis.flushdb()
|
||||
redis.close()
|
||||
# # Limpa o banco de dados após os testes
|
||||
# redis.flushdb()
|
||||
# redis.close()
|
||||
|
||||
|
||||
def kafka_searcher(topic) -> Generator[KafkaConsumer, None, None]:
|
||||
consumer = KafkaConsumer(
|
||||
topic,
|
||||
bootstrap_servers="localhost:9092",
|
||||
group_id=f"test-group-{uuid.uuid4()}",
|
||||
auto_offset_reset="earliest", # Começa a consumir apenas mensagens novas
|
||||
enable_auto_commit=True,
|
||||
)
|
||||
# def kafka_searcher(topic) -> Generator[KafkaConsumer, None, None]:
|
||||
# consumer = KafkaConsumer(
|
||||
# topic,
|
||||
# bootstrap_servers="localhost:9092",
|
||||
# group_id=f"test-group-{uuid.uuid4()}",
|
||||
# auto_offset_reset="earliest", # Começa a consumir apenas mensagens novas
|
||||
# enable_auto_commit=True,
|
||||
# )
|
||||
|
||||
yield consumer
|
||||
consumer.close()
|
||||
# yield consumer
|
||||
# consumer.close()
|
||||
|
||||
Reference in New Issue
Block a user