test_brandt nearly done

1 files changed, 8 insertions, 2 deletions

diff --git a/test_crypto.c b/test_crypto.c
index 2d1b50a..2a04c1d 100644
--- a/test_crypto.c
+++ b/test_crypto.c
@@ -209,11 +209,14 @@ test_setup_auction_data ()
 		{ \
 			handler_prep[type][oc][index] (&ad[i]); \
 			bufs[i] = handler_out[type][oc][index] (&ad[i], &lens[i]); \
-			CHECK (bufs[i], "failed to gen keyshare"); \
+			CHECK (bufs[i], "failed to gen message buffer"); \
 		} \
  \
 		for (uint16_t i = 0; i < bidders; i++) \
 		{ \
+			/* enable seller mode for receiving decryption messages */ \
+			if (msg_decrypt == index) \
+				ad[i].seller_mode = 1; \
 			for (uint16_t s = 0; s < bidders; s++) \
 			{ \
 				if (s == i) \
@@ -224,8 +227,11 @@ test_setup_auction_data ()
 				                                    lens[s] - \
 				                                    sizeof (struct msg_head), \
 				                                    s), \
-				       "failed checking keyshare"); \
+				       "failed to parse message buffer"); \
 			} \
+			/* disable seller mode again */ \
+			if (msg_decrypt == index) \
+				ad[i].seller_mode = 0; \
 		} \
  \
 		for (uint16_t i = 0; i < bidders; i++) \