test: Update main_test.go

Author: softwaredevelop

.github/workflows/go-test.yml

@@ -38,6 +38,7 @@ jobs:
           - hello
           - quarto
           - revive
+          - ruff
 
     steps:
       - name: Checkout code

ruff/main_test.go

@@ -31,7 +31,7 @@ func Test_Ruff(t *testing.T) {
 
 	t.Run("Test_ruff_with_config", func(t *testing.T) {
 		t.Parallel()
-		container := base()
+		container := base("")
 		require.NotNil(t, container)
 
 		_, err := container.
@@ -48,7 +48,7 @@ func Test_Ruff(t *testing.T) {
 	})
 	t.Run("Test_ruff_host_directory_check_file_error", func(t *testing.T) {
 		t.Parallel()
-		container := base()
+		container := base("")
 		require.NotNil(t, container)
 
 		_, err := container.
@@ -61,7 +61,7 @@ func Test_Ruff(t *testing.T) {
 	})
 	t.Run("Test_ruff_error", func(t *testing.T) {
 		t.Parallel()
-		container := base()
+		container := base("")
 		require.NotNil(t, container)
 
 		_, err := container.
@@ -76,7 +76,7 @@ func Test_Ruff(t *testing.T) {
 	})
 	t.Run("Test_ruff_version", func(t *testing.T) {
 		t.Parallel()
-		container := base()
+		container := base("")
 		require.NotNil(t, container)
 
 		out, err := container.
@@ -87,7 +87,23 @@ func Test_Ruff(t *testing.T) {
 	})
 }
 
-func base() *dagger.Container {
-	return c.Container().
-		From("ghcr.io/astral-sh/ruff:latest")
+// func base() *dagger.Container {
+// 	return c.Container().
+// 		From("ghcr.io/astral-sh/ruff:latest")
+// }
+
+func base(
+	image string,
+) *dagger.Container {
+
+	defaultImageRepository := "ghcr.io/astral-sh/ruff"
+	var ctr *dagger.Container
+
+	if image != "" {
+		ctr = c.Container().From(image)
+	} else {
+		ctr = c.Container().From(defaultImageRepository)
+	}
+
+	return ctr
 }