mirror of
https://github.com/kemko/nomad.git
synced 2026-01-02 00:15:43 +03:00
981ca36049
This PR replaces fsouza/go-dockerclient 3rd party docker client library with docker's official SDK. --------- Co-authored-by: Tim Gross <tgross@hashicorp.com> Co-authored-by: Seth Hoenig <shoenig@duck.com>
58 lines
2.3 KiB
Go
58 lines
2.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package docker
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/ci"
|
|
"github.com/shoenig/test/must"
|
|
)
|
|
|
|
func Test_DockerImageProgressManager(t *testing.T) {
|
|
ci.Parallel(t)
|
|
|
|
pm := &imageProgressManager{
|
|
imageProgress: &imageProgress{
|
|
timestamp: time.Now(),
|
|
layers: make(map[string]*layerProgress),
|
|
},
|
|
}
|
|
|
|
_, err := pm.Write([]byte(`{"status":"Pulling from library/golang","id":"1.9.5"}
|
|
{"status":"Pulling fs layer","progressDetail":{},"id":"c73ab1c6897b"}
|
|
{"status":"Pulling fs layer","progressDetail":{},"id":"1ab373b3deae"}
|
|
`))
|
|
must.NoError(t, err)
|
|
must.Eq(t, 2, len(pm.imageProgress.layers), must.Sprint("number of layers should be 2"))
|
|
|
|
cur := pm.imageProgress.currentBytes()
|
|
must.Zero(t, cur)
|
|
tot := pm.imageProgress.totalBytes()
|
|
must.Zero(t, tot)
|
|
|
|
_, err = pm.Write([]byte(`{"status":"Pulling fs layer","progress`))
|
|
must.NoError(t, err)
|
|
must.Eq(t, 2, len(pm.imageProgress.layers), must.Sprint("number of layers should be 2"))
|
|
|
|
_, err = pm.Write([]byte(`Detail":{},"id":"b542772b4177"}` + "\n"))
|
|
must.NoError(t, err)
|
|
must.Eq(t, 3, len(pm.imageProgress.layers), must.Sprint("number of layers should be 3"))
|
|
|
|
_, err = pm.Write([]byte(`{"status":"Downloading","progressDetail":{"current":45800,"total":4335495},"progress":"[\u003e ] 45.8kB/4.335MB","id":"b542772b4177"}
|
|
{"status":"Downloading","progressDetail":{"current":113576,"total":11108010},"progress":"[\u003e ] 113.6kB/11.11MB","id":"1ab373b3deae"}
|
|
{"status":"Downloading","progressDetail":{"current":694257,"total":4335495},"progress":"[========\u003e ] 694.3kB/4.335MB","id":"b542772b4177"}` + "\n"))
|
|
must.NoError(t, err)
|
|
must.Eq(t, 3, len(pm.imageProgress.layers), must.Sprint("number of layers should be 3"))
|
|
must.Eq(t, int64(807833), pm.imageProgress.currentBytes())
|
|
must.Eq(t, int64(15443505), pm.imageProgress.totalBytes())
|
|
|
|
_, err = pm.Write([]byte(`{"status":"Download complete","progressDetail":{},"id":"b542772b4177"}` + "\n"))
|
|
must.NoError(t, err)
|
|
must.Eq(t, 3, len(pm.imageProgress.layers), must.Sprint("number of layers should be 3"))
|
|
must.Eq(t, int64(4449071), pm.imageProgress.currentBytes())
|
|
must.Eq(t, int64(15443505), pm.imageProgress.totalBytes())
|
|
}
|