1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
//go:build !remote
package libimage
import (
"context"
"testing"
"time"
"github.com/containers/common/pkg/config"
"github.com/containers/storage"
"github.com/stretchr/testify/require"
)
func TestDiskUsage(t *testing.T) {
runtime := testNewRuntime(t)
ctx := context.Background()
const expectedTotalImageSize int64 = 5847966
name := "quay.io/libpod/alpine:3.10.2"
pullOptions := &PullOptions{}
pulledImages, err := runtime.Pull(ctx, name, config.PullPolicyAlways, pullOptions)
require.NoError(t, err)
require.Len(t, pulledImages, 1)
imgID := pulledImages[0].storageImage.ID
layerID := pulledImages[0].storageImage.TopLayer
digest := pulledImages[0].storageImage.Digest
img, err := pulledImages[0].storageReference.NewImageSource(ctx, &runtime.systemContext)
require.NoError(t, err)
defer img.Close()
manifest, _, err := img.GetManifest(ctx, nil)
require.NoError(t, err)
expectedImageDiskUsage := ImageDiskUsage{
ID: imgID,
Repository: "quay.io/libpod/alpine",
Tag: "3.10.2",
SharedSize: 0,
UniqueSize: expectedTotalImageSize,
Size: expectedTotalImageSize,
}
res, size, err := runtime.DiskUsage(ctx)
require.NoError(t, err)
require.Equal(t, expectedTotalImageSize, size)
require.Len(t, res, 1)
// intentionally unsetting the time here, we cannot really equal the time
// because of the local information that is part of the struct and that
// can differ even when the time is the same
res[0].Created = time.Time{}
require.Equal(t, expectedImageDiskUsage, res[0])
opts := &storage.ImageOptions{
BigData: []storage.ImageBigDataOption{
{
Key: storage.ImageDigestBigDataKey,
Data: manifest,
Digest: digest,
},
},
}
img2, err := runtime.store.CreateImage("", []string{"localhost/test:123"}, layerID, "", opts)
require.NoError(t, err)
const sharedSize int64 = 5843968
// copy the expected and update the expected values
expectedImageDiskUsage2 := ImageDiskUsage{
ID: img2.ID,
Repository: "localhost/test",
Tag: "123",
SharedSize: sharedSize,
UniqueSize: int64(len(manifest)),
Size: sharedSize + int64(len(manifest)),
}
expectedImageDiskUsage.SharedSize = sharedSize
expectedImageDiskUsage.UniqueSize = expectedImageDiskUsage.Size - sharedSize
res, size, err = runtime.DiskUsage(ctx)
require.NoError(t, err)
require.Equal(t, expectedTotalImageSize+int64(len(manifest)), size)
require.Len(t, res, 2)
res[0].Created = time.Time{}
res[1].Created = time.Time{}
require.ElementsMatch(t, []ImageDiskUsage{expectedImageDiskUsage, expectedImageDiskUsage2}, res)
}
|
