From 5c952d0c5ac4dbbb6e8366cd1bbb56023812fddb Mon Sep 17 00:00:00 2001
From: Matthew Tsai <matthewtsai@google.com>
Date: Wed, 13 Jul 2022 00:28:27 +0000
Subject: [PATCH] Resolved all outstanding nits.

---
 direct/tpm2/certify_test.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/direct/tpm2/certify_test.go b/direct/tpm2/certify_test.go
index 9613ff27..b1dee2ca 100644
--- a/direct/tpm2/certify_test.go
+++ b/direct/tpm2/certify_test.go
@@ -88,7 +88,7 @@ func TestCertify(t *testing.T) {
 	defer flushContextSigner.Execute(thetpm)
 
 	createPrimarySubject := CreatePrimary{
-		PrimaryHandle: tpm.RHPlatform,
+		PrimaryHandle: tpm.RHOwner,
 		InSensitive: tpm2b.SensitiveCreate{
 			Sensitive: tpms.SensitiveCreate{
 				UserAuth: tpm2b.Auth{
@@ -99,6 +99,12 @@ func TestCertify(t *testing.T) {
 		InPublic:    public,
 		CreationPCR: pcrSelection,
 	}
+	uniqueIdentifier := tpmu.PublicID{
+		KeyedHash: &tpm2b.Digest{
+			Buffer: decodeHex(t, "ed4fe8e2bff97665e7bfbe27c2365d07a9be91dd92d997cd91cc706b6074eb08"),
+		},
+	}
+	createPrimarySubject.InPublic.PublicArea.Unique = uniqueIdentifier
 	rspSubject, err := createPrimarySubject.Execute(thetpm)
 	if err != nil {
 		t.Fatalf("Failed to create primary: %v", err)
@@ -118,7 +124,7 @@ func TestCertify(t *testing.T) {
 			Auth:   PasswordAuth(Auth),
 		},
 		QualifyingData: tpm2b.Data{
-			Buffer: nil,
+			Buffer: make([]byte, 16),
 		},
 		InScheme: tpmt.SigScheme{
 			Scheme: tpm.AlgNull,