migrate_test.go

package txlib

import (
	"fmt"
	"io"
	"log"
	"os"
	"path/filepath"
	"regexp"
	"strings"
	"testing"

	"github.com/transifex/cli/internal/txlib/config"
	"github.com/transifex/cli/pkg/assert"
	"github.com/transifex/cli/pkg/jsonapi"
)

func TestSuccessfulFindOrganizationSlug(t *testing.T) {
	org1ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug"
	org2ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg2&" +
		"filter%5Bslug%5D=projslug"
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(
			`{"data": [{"type": "organizations",
			            "id": "o:org",
									"attributes": {"slug": "org"}},
			           {"type": "organizations",
								  "id": "o:org2",
									"attributes": {"slug": "org2"}}]}`,
		),
		org1ProjectsUrl: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		org2ProjectsUrl: jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	api := jsonapi.GetTestConnection(mockData)
	resource := config.Resource{
		ProjectSlug: "projslug",
	}
	res, err := getOrganizationSlug(api, &resource)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, res, "org")
}

func TestFailToFindOrganizationSlug(t *testing.T) {
	org1ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug"
	org2ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg2&" +
		"filter%5Bslug%5D=projslug"
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(
			`{"data": [{"type": "organizations",
			            "id": "o:org",
									"attributes": {"slug": "org"}},
			           {"type": "organizations",
								  "id": "o:org2",
									"attributes": {"slug": "org2"}}]}`,
		),
		org1ProjectsUrl: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		org2ProjectsUrl: jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	api := jsonapi.GetTestConnection(mockData)
	resource := config.Resource{
		ProjectSlug: "projslug3",
	}
	res, err := getOrganizationSlug(api, &resource)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, res, "")
}

func TestSuccessfulMigration(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}

	// Requests Data
	org1ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug"
	org2ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg2&" +
		"filter%5Bslug%5D=projslug"
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(
			`{"data": [{"type": "organizations",
			            "id": "o:org",
									"attributes": {"slug": "org"}},
			           {"type": "organizations",
								  "id": "o:org2",
									"attributes": {"slug": "org2"}}]}`,
		),
		org1ProjectsUrl: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		org2ProjectsUrl: jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, _ := os.Getwd()
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create(".transifexrc")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[https://app.transifex.com]
		api_hostname  = https://api.transifex.com
		hostname      = https://app.transifex.com
		username      = api
		password      = apassword
	`)

	if err2 != nil {
		log.Fatal(err2)
	}

	f, err = os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 = f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_file = locale/en.po
		source_lang = en
		type = PO
		trans.pt-pt = locale/other/pt_PT/en.po
		trans.fr_CA = locale/other/fr_CA/ui.po
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	assert.Equal(t, cfg.GetActiveHost().Token, "")
	assert.Equal(t, cfg.GetActiveHost().RestHostname, "")
	assert.Equal(t, cfg.Local.Resources[0].OrganizationSlug, "")

	_, err = MigrateLegacyConfigFile(&cfg, api)
	if err != nil {
		t.Error(err)
	}

	// Load for the first time configs
	cfgReloaded, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	assert.Equal(t, cfgReloaded.GetActiveHost().Token, "apassword")
	assert.Equal(t, cfgReloaded.GetActiveHost().RestHostname,
		"https://rest.api.transifex.com")
	assert.Equal(t, cfgReloaded.Local.Resources[0].OrganizationSlug, "org")
	assert.Equal(t, len(cfgReloaded.Local.Resources[0].Overrides), 2)
}

func TestSuccessfulMigrationWithSourceFileConstruction(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}

	// Requests Data
	org1ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug"
	org2ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg2&" +
		"filter%5Bslug%5D=projslug"
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(
			`{"data": [{"type": "organizations",
			            "id": "o:org",
									"attributes": {"slug": "org"}},
			           {"type": "organizations",
								  "id": "o:org2",
									"attributes": {"slug": "org2"}}]}`,
		),
		org1ProjectsUrl: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		org2ProjectsUrl: jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, _ := os.Getwd()
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create(".transifexrc")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[https://app.transifex.com]
		api_hostname  = https://api.transifex.com
		hostname      = https://app.transifex.com
		username      = api
		password      = apassword
	`)

	if err2 != nil {
		log.Fatal(err2)
	}

	f, err = os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 = f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_lang = en
		type = PO
		trans.pt-pt = locale/other/pt_PT/en.po
		trans.fr_CA = locale/other/fr_CA/ui.po
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	assert.Equal(t, cfg.GetActiveHost().Token, "")
	assert.Equal(t, cfg.GetActiveHost().RestHostname, "")
	assert.Equal(t, cfg.Local.Resources[0].OrganizationSlug, "")

	_, err = MigrateLegacyConfigFile(&cfg, api)
	if err != nil {
		t.Error(err)
	}

	// Load for the first time configs
	cfgReloaded, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	assert.Equal(t, cfgReloaded.Local.Resources[0].OrganizationSlug, "org")
	assert.Equal(t, cfgReloaded.Local.Resources[0].SourceFile, "locale/en.po")
	assert.Equal(t, len(cfgReloaded.Local.Resources[0].Overrides), 2)
}

