You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.0 KiB
106 lines
2.0 KiB
6 months ago
|
package wakeup
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
type procfsMock struct{}
|
||
|
|
|
||
|
|
func (p *procfsMock) ACPIWakeup() (io.ReadCloser, error) {
|
||
|
|
const testFileRaw = `Device S-state Status Sysfs node
|
||
|
|
PEG0 S3 *disabled
|
||
|
|
EC S4 *disabled platform:PNP0C09:00
|
||
|
|
HDEF S3 *disabled pci:0000:00:1b.0
|
||
|
|
RP01 S3 *enabled pci:0000:00:1c.0
|
||
|
|
RP02 S3 *enabled pci:0000:00:1c.1
|
||
|
|
RP03 S3 *enabled pci:0000:00:1c.2
|
||
|
|
ARPT S4 *enabled pci:0000:03:00.0
|
||
|
|
RP05 S3 *enabled pci:0000:00:1c.4
|
||
|
|
RP06 S3 *enabled pci:0000:00:1c.5
|
||
|
|
SPIT S3 *disabled spi:spi-APP000D:00
|
||
|
|
XHC1 S3 *disabled pci:0000:00:14.0
|
||
|
|
ADP1 S4 *disabled platform:ACPI0003:00
|
||
|
|
LID0 S4 *disabled platform:PNP0C0D:00
|
||
|
|
`
|
||
|
|
|
||
|
|
return io.NopCloser(strings.NewReader(testFileRaw)), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *procfsMock) ACPIWakeupWrite() (io.WriteCloser, error) {
|
||
|
|
return nil, fmt.Errorf("not implemented")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSomething(t *testing.T) {
|
||
|
|
|
||
|
|
tt := []struct {
|
||
|
|
name string
|
||
|
|
wantStatus Status
|
||
|
|
wantDevices []string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Read and parse successfully",
|
||
|
|
wantStatus: StatusEnabled,
|
||
|
|
wantDevices: []string{
|
||
|
|
"RP01",
|
||
|
|
"RP02",
|
||
|
|
"RP03",
|
||
|
|
"RP05",
|
||
|
|
"RP06",
|
||
|
|
"ARPT",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Read and parse disabled successfully",
|
||
|
|
wantStatus: StatusDisabled,
|
||
|
|
wantDevices: []string{
|
||
|
|
"PEG0",
|
||
|
|
"EC",
|
||
|
|
"HDEF",
|
||
|
|
"SPIT",
|
||
|
|
"XHC1",
|
||
|
|
"ADP1",
|
||
|
|
"LID0",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Read and parse all successfully",
|
||
|
|
wantStatus: StatusAll,
|
||
|
|
wantDevices: []string{
|
||
|
|
"RP01",
|
||
|
|
"RP02",
|
||
|
|
"RP03",
|
||
|
|
"RP05",
|
||
|
|
"RP06",
|
||
|
|
"ARPT",
|
||
|
|
"PEG0",
|
||
|
|
"EC",
|
||
|
|
"HDEF",
|
||
|
|
"SPIT",
|
||
|
|
"XHC1",
|
||
|
|
"ADP1",
|
||
|
|
"LID0",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tc := range tt {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
c := NewWakeupController()
|
||
|
|
c.ProcfsProvider = &procfsMock{}
|
||
|
|
|
||
|
|
devices, err := c.GetWakeupDevices(tc.wantStatus)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
assert.Len(t, devices, len(tc.wantDevices))
|
||
|
|
for i, wantDevice := range tc.wantDevices {
|
||
|
|
assert.Contains(t, devices, wantDevice, "device %d not found", i)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|