tests: diskusage command

Signed-off-by: CrazyMax <[email protected]>

Author: crazy-max

tests/diskusage.go

@@ -0,0 +1,42 @@
+package tests
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/moby/buildkit/client"
+	"github.com/moby/buildkit/util/testutil/integration"
+	"github.com/stretchr/testify/require"
+)
+
+var diskusageTests = []func(t *testing.T, sb integration.Sandbox){
+	testDiskusage,
+	testDiskusageVerboseRaw,
+	testDiskusageVerboseJSON,
+}
+
+func testDiskusage(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+}
+
+func testDiskusageVerboseRaw(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--verbose", "--format=raw"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+}
+
+func testDiskusageVerboseJSON(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--verbose", "--format=json"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+
+	var du []*client.UsageInfo
+	err = json.Unmarshal(out, &du)
+	require.NoError(t, err)
+	require.NotEmpty(t, du)
+}

tests/integration_test.go

@@ -33,6 +33,7 @@ func TestIntegration(t *testing.T) {
 	tests = append(tests, rmTests...)
 	tests = append(tests, dialstdioTests...)
 	tests = append(tests, composeTests...)
+	tests = append(tests, diskusageTests...)
 	testIntegration(t, tests...)
 }