fixup! keydata*test.go: fix/add tests

canonical · Jan 16, 2024 · d891631 · d891631
1 parent 1c0fc52
commit d891631
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keydata_test.go b/keydata_test.go
@@ -267,11 +267,11 @@ func (s *keyDataTestBase) TearDownSuite(c *C) {
 }
 
 func (s *keyDataTestBase) newPrimaryKey(c *C, sz1 int) PrimaryKey {
-	primaryKey := make([]byte, sz1)
+	primaryKey := make(PrimaryKey, sz1)
 	_, err := rand.Read(primaryKey)
 	c.Assert(err, IsNil)
 
-	return PrimaryKey(primaryKey)
+	return primaryKey
 }
 
 func (s *keyDataTestBase) mockProtectKeys(c *C, primaryKey PrimaryKey, kdfAlg crypto.Hash, modelAuthHash crypto.Hash) (out *KeyParams, unlockKey DiskUnlockKey) {