| package binary |
| |
| import ( |
| "testing" |
| |
| "cos.googlesource.com/cos/tools/src/cmd/cos_image_analyzer/internal/input" |
| ) |
| |
| // test ReadFileToMap function |
| func TestGetBinaryInfo(t *testing.T) { |
| // test normal file |
| testImage1 := &input.ImageInfo{RootfsPartition3: "../testdata/image1"} |
| testImage2 := &input.ImageInfo{} |
| testImage3 := &input.ImageInfo{RootfsPartition3: "../testdata/image2"} |
| |
| expectedImage1 := &input.ImageInfo{RootfsPartition3: "../testdata/image1", Version: "81", BuildID: "12871.119.0"} |
| expectedImage2 := &input.ImageInfo{} |
| expectedImage3 := &input.ImageInfo{RootfsPartition3: "../testdata/image2", Version: "77", BuildID: "12371.273.0"} |
| |
| type test struct { |
| Image1 *input.ImageInfo |
| want *input.ImageInfo |
| } |
| |
| tests := []test{ |
| {Image1: testImage1, want: expectedImage1}, |
| {Image1: testImage2, want: expectedImage2}, |
| {Image1: testImage3, want: expectedImage3}, |
| } |
| |
| for _, tc := range tests { |
| GetBinaryInfo(tc.Image1) |
| |
| if tc.want.Version != tc.Image1.Version { |
| t.Fatalf("Diff expected: %v, got: %v", tc.want.Version, tc.Image1.Version) |
| } |
| if tc.want.BuildID != tc.Image1.BuildID { |
| t.Fatalf("Diff expected: %v, got: %v", tc.want.BuildID, tc.Image1.BuildID) |
| } |
| } |
| |
| } |