trusttier: add "identity" conversions to ToTrustTier

Allow TrustTier and *TrustTier as valid input types for ToTrustTier(). This is something that already exists for ToTrustClaim and should have been implemented for ToTrustTier as well. Signed-off-by: Sergei Trofimov <[email protected]>
veraison · Sep 22, 2023 · 74b034b · 74b034b
1 parent e82a194
commit 74b034b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/trusttier.go b/trusttier.go
@@ -152,6 +152,10 @@ func ToTrustTier(v interface{}) (*TrustTier, error) {
 				err = fmt.Errorf("not a valid TrustTier value: %v (%d)", t, int(i))
 			}
 		}
+	case TrustTier:
+		tier = t
+	case *TrustTier:
+		tier = *t
 	default:
 		err = fmt.Errorf("cannot convert %v (type %T) to TrustTier", t, t)
 	}

diff --git a/trusttier_test.go b/trusttier_test.go
@@ -163,4 +163,13 @@ func TestTrustTier_ToTrustTier(t *testing.T) {
 	tt, err = ToTrustTier(UnrecognizedHardwareClaim)
 	require.NoError(t, err)
 	assert.Equal(t, TrustTierContraindicated, *tt)
+
+	tt, err = ToTrustTier(TrustTierContraindicated)
+	require.NoError(t, err)
+	assert.Equal(t, TrustTierContraindicated, *tt)
+
+	taff := TrustTierAffirming
+	tt, err = ToTrustTier(&taff)
+	require.NoError(t, err)
+	assert.Equal(t, TrustTierAffirming, *tt)
 }