MCPcopy Create free account
hub / github.com/NVIDIA/gpu-operator / TestGetDriverAppName

Function TestGetDriverAppName

internal/state/driver_test.go:525–580  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

523}
524
525func TestGetDriverAppName(t *testing.T) {
526 cr := &nvidiav1alpha1.NVIDIADriver{
527 ObjectMeta: metav1.ObjectMeta{
528 UID: apitypes.UID("bfac7359-6033-45ce-88d6-53db0078526e"),
529 },
530 Spec: nvidiav1alpha1.NVIDIADriverSpec{
531 DriverType: nvidiav1alpha1.GPU,
532 },
533 }
534
535 pool := nodePool{
536 osRelease: "ubuntu",
537 osVersion: "20.04",
538 }
539 var err error
540 pool.osTag, err = getOSTag(pool.osRelease, pool.osVersion)
541 assert.NoError(t, err)
542
543 actual := getDriverAppName(cr, pool)
544 expected := "nvidia-gpu-driver-ubuntu20.04-67cc6dbb79"
545 assert.Equal(t, expected, actual)
546
547 // Modify nodePool to include kernelVersion
548 pool.kernel = "5.15.0-70-generic"
549
550 actual = getDriverAppName(cr, pool)
551 expected = "nvidia-gpu-driver-ubuntu20.04-59b779bcc5"
552 assert.Equal(t, expected, actual)
553
554 // Now set the osVersion to a really long string
555 pool.osRelease = "redhatCoreOS"
556 pool.osVersion = "4.14-414.92.202309282257"
557 pool.osTag, err = getOSTag(pool.osRelease, pool.osVersion)
558 assert.NoError(t, err)
559
560 actual = getDriverAppName(cr, pool)
561 expected = "nvidia-gpu-driver-redhatCoreOS4.14-414.92.2023092822-59b779bcc5"
562 assert.Equal(t, expected, actual)
563 assert.Equal(t, 63, len(actual))
564
565 // RockyLinux
566 pool.osRelease = "rocky"
567 pool.osVersion = "9.6"
568 pool.osTag, err = getOSTag(pool.osRelease, pool.osVersion)
569 assert.NoError(t, err)
570 actual = getDriverAppName(cr, pool)
571 assert.Equal(t, "nvidia-gpu-driver-rocky9-59b779bcc5", actual)
572
573 // RHEL10
574 pool.osRelease = "rhel"
575 pool.osVersion = "10.1"
576 pool.osTag, err = getOSTag(pool.osRelease, pool.osVersion)
577 assert.NoError(t, err)
578 actual = getDriverAppName(cr, pool)
579 assert.Equal(t, "nvidia-gpu-driver-rhel10-59b779bcc5", actual)
580}
581
582func TestGetDriverAppNameRHCOS(t *testing.T) {

Callers

nothing calls this directly

Calls 2

getOSTagFunction · 0.85
getDriverAppNameFunction · 0.85

Tested by

no test coverage detected