add test

99designs · Feb 18, 2020 · 3abe5b3 · 3abe5b3
1 parent 6ecdb88
commit 3abe5b3
Showing 1 changed file with 33 additions and 25 deletions.
diff --git a/internal/rewrite/rewriter_test.go b/internal/rewrite/rewriter_test.go
@@ -8,29 +8,37 @@ import (
 )
 
 func TestRewriter(t *testing.T) {
-	r, err := New("github.com/99designs/gqlgen/internal/rewrite/testdata")
-	require.NoError(t, err)
-
-	body := r.GetMethodBody("Foo", "Method")
-	require.Equal(t, `
-	// leading comment
-
-	// field comment
-	m.Field++
-
-	// trailing comment
-`, body)
-
-	imps := r.ExistingImports("testdata/example.go")
-	require.Len(t, imps, 2)
-	assert.Equal(t, []Import{
-		{
-			Alias:      "",
-			ImportPath: "fmt",
-		},
-		{
-			Alias:      "lol",
-			ImportPath: "bytes",
-		},
-	}, imps)
+	t.Run("default", func(t *testing.T) {
+		r, err := New("github.com/99designs/gqlgen/internal/rewrite/testdata")
+		require.NoError(t, err)
+
+		body := r.GetMethodBody("Foo", "Method")
+		require.Equal(t, `
+			// leading comment
+
+			// field comment
+			m.Field++
+
+			// trailing comment
+		`, body)
+
+		imps := r.ExistingImports("testdata/example.go")
+		require.Len(t, imps, 2)
+		assert.Equal(t, []Import{
+			{
+				Alias:      "",
+				ImportPath: "fmt",
+			},
+			{
+				Alias:      "lol",
+				ImportPath: "bytes",
+			},
+		}, imps)
+
+	})
+
+	t.Run("out of scope dir", func(t *testing.T) {
+		_, err := New("../../../out-of-gomod/package")
+		require.Error(t, err)
+	})
 }