| package image |
| |
| import ( |
| "encoding/json" |
| "runtime" |
| "sort" |
| "strings" |
| "testing" |
| |
| "github.com/docker/docker/api/types/container" |
| "github.com/docker/docker/layer" |
| "github.com/stretchr/testify/assert" |
| "github.com/stretchr/testify/require" |
| ) |
| |
| const sampleImageJSON = `{ |
| "architecture": "amd64", |
| "os": "linux", |
| "config": {}, |
| "rootfs": { |
| "type": "layers", |
| "diff_ids": [] |
| } |
| }` |
| |
| func TestNewFromJSON(t *testing.T) { |
| img, err := NewFromJSON([]byte(sampleImageJSON)) |
| require.NoError(t, err) |
| assert.Equal(t, sampleImageJSON, string(img.RawJSON())) |
| } |
| |
| func TestNewFromJSONWithInvalidJSON(t *testing.T) { |
| _, err := NewFromJSON([]byte("{}")) |
| assert.EqualError(t, err, "invalid image JSON, no RootFS key") |
| } |
| |
| func TestMarshalKeyOrder(t *testing.T) { |
| b, err := json.Marshal(&Image{ |
| V1Image: V1Image{ |
| Comment: "a", |
| Author: "b", |
| Architecture: "c", |
| }, |
| }) |
| assert.NoError(t, err) |
| |
| expectedOrder := []string{"architecture", "author", "comment"} |
| var indexes []int |
| for _, k := range expectedOrder { |
| indexes = append(indexes, strings.Index(string(b), k)) |
| } |
| |
| if !sort.IntsAreSorted(indexes) { |
| t.Fatal("invalid key order in JSON: ", string(b)) |
| } |
| } |
| |
| func TestImage(t *testing.T) { |
| cid := "50a16564e727" |
| config := &container.Config{ |
| Hostname: "hostname", |
| Domainname: "domain", |
| User: "root", |
| } |
| os := runtime.GOOS |
| |
| img := &Image{ |
| V1Image: V1Image{ |
| Config: config, |
| }, |
| computedID: ID(cid), |
| } |
| |
| assert.Equal(t, cid, img.ImageID()) |
| assert.Equal(t, cid, img.ID().String()) |
| assert.Equal(t, os, img.OperatingSystem()) |
| assert.Equal(t, config, img.RunConfig()) |
| } |
| |
| func TestImageOSNotEmpty(t *testing.T) { |
| os := "os" |
| img := &Image{ |
| V1Image: V1Image{ |
| OS: os, |
| }, |
| OSVersion: "osversion", |
| } |
| assert.Equal(t, os, img.OperatingSystem()) |
| } |
| |
| func TestNewChildImageFromImageWithRootFS(t *testing.T) { |
| rootFS := NewRootFS() |
| rootFS.Append(layer.DiffID("ba5e")) |
| parent := &Image{ |
| RootFS: rootFS, |
| History: []History{ |
| NewHistory("a", "c", "r", false), |
| }, |
| } |
| childConfig := ChildConfig{ |
| DiffID: layer.DiffID("abcdef"), |
| Author: "author", |
| Comment: "comment", |
| ContainerConfig: &container.Config{ |
| Cmd: []string{"echo", "foo"}, |
| }, |
| Config: &container.Config{}, |
| } |
| |
| newImage := NewChildImage(parent, childConfig, "platform") |
| expectedDiffIDs := []layer.DiffID{layer.DiffID("ba5e"), layer.DiffID("abcdef")} |
| assert.Equal(t, expectedDiffIDs, newImage.RootFS.DiffIDs) |
| assert.Equal(t, childConfig.Author, newImage.Author) |
| assert.Equal(t, childConfig.Config, newImage.Config) |
| assert.Equal(t, *childConfig.ContainerConfig, newImage.ContainerConfig) |
| assert.Equal(t, "platform", newImage.OS) |
| assert.Equal(t, childConfig.Config, newImage.Config) |
| |
| assert.Len(t, newImage.History, 2) |
| assert.Equal(t, childConfig.Comment, newImage.History[1].Comment) |
| |
| // RootFS should be copied not mutated |
| assert.NotEqual(t, parent.RootFS.DiffIDs, newImage.RootFS.DiffIDs) |
| } |