from beaconcrypt import BeaconCryptBeacon, BeaconCryptServer
def register_beacon(
server: BeaconCryptServer,
beacon: BeaconCryptBeacon,
) -> bytes | None:
message = bytes(0xFF * 32)
phase_1 = beacon.generate_registration()
assert phase_1 is not None
reg_out = server.register_beacon(phase_1, message)
assert reg_out is not None
phase2 = beacon.process_initial_message(reg_out.serialized())
assert phase2 is not None
assert phase2 == message
return phase2
def test_register():
server = BeaconCryptServer(0, None)
server_pk = server.id_pk()
beacon = BeaconCryptBeacon(0, server_pk)
assert register_beacon(server, beacon) is not None