@@ -535,63 +535,55 @@ func TestDefaultNodePoolGCPImage(t *testing.T) {
535535 testCases := []struct {
536536 name string
537537 arch string
538- releaseImage * releaseinfo. ReleaseImage
538+ streamMeta * stream. Stream
539539 expectedImage string
540540 expectedErr bool
541541 expectedErrMsg string
542542 }{
543543 {
544544 name : "When stream metadata has project and name for amd64, it should construct image path" ,
545545 arch : hyperv1 .ArchitectureAMD64 ,
546- releaseImage : & releaseinfo.ReleaseImage {
547- StreamMetadata : & stream.Stream {
548- Architectures : map [string ]stream.Arch {
549- "x86_64" : {
550- Images : stream.Images {
551- Gcp : & stream.GcpImage {
552- Project : "rhcos-cloud" ,
553- Name : "rhcos-9-6-20251023-0-gcp-x86-64" ,
554- },
546+ streamMeta : & stream.Stream {
547+ Architectures : map [string ]stream.Arch {
548+ "x86_64" : {
549+ Images : stream.Images {
550+ Gcp : & stream.GcpImage {
551+ Project : "rhcos-cloud" ,
552+ Name : "rhcos-9-6-20251023-0-gcp-x86-64" ,
555553 },
556554 },
557555 },
558556 },
559557 },
560558 expectedImage : "projects/rhcos-cloud/global/images/rhcos-9-6-20251023-0-gcp-x86-64" ,
561- expectedErr : false ,
562559 },
563560 {
564561 name : "When stream metadata has project and name for arm64, it should construct image path" ,
565562 arch : hyperv1 .ArchitectureARM64 ,
566- releaseImage : & releaseinfo.ReleaseImage {
567- StreamMetadata : & stream.Stream {
568- Architectures : map [string ]stream.Arch {
569- "aarch64" : {
570- Images : stream.Images {
571- Gcp : & stream.GcpImage {
572- Project : "rhcos-cloud" ,
573- Name : "rhcos-9-6-20251023-0-gcp-aarch64" ,
574- },
563+ streamMeta : & stream.Stream {
564+ Architectures : map [string ]stream.Arch {
565+ "aarch64" : {
566+ Images : stream.Images {
567+ Gcp : & stream.GcpImage {
568+ Project : "rhcos-cloud" ,
569+ Name : "rhcos-9-6-20251023-0-gcp-aarch64" ,
575570 },
576571 },
577572 },
578573 },
579574 },
580575 expectedImage : "projects/rhcos-cloud/global/images/rhcos-9-6-20251023-0-gcp-aarch64" ,
581- expectedErr : false ,
582576 },
583577 {
584- name : "When architecture is not found in release metadata, it should return error" ,
578+ name : "When architecture is not found in stream metadata, it should return error" ,
585579 arch : "unsupported-arch" ,
586- releaseImage : & releaseinfo.ReleaseImage {
587- StreamMetadata : & stream.Stream {
588- Architectures : map [string ]stream.Arch {
589- "x86_64" : {
590- Images : stream.Images {
591- Gcp : & stream.GcpImage {
592- Project : "rhcos-cloud" ,
593- Name : "rhcos-9-6-20251023-0-gcp-x86-64" ,
594- },
580+ streamMeta : & stream.Stream {
581+ Architectures : map [string ]stream.Arch {
582+ "x86_64" : {
583+ Images : stream.Images {
584+ Gcp : & stream.GcpImage {
585+ Project : "rhcos-cloud" ,
586+ Name : "rhcos-9-6-20251023-0-gcp-x86-64" ,
595587 },
596588 },
597589 },
@@ -603,13 +595,11 @@ func TestDefaultNodePoolGCPImage(t *testing.T) {
603595 {
604596 name : "When GCP project and name are empty, it should return error" ,
605597 arch : hyperv1 .ArchitectureAMD64 ,
606- releaseImage : & releaseinfo.ReleaseImage {
607- StreamMetadata : & stream.Stream {
608- Architectures : map [string ]stream.Arch {
609- "x86_64" : {
610- Images : stream.Images {
611- Gcp : & stream.GcpImage {},
612- },
598+ streamMeta : & stream.Stream {
599+ Architectures : map [string ]stream.Arch {
600+ "x86_64" : {
601+ Images : stream.Images {
602+ Gcp : & stream.GcpImage {},
613603 },
614604 },
615605 },
@@ -620,14 +610,12 @@ func TestDefaultNodePoolGCPImage(t *testing.T) {
620610 {
621611 name : "When GCP project is empty but name is set, it should return error" ,
622612 arch : hyperv1 .ArchitectureAMD64 ,
623- releaseImage : & releaseinfo.ReleaseImage {
624- StreamMetadata : & stream.Stream {
625- Architectures : map [string ]stream.Arch {
626- "x86_64" : {
627- Images : stream.Images {
628- Gcp : & stream.GcpImage {
629- Name : "rhcos-x86-64" ,
630- },
613+ streamMeta : & stream.Stream {
614+ Architectures : map [string ]stream.Arch {
615+ "x86_64" : {
616+ Images : stream.Images {
617+ Gcp : & stream.GcpImage {
618+ Name : "rhcos-x86-64" ,
631619 },
632620 },
633621 },
@@ -639,14 +627,12 @@ func TestDefaultNodePoolGCPImage(t *testing.T) {
639627 {
640628 name : "When GCP name is empty but project is set, it should return error" ,
641629 arch : hyperv1 .ArchitectureAMD64 ,
642- releaseImage : & releaseinfo.ReleaseImage {
643- StreamMetadata : & stream.Stream {
644- Architectures : map [string ]stream.Arch {
645- "x86_64" : {
646- Images : stream.Images {
647- Gcp : & stream.GcpImage {
648- Project : "rhcos-cloud" ,
649- },
630+ streamMeta : & stream.Stream {
631+ Architectures : map [string ]stream.Arch {
632+ "x86_64" : {
633+ Images : stream.Images {
634+ Gcp : & stream.GcpImage {
635+ Project : "rhcos-cloud" ,
650636 },
651637 },
652638 },
@@ -655,13 +641,20 @@ func TestDefaultNodePoolGCPImage(t *testing.T) {
655641 expectedErr : true ,
656642 expectedErrMsg : "release image metadata has no GCP image for architecture \" amd64\" " ,
657643 },
644+ {
645+ name : "When stream metadata is nil, it should return error" ,
646+ arch : hyperv1 .ArchitectureAMD64 ,
647+ streamMeta : nil ,
648+ expectedErr : true ,
649+ expectedErrMsg : "stream metadata is nil" ,
650+ },
658651 }
659652
660653 for _ , tc := range testCases {
661654 t .Run (tc .name , func (t * testing.T ) {
662655 g := NewWithT (t )
663656
664- image , err := defaultNodePoolGCPImage (tc .arch , tc .releaseImage )
657+ image , err := DefaultNodePoolGCPImage (tc .arch , tc .streamMeta )
665658
666659 if tc .expectedErr {
667660 g .Expect (err ).To (HaveOccurred ())
0 commit comments