def curvezmq_client(): with get_server_auth_files() as auth_files: curve_dir, server_key_id, server_public_key, server_private_key = auth_files client_key_id = "id_client_{}_curve".format(socket.gethostname()) key_generator = KeyGenerator(destination_dir=curve_dir, key_id=client_key_id) key_generator.generate() client_public_key = os.path.join(curve_dir, "{}.key".format(client_key_id)) client_private_key = os.path.join( curve_dir, "{}.key_secret".format(client_key_id)) shutil.copyfile( client_public_key, os.path.join(curve_dir, 'authorized_clients', "{}.key".format(client_key_id))) auth_config = {sqlite3.SQLITE_OK: {sqlite3.SQLITE_DROP_TABLE}} server = SQLiteServer(bind_address="tcp://127.0.0.1:5002", use_encryption=True, curve_dir=curve_dir, server_curve_id=server_key_id, auth_config=auth_config, database=":memory:") client = SQLiteClient(connect_address="tcp://127.0.0.1:5002", server_curve_id=server_key_id, client_curve_id=client_key_id, curve_dir=curve_dir, use_encryption=True) server.start() LOG.info("Started Test SQLiteServer") yield client os.kill(server.pid, signal.SIGINT) server.join() client.shutdown()
def plain_client(): auth_config = { sqlite3.SQLITE_OK: { sqlite3.SQLITE_DROP_TABLE } } server = SQLiteServer(bind_address="tcp://127.0.0.1:5003", database=":memory:", auth_config=auth_config) # server.daemon = True client = SQLiteClient(connect_address="tcp://127.0.0.1:5003") server.start() # server.join() LOG.info("Started Test SQLiteServer") yield client if platform.system().lower() == 'windows': os.system("taskkill /F /pid "+str(server.pid)) else: os.kill(server.pid, signal.SIGINT) server.join() client.cleanup()
def plain_client(): auth_config = {sqlite3.SQLITE_OK: {sqlite3.SQLITE_DROP_TABLE}} server = SQLiteServer(bind_address="tcp://127.0.0.1:5003", database=":memory:", auth_config=auth_config) client = SQLiteClient(connect_address="tcp://127.0.0.1:5003") server.start() LOG.info("Started Test SQLiteServer") yield client os.kill(server.pid, signal.SIGINT) server.join() client.shutdown()
def plain_client(): auth_config = {sqlite3.SQLITE_OK: {sqlite3.SQLITE_DROP_TABLE}} with tempfile.TemporaryDirectory() as base_dir: main_db_file = os.path.join(base_dir, 'main.db') backup_db_file = os.path.join(base_dir, 'backup.db') if (sys.version_info.major == 3 and sys.version_info.minor >= 7): server = SQLiteServer(bind_address="tcp://127.0.0.1:5003", database=main_db_file, auth_config=auth_config, backup_database=backup_db_file, backup_interval=1) else: server = SQLiteServer(bind_address="tcp://127.0.0.1:5003", database=main_db_file, auth_config=auth_config) client = SQLiteClient(connect_address="tcp://127.0.0.1:5003") event = backup_event(client=client, backup_database=backup_db_file) server.start() LOG.info("Started Test SQLiteServer") yield event if platform.system().lower() == 'windows': os.system("taskkill /F /pid " + str(server.pid)) else: os.kill(server.pid, signal.SIGINT) server.join() client.cleanup()
def error_client(): LOG.info("Initializing SQLite Client") client = SQLiteClient(connect_address="tcp://127.0.0.1:5004") yield client LOG.info("Shutting down SQLite Client") client.cleanup()