added processing of go files without using ast

1 files changed, 20 insertions, 2 deletions

diff --git a/main_test.go b/main_test.go
index b0ede2f..17a8219 100644
--- a/main_test.go
+++ b/main_test.go
@@ -7,12 +7,30 @@ import (
 	"testing"
 )
 
-func TestApp(t *testing.T) {
+func TestAppAST(t *testing.T) {
 	fileContent := "package testing\n\nimport \"fmt\"\n\nfunc HelloWorld() {\nfmt.Println(\"Hello World!\")\n}\n"
 
 	ioutil.WriteFile("hello.go", []byte(fileContent), os.ModePerm)
 
-	ProcessFile("hello.go", "fmt", "replacedImport")
+	ProcessFileAST("hello.go", "fmt", "replacedImport")
+
+	result, err := ioutil.ReadFile("hello.go")
+	if err != nil {
+		t.Error("Failed to read written file.")
+	}
+
+	if !strings.Contains(string(result), "replacedImport") {
+		t.Error("Got different results")
+	}
+	os.Remove("hello.go")
+}
+
+func TestAppNative(t *testing.T) {
+	fileContent := "package testing\n\nimport \"fmt\"\n\nfunc HelloWorld() {\nfmt.Println(\"Hello World!\")\n}\n"
+
+	ioutil.WriteFile("hello.go", []byte(fileContent), os.ModePerm)
+
+	ProcessFileNative("hello.go", "fmt", "replacedImport")
 
 	result, err := ioutil.ReadFile("hello.go")
 	if err != nil {