From 1a13e0a9bcc0d82603ef4c5fd6e8cf0465a224e6 Mon Sep 17 00:00:00 2001
From: melsonic <work.melsonic@gmail.com>
Date: Thu, 3 Apr 2025 21:04:49 +0530
Subject: [PATCH] added tests

---
 e2etests-cli/main_test.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/e2etests-cli/main_test.go b/e2etests-cli/main_test.go
index 441fbd986..6b06b63c5 100644
--- a/e2etests-cli/main_test.go
+++ b/e2etests-cli/main_test.go
@@ -1362,6 +1362,29 @@ c
 				assert.Error(t, err)
 			},
 		},
+		{
+			name: "omit-version",
+			run: func(t *testing.T, ctx context.Context, dir string, env *xos.Env) {
+				writeFile(t, dir, "test.d2", `x -> y`)
+				err := runTestMain(t, ctx, dir, env, "--omit-version", "test.d2", "no-version.svg")
+				assert.Success(t, err)
+				noVersionSvg := readFile(t, dir, "no-version.svg")
+				assert.False(t, strings.Contains(string(noVersionSvg), "data-d2-version="))
+
+				writeFile(t, dir, "test.d2", `x -> y`)
+				err = runTestMain(t, ctx, dir, env, "test.d2", "with-version.svg")
+				assert.Success(t, err)
+				withVersionSvg := readFile(t, dir, "with-version.svg")
+				assert.True(t, strings.Contains(string(withVersionSvg), "data-d2-version="))
+
+				env.Setenv("OMIT_VERSION", "1")
+				writeFile(t, dir, "test.d2", `x -> y`)
+				err = runTestMain(t, ctx, dir, env, "test.d2", "no-version-env.svg")
+				assert.Success(t, err)
+				noVersionEnvSvg := readFile(t, dir, "no-version-env.svg")
+				assert.False(t, strings.Contains(string(noVersionEnvSvg), "data-d2-version="))
+			},
+		},
 	}
 
 	ctx := context.Background()