1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright 2023 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package report
import (
"bytes"
"strings"
"testing"
"github.com/google/go-configfs-tsm/configfs/configfsi"
"github.com/google/go-configfs-tsm/configfs/faketsm"
)
func TestGet(t *testing.T) {
c := &faketsm.Client{Subsystems: map[string]configfsi.Client{"report": faketsm.ReportV7(0)}}
req := &Request{
InBlob: []byte("lessthan64bytesok"),
GetAuxBlob: true,
}
resp, err := Get(c, req)
if err != nil {
t.Fatalf("Get(%+v) = %+v, %v, want nil", req, resp, err)
}
wantOut := "privlevel: 0\ninblob: 6c6573737468616e363462797465736f6b"
if !bytes.Equal(resp.OutBlob, []byte(wantOut)) {
t.Errorf("OutBlob %v is not %v", string(resp.OutBlob), wantOut)
}
wantProvider := "fake\n"
if resp.Provider != wantProvider {
t.Errorf("provider = %q, want %q", resp.Provider, wantProvider)
}
if !bytes.Equal(resp.AuxBlob, []byte(`auxblob`)) {
t.Errorf("auxblob = %v, want %v", resp.AuxBlob, []byte(`auxblob`))
}
}
func TestGetErr(t *testing.T) {
tcs := []struct {
name string
req *Request
floor uint
wantErr string
}{
{
name: "inblob too big",
req: &Request{
InBlob: make([]byte, 4096),
},
wantErr: "invalid argument",
},
{
name: "privlevel too high",
req: &Request{
InBlob: make([]byte, 64),
Privilege: &Privilege{Level: 300},
},
wantErr: "privlevel must be 0-3",
},
{
name: "missing inblob",
req: &Request{},
wantErr: "invalid argument",
},
{
name: "privlevel too low",
req: &Request{
InBlob: make([]byte, 64),
Privilege: &Privilege{Level: 0},
},
floor: 1,
wantErr: "privlevel 0 cannot be less than 1",
},
}
for _, tc := range tcs {
t.Run(tc.name, func(t *testing.T) {
c := &faketsm.Client{Subsystems: map[string]configfsi.Client{"report": faketsm.ReportV7(tc.floor)}}
resp, err := Get(c, tc.req)
if err == nil || !strings.Contains(err.Error(), tc.wantErr) {
t.Fatalf("Get(%+v) = %+v, %v, want %q", tc.req, resp, err, tc.wantErr)
}
})
}
}
|
