(t *testing.T)
| 612 | } |
| 613 | |
| 614 | func TestIsUnknownGpuResourceType(t *testing.T) { |
| 615 | g := gomega.NewGomegaWithT(t) |
| 616 | |
| 617 | scenarios := map[string]struct { |
| 618 | resources corev1.ResourceRequirements |
| 619 | annotations map[string]string |
| 620 | expectedUnknown bool |
| 621 | }{ |
| 622 | "OnlyBasicResources": { |
| 623 | resources: corev1.ResourceRequirements{ |
| 624 | Limits: corev1.ResourceList{ |
| 625 | corev1.ResourceCPU: resource.MustParse("1"), |
| 626 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 627 | }, |
| 628 | Requests: corev1.ResourceList{ |
| 629 | corev1.ResourceCPU: resource.MustParse("1"), |
| 630 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 631 | }, |
| 632 | }, |
| 633 | annotations: map[string]string{}, |
| 634 | expectedUnknown: false, |
| 635 | }, |
| 636 | "ValidGpuResource": { |
| 637 | resources: corev1.ResourceRequirements{ |
| 638 | Limits: corev1.ResourceList{ |
| 639 | corev1.ResourceCPU: resource.MustParse("1"), |
| 640 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 641 | corev1.ResourceName("nvidia.com/gpu"): resource.MustParse("1"), |
| 642 | }, |
| 643 | Requests: corev1.ResourceList{ |
| 644 | corev1.ResourceCPU: resource.MustParse("1"), |
| 645 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 646 | corev1.ResourceName("nvidia.com/gpu"): resource.MustParse("1"), |
| 647 | }, |
| 648 | }, |
| 649 | annotations: map[string]string{}, |
| 650 | expectedUnknown: false, |
| 651 | }, |
| 652 | "UnknownGpuResource": { |
| 653 | resources: corev1.ResourceRequirements{ |
| 654 | Limits: corev1.ResourceList{ |
| 655 | corev1.ResourceCPU: resource.MustParse("1"), |
| 656 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 657 | corev1.ResourceName("unknown.com/gpu"): resource.MustParse("1"), |
| 658 | }, |
| 659 | Requests: corev1.ResourceList{ |
| 660 | corev1.ResourceCPU: resource.MustParse("1"), |
| 661 | corev1.ResourceMemory: resource.MustParse("1Gi"), |
| 662 | corev1.ResourceName("unknown.com/gpu"): resource.MustParse("1"), |
| 663 | }, |
| 664 | }, |
| 665 | annotations: map[string]string{}, |
| 666 | expectedUnknown: true, |
| 667 | }, |
| 668 | "UnknownGpuResourceWithAnnotation": { |
| 669 | resources: corev1.ResourceRequirements{ |
| 670 | Limits: corev1.ResourceList{ |
| 671 | corev1.ResourceCPU: resource.MustParse("1"), |
nothing calls this directly
no test coverage detected