From f0646eeab58067d672a6b6b236657036ccd3cd52 Mon Sep 17 00:00:00 2001
From: CrazyMax <1951866+crazy-max@users.noreply.github.com>
Date: Fri, 22 Aug 2025 15:05:41 +0200
Subject: [PATCH] tests: diskusage command

Signed-off-by: CrazyMax <1951866+crazy-max@users.noreply.github.com>
---
 tests/diskusage.go        | 51 +++++++++++++++++++++++++++++++++++++++
 tests/integration_test.go |  1 +
 2 files changed, 52 insertions(+)
 create mode 100644 tests/diskusage.go

diff --git a/tests/diskusage.go b/tests/diskusage.go
new file mode 100644
index 000000000..8cfb4caa9
--- /dev/null
+++ b/tests/diskusage.go
@@ -0,0 +1,51 @@
+package tests
+
+import (
+	"testing"
+
+	"github.com/moby/buildkit/util/testutil/integration"
+	"github.com/stretchr/testify/require"
+)
+
+var diskusageTests = []func(t *testing.T, sb integration.Sandbox){
+	testDiskusage,
+	testDiskusageVerbose,
+	testDiskusageVerboseFormatError,
+	testDiskusageFormatJSON,
+	testDiskusageFormatGoTemplate,
+}
+
+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 testDiskusageVerbose(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--verbose"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+}
+
+func testDiskusageVerboseFormatError(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--verbose", "--format=json"))
+	out, err := cmd.Output()
+	require.Error(t, err, string(out))
+}
+
+func testDiskusageFormatJSON(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--format=json"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+}
+
+func testDiskusageFormatGoTemplate(t *testing.T, sb integration.Sandbox) {
+	buildTestProject(t, sb)
+	cmd := buildxCmd(sb, withArgs("du", "--format={{.ID}}: {{.Size}}"))
+	out, err := cmd.Output()
+	require.NoError(t, err, string(out))
+}
diff --git a/tests/integration_test.go b/tests/integration_test.go
index 44375b45a..e77a1a434 100644
--- a/tests/integration_test.go
+++ b/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...)
 }