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()
Exemple #2
0
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()
    
Exemple #3
0
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()
Exemple #4
0
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()
Exemple #5
0
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()