diff --git a/features/steps/shamir_steps.py b/features/steps/shamir_steps.py index ff77276..498ee94 100644 --- a/features/steps/shamir_steps.py +++ b/features/steps/shamir_steps.py @@ -39,7 +39,7 @@ def operation(communicator): context.shares = [] for i in range(count): - context.shares.append(cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation)) + context.shares.append(cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation, identities=context.identities, trusted=context.trusted)) context.shares = numpy.array(context.shares) @@ -52,7 +52,7 @@ def operation(communicator, count): shares = [protocol.share(src=0, k=context.k, secret=numpy.array(5))._storage for i in range(count)] return shares - context.shares = numpy.column_stack(cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation, args=(count,))) + context.shares = numpy.column_stack(cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation, args=(count,), identities=context.identities, trusted=context.trusted)) @when(u'player {player} shamir shares {secret} with {recipients} and {senders} reveal their shares to {destinations}') @@ -68,6 +68,6 @@ def operation(communicator): share = protocol.share(src=player, k=context.k, secret=secret) return protocol.reveal(src=senders, share=share, dst=destinations) - context.results = cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation) + context.results = cicada.communicator.SocketCommunicator.run(world_size=context.players, fn=operation, identities=context.identities, trusted=context.trusted)