func TestNeedsTokenInRootConfig(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}
	// Requests Data
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, _ := os.Getwd()
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create(".transifexrc")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[https://app.transifex.com]
		api_hostname  = https://api.transifex.com
		hostname      = https://app.transifex.com
		username      = tk
		password      = apassword
	`)

	if err2 != nil {
		log.Fatal(err2)
	}

	f, err = os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 = f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_file = locale/en.po
		source_lang = en
		type = PO
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	rescueStdout := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	_, _ = MigrateLegacyConfigFile(&cfg, api)

	w.Close()
	out, _ := io.ReadAll(r)
	os.Stdout = rescueStdout

	assert.True(t, strings.Contains(string(out), "API token not found."))
}

func TestNoTransifexRcFile(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}
	// Requests Data
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, _ := os.Getwd()
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_file = locale/en.po
		source_lang = en
		type = PO
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths("", filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	rescueStdout := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	_, _ = MigrateLegacyConfigFile(&cfg, api)

	w.Close()
	out, _ := io.ReadAll(r)
	os.Stdout = rescueStdout

	assert.True(t, strings.Contains(string(out), "Please provide an API token to continue."))
}

func TestResourceMigrationFailed(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}
	// Requests Data
	project1Url := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug"
	project2Url := "/projects?filter%5Borganization%5D=o%3Aorg&" +
		"filter%5Bslug%5D=projslug2"
	org2ProjectsUrl := "/projects?filter%5Borganization%5D=o%3Aorg2&" +
		"filter%5Bslug%5D=projslug"
	mockData := jsonapi.MockData{
		"/organizations": &jsonapi.MockEndpoint{
			Requests: []jsonapi.MockRequest{
				{
					Response: jsonapi.MockResponse{
						Text: `{"data": [
							{"type": "organizations",
							 "id": "o:org",
							 "attributes": {"slug": "org"}},
							{"type": "organizations",
							 "id": "o:org2",
							 "attributes": {"slug": "org2"}}
						]}`,
					},
				},
				{
					Response: jsonapi.MockResponse{
						Text: `{"data": [
							{"type": "organizations",
							 "id": "o:org",
							 "attributes": {"slug": "org"}},
							{"type": "organizations",
							 "id": "o:org2",
							 "attributes": {"slug": "org2"}}
						]}`,
					},
				},
			},
		},
		project1Url: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		project2Url: jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug2",
				"attributes": {"name": "Proj Name 2", "slug": "projslug2"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		org2ProjectsUrl: jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, _ := os.Getwd()
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create(".transifexrc")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[https://app.transifex.com]
		api_hostname  = https://api.transifex.com
		hostname      = https://app.transifex.com
		username      = api
		password      = apassword
	`)

	if err2 != nil {
		log.Fatal(err2)
	}

	f, err = os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 = f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug1.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 10
		source_file = locale/en.po
		source_lang = en
		type = PO
		[projslug2.ares2]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_file = locale/en.po
		source_lang = en
		type = PO
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	assert.Equal(t, cfg.Local.Resources[0].OrganizationSlug, "")
	assert.Equal(t, cfg.Local.Resources[1].OrganizationSlug, "")

	rescueStdout := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	_, err = MigrateLegacyConfigFile(&cfg, api)
	if err != nil {
		t.Error(err)
	}

	w.Close()
	out, _ := io.ReadAll(r)
	os.Stdout = rescueStdout
	assert.True(t, strings.Contains(
		string(out), "Could not migrate resource `ares`"))

	content, err := os.ReadFile(filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}
	assert.True(t, strings.Contains(
		string(content), "projslug1.ares"))
	assert.True(t, strings.Contains(
		string(content), "o:org:p:projslug2:r:ares2"))
	matched, err := regexp.MatchString(`minimum_perc\s*=\s*10`, string(content))
	if err != nil {
		t.Error(err)
	}
	assert.True(t, matched)
	matched, err = regexp.MatchString(`minimum_perc\s*=\s*0`, string(content))
	if err != nil {
		t.Error(err)
	}
	assert.True(t, matched)
}

