Add additional parameters to update and finish messages.
Change-Id: I3552d9c42bb781f80e7475edf295c1eb07a50b17
diff --git a/google_keymaster_messages_test.cpp b/google_keymaster_messages_test.cpp
index 01fcf7b..60c5f08 100644
--- a/google_keymaster_messages_test.cpp
+++ b/google_keymaster_messages_test.cpp
@@ -256,7 +256,17 @@
msg.op_handle = 0xDEADBEEF;
msg.input.Reinitialize("foo", 3);
- UniquePtr<UpdateOperationRequest> deserialized(round_trip(ver, msg, 15));
+ UniquePtr<UpdateOperationRequest> deserialized;
+ switch(ver) {
+ case 0:
+ deserialized.reset(round_trip(ver, msg, 15));
+ break;
+ case 1:
+ deserialized.reset(round_trip(ver, msg, 27));
+ break;
+ default:
+ FAIL();
+ }
EXPECT_EQ(3U, deserialized->input.available_read());
EXPECT_EQ(0, memcmp(deserialized->input.peek_read(), "foo", 3));
}
@@ -303,7 +313,17 @@
msg.op_handle = 0xDEADBEEF;
msg.signature.Reinitialize("bar", 3);
- UniquePtr<FinishOperationRequest> deserialized(round_trip(ver, msg, 15));
+ UniquePtr<FinishOperationRequest> deserialized;
+ switch(ver) {
+ case 0:
+ deserialized.reset(round_trip(ver, msg, 15));
+ break;
+ case 1:
+ deserialized.reset(round_trip(ver, msg, 27));
+ break;
+ default:
+ FAIL();
+ }
EXPECT_EQ(0xDEADBEEF, deserialized->op_handle);
EXPECT_EQ(3U, deserialized->signature.available_read());
EXPECT_EQ(0, memcmp(deserialized->signature.peek_read(), "bar", 3));