84 lines
2.8 KiB
Go
84 lines
2.8 KiB
Go
// Copyright 2023 Clayton Craft <clayton@craftyguy.net>
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package modules
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestStripExts(t *testing.T) {
|
|
tables := []struct {
|
|
in string
|
|
expected string
|
|
}{
|
|
{"/foo/bar/bazz.tar", "/foo/bar/bazz"},
|
|
{"file.tar.gz.xz.zip", "file"},
|
|
{"another_file", "another_file"},
|
|
{"a.b.c.d.e.f.g.h.i", "a"},
|
|
{"virtio_blk.ko", "virtio_blk"},
|
|
{"virtio_blk.ko ", "virtio_blk"},
|
|
}
|
|
for _, table := range tables {
|
|
out := stripExts(table.in)
|
|
if out != table.expected {
|
|
t.Errorf("Expected: %q, got: %q", table.expected, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
var testModuleDep string = `
|
|
kernel/sound/soc/codecs/snd-soc-msm8916-digital.ko:
|
|
kernel/net/sched/act_ipt.ko.xz: kernel/net/netfilter/x_tables.ko.xz
|
|
kernel/drivers/watchdog/watchdog.ko.xz:
|
|
kernel/drivers/usb/serial/ir-usb.ko.xz: kernel/drivers/usb/serial/usbserial.ko.xz
|
|
kernel/drivers/gpu/drm/scheduler/gpu-sched.ko.xz:
|
|
kernel/drivers/hid/hid-alps.ko.xz:
|
|
kernel/net/netfilter/xt_u32.ko.xz: kernel/net/netfilter/x_tables.ko.xz
|
|
kernel/net/netfilter/xt_sctp.ko.xz: kernel/net/netfilter/x_tables.ko.xz
|
|
kernel/drivers/hwmon/gl518sm.ko.xz:
|
|
kernel/drivers/watchdog/dw_wdt.ko.xz: kernel/drivers/watchdog/watchdog.ko.xz
|
|
kernel/net/bluetooth/hidp/hidp.ko.xz: kernel/net/bluetooth/bluetooth.ko.xz kernel/net/rfkill/rfkill.ko.xz kernel/crypto/ecdh_generic.ko.xz kernel/crypto/ecc.ko.xz
|
|
kernel/fs/nls/nls_iso8859-1.ko.xz:
|
|
kernel/net/vmw_vsock/vmw_vsock_virtio_transport.ko.xz: kernel/net/vmw_vsock/vmw_vsock_virtio_transport_common.ko.xz kernel/drivers/virtio/virtio.ko.xz kernel/drivers/virtio/virtio_ring.ko.xz kernel/net/vmw_vsock/vsock.ko.xz
|
|
kernel/drivers/gpu/drm/panfrost/panfrost.ko.xz: kernel/drivers/gpu/drm/scheduler/gpu-sched.ko.xz
|
|
kernel/drivers/gpu/drm/msm/msm.ko: kernel/drivers/gpu/drm/drm_kms_helper.ko
|
|
`
|
|
|
|
func TestGetModuleDeps(t *testing.T) {
|
|
tables := []struct {
|
|
in string
|
|
expected []string
|
|
}{
|
|
{"nls-iso8859-1", []string{"kernel/fs/nls/nls_iso8859-1.ko.xz"}},
|
|
{"gpu_sched", []string{"kernel/drivers/gpu/drm/scheduler/gpu-sched.ko.xz"}},
|
|
{"dw-wdt", []string{"kernel/drivers/watchdog/dw_wdt.ko.xz",
|
|
"kernel/drivers/watchdog/watchdog.ko.xz"}},
|
|
{"gl518sm", []string{"kernel/drivers/hwmon/gl518sm.ko.xz"}},
|
|
{"msm", []string{"kernel/drivers/gpu/drm/msm/msm.ko",
|
|
"kernel/drivers/gpu/drm/drm_kms_helper.ko"}},
|
|
}
|
|
for _, table := range tables {
|
|
out, err := getModuleDeps(table.in, strings.NewReader(testModuleDep))
|
|
if err != nil {
|
|
t.Errorf("unexpected error with input: %q, error: %q", table.expected, err)
|
|
}
|
|
if !stringSlicesEqual(out, table.expected) {
|
|
t.Errorf("Expected: %q, got: %q", table.expected, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func stringSlicesEqual(a []string, b []string) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for i, v := range a {
|
|
if v != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|