func TestBackUpFileCreated(t *testing.T) {
	var afterTest = func(pkgDir string, tmpDir string) {
		err := os.Chdir(pkgDir)
		if err != nil {
			t.Error(err)
		}
		err = os.RemoveAll(tmpDir)
		if err != nil {
			fmt.Println("Delete error:", err)
		}
	}

	// Requests Data
	mockData := jsonapi.MockData{
		"/organizations": jsonapi.GetMockTextResponse(
			`{"data": [{"type": "organizations",
			            "id": "o:org",
									"attributes": {"slug": "org"}},
			           {"type": "organizations",
								  "id": "o:org2",
									"attributes": {"slug": "org2"}}]}`,
		),
		"/projects?filter%5Borganization%5D=o%3Aorg&filter%5Bslug%5D=projslug": jsonapi.GetMockTextResponse(
			`{"data": [{
				"type": "projects",
				"id": "o:orgslug:p:projslug",
				"attributes": {"name": "Proj Name", "slug": "projslug"},
				"relationships": {"organization": {
					"data": {"type": "organizations", "id": "o:orgslug"},
					"links": {"related": "/organizations/o:orgslug"}
				}}
			}]}`,
		),
		"/projects?filter%5Borganization%5D=o%3Aorg2&filter%5Bslug%5D=projslug": jsonapi.GetMockTextResponse(`{"data": []}`),
	}

	// Create deprecated config & .transifexrc
	pkgDir, err := os.Getwd()
	if err != nil {
		t.Error(err)
	}
	tmpDir, err := os.MkdirTemp("", "")
	if err != nil {
		log.Fatal(err)
	}

	err = os.Chdir(tmpDir)
	if err != nil {
		t.Error(err)
	}
	defer afterTest(pkgDir, tmpDir)

	f, err := os.Create(".transifexrc")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 := f.WriteString(`
		[https://app.transifex.com]
		api_hostname  = https://api.transifex.com
		hostname      = https://app.transifex.com
		username      = api
		password      = apassword
	`)

	if err2 != nil {
		log.Fatal(err2)
	}

	f, err = os.Create("config")

	if err != nil {
		log.Fatal(err)
	}

	defer f.Close()

	_, err2 = f.WriteString(`
		[main]
		host = https://app.transifex.com
		[projslug.ares]
		file_filter = locale/<lang>.po
		minimum_perc = 0
		source_file = locale/en.po
		source_lang = en
		type = PO
	`)
	if err2 != nil {
		log.Fatal(err2)
	}

	// Load for the first time configs
	cfg, err := config.LoadFromPaths(
		filepath.Join(tmpDir, ".transifexrc"), filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}

	api := jsonapi.GetTestConnection(mockData)

	backupFilePath, _ := MigrateLegacyConfigFile(&cfg, api)

	newContent, err := os.ReadFile(filepath.Join(tmpDir, "config"))
	if err != nil {
		t.Error(err)
	}
	buContent, err := os.ReadFile(filepath.Join(backupFilePath))
	if err != nil {
		t.Error(err)
	}

	if err != nil {
		t.Errorf("A backup file was expected %s", err.Error())
	}

	assert.True(t, strings.Contains(string(buContent), "[projslug.ares]"))
	assert.True(t, strings.Contains(string(newContent),
		"o:org:p:projslug:r:ares"))
}