from tonga.stores.store_builder.store_builder import StoreBuilder
# StoreRecord import
from tonga.models.store_record.store_record import StoreRecord
from tonga.models.store_record.store_record_handler import StoreRecordHandler
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
t_loop = uvloop.new_event_loop()
# Store test
test_local_memory_store = LocalStoreMemory(name='local_store_memory_test')
test_global_memory_store = GlobalStoreMemory(name='global_store_memory_test')
# Avro Serializer test
test_serializer = AvroSerializer(BASE_DIR + '/misc/schemas')
test_serializer_local_memory_store = LocalStoreMemory(
name='local_store_memory_serializer_test')
test_serializer_global_memory_store = GlobalStoreMemory(
name='global_store_memory_serializer_test')
# StatefulsetPartitionAssignor test
assignor_kafka_client = KafkaClient(bootstrap_servers='localhost:9092',
client_id='test_client')
assignor_cluster_metadata = ClusterMetadata(bootstrap_servers='localhost:9092')
# StoreBuilder test
store_builder_serializer = AvroSerializer(BASE_DIR + '/misc/schemas')
admin_client = KafkaAdminClient(bootstrap_servers='localhost:9092',
client_id='test_store_builder')
# Register bartender app info
bartender_app['instance'] = cur_instance
bartender_app['nb_replica'] = nb_replica
# Registers logger
bartender_app['logger'] = setup_logger()
bartender_app['logger'].info(
f'Bartender current instance : {cur_instance}')
# Creates & registers event loop
bartender_app['loop'] = uvloop.new_event_loop()
asyncio.set_event_loop(bartender_app['loop'])
bartender_app['serializer'] = AvroSerializer(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'examples/coffee_bar/avro_schemas'))
# Creates & register KafkaProducer
bartender_app['transactional_producer'] = KafkaProducer(
name=f'bartender-{cur_instance}',
bootstrap_servers='localhost:9092',
client_id=f'bartender-{cur_instance}',
serializer=bartender_app['serializer'],
loop=bartender_app['loop'],
partitioner=KeyPartitioner(),
acks='all',
transactional_id=f'bartender')
# Initializes bartender handlers
coffee_finished_handler = CoffeeFinishedHandler()
def test_register_more_than_once_avro_serializer():
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
with pytest.raises(AvroAlreadyRegister):
serializer = AvroSerializer(BASE_DIR + '/misc/schemas/bad')