Add translate test

internal/config/config_test.go

@@ -1,9 +1,10 @@
 package config
 
 import (
-	"os"
 	"path/filepath"
 	"testing"
+
+	"github.com/candy12t/deepl-cli/test"
 )
 
 func TestParseConfig(t *testing.T) {
@@ -20,7 +21,7 @@ func TestParseConfig(t *testing.T) {
 			BaseURL: "https://api-free.deepl.com/v2",
 		}
 
-		configPath := filepath.Join(projectDirPath(), "testdata", "config", "free.yaml")
+		configPath := filepath.Join(test.ProjectDirPath(), "test", "testdata", "config", "free.yaml")
 		err := ParseConfig(configPath)
 		gotConfig := CachedConfig()
 
@@ -43,7 +44,7 @@ func TestParseConfig(t *testing.T) {
 			BaseURL: "https://api.deepl.com/v2",
 		}
 
-		configPath := filepath.Join(projectDirPath(), "testdata", "config", "pro.yaml")
+		configPath := filepath.Join(test.ProjectDirPath(), "test", "testdata", "config", "pro.yaml")
 		err := ParseConfig(configPath)
 		gotConfig := CachedConfig()
 
@@ -56,7 +57,7 @@ func TestParseConfig(t *testing.T) {
 	t.Run("can not read config file", func(t *testing.T) {
 		wantConfig := Config{}
 
-		configPath := filepath.Join(projectDirPath(), "testdata", "config", "not_read.yaml")
+		configPath := filepath.Join(test.ProjectDirPath(), "test", "testdata", "config", "not_read.yaml")
 		err := ParseConfig(configPath)
 		gotConfig := CachedConfig()
 
@@ -69,7 +70,7 @@ func TestParseConfig(t *testing.T) {
 	t.Run("can not unmarshal config file", func(t *testing.T) {
 		wantConfig := Config{}
 
-		configPath := filepath.Join(projectDirPath(), "testdata", "config", "not_unmarshal.yaml")
+		configPath := filepath.Join(test.ProjectDirPath(), "test", "testdata", "config", "not_unmarshal.yaml")
 		err := ParseConfig(configPath)
 		gotConfig := CachedConfig()
 
@@ -114,11 +115,6 @@ func assertUnmarshalError(t *testing.T, got, want error) {
 	}
 }
 
-func projectDirPath() string {
-	currentDirPath, _ := os.Getwd()
-	return filepath.Join(currentDirPath, "..", "..")
-}
-
 func cleanCachedConfig(t *testing.T) {
 	t.Cleanup(func() {
 		config = Config{}

internal/deepl/translate_test.go

@@ -0,0 +1,76 @@
+package deepl
+
+import (
+	"context"
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"testing"
+
+	"github.com/candy12t/deepl-cli/test"
+)
+
+func TestTranslate(t *testing.T) {
+	t.Run("success translate", func(t *testing.T) {
+		client, server, err := deeplMockServer(t, "test-auth-key", "success-body", "success-header")
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer server.Close()
+
+		ctx := context.Background()
+		got, err := client.Translate(ctx, "hello", "EN", "JA")
+		if err != nil {
+			t.Fatal(err)
+		}
+		want := "こんにちわ"
+
+		if got != want {
+			t.Fatalf("got %v, want %v", got, want)
+		}
+	})
+}
+
+func deeplMockServer(t *testing.T, mockAuthKey string, mockResponseBody, mockResponseHeader string) (*Client, *httptest.Server, error) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		header := filepath.Join(test.ProjectDirPath(), "test", "testdata", "deepl", mockResponseHeader)
+		headerBytes, err := ioutil.ReadFile(header)
+		h := strings.Split(string(headerBytes), "\n")
+		for _, line := range h {
+			if strings.Contains(line, "HTTP") {
+				statusCode, err := strconv.Atoi(strings.Split(line, " ")[1])
+				if err != nil {
+					t.Fatal(err)
+				}
+				w.WriteHeader(statusCode)
+			}
+
+			if strings.Contains(line, "content-type") {
+				contentType := strings.Split(line, " ")[1]
+				w.Header().Add("content-type", contentType)
+			}
+		}
+
+		body := filepath.Join(test.ProjectDirPath(), "test", "testdata", "deepl", mockResponseBody)
+		bodyBytes, err := ioutil.ReadFile(body)
+		if err != nil {
+			t.Fatal(err)
+		}
+		w.Write(bodyBytes)
+	}))
+
+	serverURL, err := url.ParseRequestURI(server.URL)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	return &Client{
+		BaseURL:    serverURL,
+		HTTPClient: server.Client(),
+		AuthKey:    mockAuthKey,
+	}, server, nil
+}

test/helper.go

@@ -0,0 +1,11 @@
+package test
+
+import (
+	"os"
+	"path/filepath"
+)
+
+func ProjectDirPath() string {
+	currentDirPath, _ := os.Getwd()
+	return filepath.Join(currentDirPath, "..", "..")
+}

test/testdata/deepl/non_target_lang-body

@@ -0,0 +1 @@
+{"message":"\"Value for 'target_lang' not supported.\""}

test/testdata/deepl/non_target_lang-header

@@ -0,0 +1,7 @@
+HTTP/2 400 
+server: nginx
+date: Sun, 14 Nov 2021 13:34:43 GMT
+content-type: application/json
+content-length: 56
+access-control-allow-origin: *
+

test/testdata/deepl/success-body

@@ -0,0 +1 @@
+{"translations":[{"detected_source_language":"EN","text":"こんにちわ"}]}

test/testdata/deepl/success-header

@@ -0,0 +1,7 @@
+HTTP/2 200 
+server: nginx
+date: Sun, 14 Nov 2021 13:29:37 GMT
+content-type: application/json
+content-length: 77
+access-control-allow-origin: *
+