test: test cases for the new wrapper

helpers_test.go

@@ -0,0 +1,59 @@
+package libbpfgo
+
+import (
+	"testing"
+
+	"github.com/aquasecurity/libbpfgo/helpers"
+)
+
+func TestFuncSupportbyType(t *testing.T) {
+	tt := []struct {
+		progType  BPFProgType
+		funcId    helpers.BPFFunc
+		supported bool
+	}{
+		{
+			progType:  BPFProgTypeKprobe,
+			funcId:    helpers.BPFFuncMapLookupElem,
+			supported: true,
+		},
+		{
+			progType:  BPFProgTypeKprobe,
+			funcId:    helpers.BPFFuncLoop,
+			supported: true,
+		},
+		{
+			progType:  BPFProgTypeKprobe,
+			funcId:    helpers.BPFFuncKtimeGetNs,
+			supported: true,
+		},
+		{
+			progType:  BPFProgTypeKprobe,
+			funcId:    helpers.BPFFuncSysBpf,
+			supported: false,
+		},
+		{
+			progType:  BPFProgTypeLsm,
+			funcId:    helpers.BPFFuncGetCurrentCgroupId,
+			supported: false,
+		},
+		{
+			progType:  BPFProgTypeSkLookup,
+			funcId:    helpers.BPFFuncGetCurrentCgroupId,
+			supported: true,
+		},
+		{
+			progType:  BPFProgTypeKprobe,
+			funcId:    helpers.BPFFuncSockMapUpdate,
+			supported: false,
+		},
+	}
+
+	for _, tc := range tt {
+		support, _ := BPFFuncTypeIsSupported(tc.progType, tc.funcId)
+
+		if support != tc.supported {
+			t.Errorf("expected %v, got %v", tc.supported, support)
+		}
+	}
+}