talm

1
// This Source Code Form is subject to the terms of the Mozilla Public
2
// License, v. 2.0. If a copy of the MPL was not distributed with this
3
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
4

5
// Package tpm2 provides TPM2.0 related functionality helpers.
6
package tpm2_test
7

8
import (
9
	"testing"
10

11
	"github.com/stretchr/testify/require"
12

13
	tpm2internal "github.com/aenix-io/talm/internal/pkg/secureboot/tpm2"
14
)
15

16
func TestGetSelection(t *testing.T) {
17
	t.Parallel()
18

19
	for _, tt := range []struct {
20
		name     string
21
		pcrs     []int
22
		expected []byte
23
	}{
24
		{
25
			name:     "empty",
26
			expected: []byte{0, 0, 0},
27
		},
28
		{
29
			name:     "1, 3, 5",
30
			pcrs:     []int{1, 3, 5},
31
			expected: []byte{42, 0, 0},
32
		},
33
		{
34
			name:     "21, 22, 23",
35
			pcrs:     []int{21, 22, 23},
36
			expected: []byte{0, 0, 0xe0},
37
		},
38
	} {
39
		t.Run(tt.name, func(t *testing.T) {
40
			t.Parallel()
41

42
			actual, err := tpm2internal.CreateSelector(tt.pcrs)
43
			require.NoError(t, err)
44

45
			require.Equal(t, tt.expected, actual)
46
		})
47
	}
48
}
49