diff --git a/pkg/labels/labels.go b/pkg/labels/labels.go new file mode 100644 index 000000000..fafcbea38 --- /dev/null +++ b/pkg/labels/labels.go @@ -0,0 +1,13 @@ +package labels + +import ( + "github.com/rh-ecosystem-edge/kernel-module-management/internal/utils" +) + +func GetKernelModuleReadyNodeLabel(namespace, moduleName string) string { + return utils.GetKernelModuleReadyNodeLabel(namespace, moduleName) +} + +func GetDevicePluginNodeLabel(namespace, moduleName string) string { + return utils.GetDevicePluginNodeLabel(namespace, moduleName) +} diff --git a/pkg/labels/labels_test.go b/pkg/labels/labels_test.go new file mode 100644 index 000000000..6afe7cc2c --- /dev/null +++ b/pkg/labels/labels_test.go @@ -0,0 +1,18 @@ +package labels + +import ( + . "github.com/onsi/ginkgo/v2" + . "github.com/onsi/gomega" +) + +var _ = Describe("GetModuleReadyAndDevicePluginReadyLabels", func() { + It("module ready label", func() { + res := GetKernelModuleReadyNodeLabel("some-namespace", "some-module") + Expect(res).To(Equal("kmm.node.kubernetes.io/some-namespace.some-module.ready")) + }) + + It("device-plugin ready label", func() { + res := GetDevicePluginNodeLabel("some-namespace", "some-module") + Expect(res).To(Equal("kmm.node.kubernetes.io/some-namespace.some-module.device-plugin-ready")) + }) +}) diff --git a/pkg/labels/suite_test.go b/pkg/labels/suite_test.go new file mode 100644 index 000000000..f3320ffc1 --- /dev/null +++ b/pkg/labels/suite_test.go @@ -0,0 +1,23 @@ +package labels + +import ( + "testing" + + . "github.com/onsi/ginkgo/v2" + . "github.com/onsi/gomega" + "github.com/rh-ecosystem-edge/kernel-module-management/internal/test" + "k8s.io/apimachinery/pkg/runtime" +) + +var scheme *runtime.Scheme + +func TestSuite(t *testing.T) { + RegisterFailHandler(Fail) + + var err error + + scheme, err = test.TestScheme() + Expect(err).NotTo(HaveOccurred()) + + RunSpecs(t, "Job Suite") +}