// Copyright 2020 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package spec

import (
	"fmt"
	"testing"

	"github.com/joomcode/errorx"
	"github.com/pingcap/errors"
	"github.com/pingcap/tiup/pkg/utils"
	"github.com/stretchr/testify/require"
	"gopkg.in/yaml.v3"
)

func TestDirectoryConflicts1(t *testing.T) {
	topo := Specification{}

	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tidb_servers:
  - host: 172.16.5.138
    deploy_dir: "/test-1"
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-1"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "directory conflict for '/test-1' between 'tidb_servers:172.16.5.138.deploy_dir' and 'pd_servers:172.16.5.138.data_dir'", err.Error())
}

func TestPortConflicts(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tidb_servers:
  - host: 172.16.5.138
    port: 1234
tikv_servers:
  - host: 172.16.5.138
    status_port: 1234
`), &topo)
	require.Error(t, err)
	require.Equal(t, "port conflict for '1234' between 'tidb_servers:172.16.5.138.port' and 'tikv_servers:172.16.5.138.status_port'", err.Error())

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    tcp_port: 111
    http_port: 222
    flash_service_port: 1234
    flash_proxy_port: 444
    flash_proxy_status_port: 555
    metrics_port: 666
  - host: 172.16.5.138
    tcp_port: 1111
    http_port: 1222
    flash_service_port: 1333
    flash_proxy_port: 1444
    flash_proxy_status_port: 1234
    metrics_port: 1666
`), &topo)
	require.Error(t, err)
	require.Equal(t, "port conflict for '1234' between 'tiflash_servers:172.16.5.138.flash_service_port' and 'tiflash_servers:172.16.5.138.flash_proxy_status_port'", err.Error())

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    tcp_port: 111
    http_port: 222
    flash_service_port: 333
    flash_proxy_port: 1234
    flash_proxy_status_port: 555
    metrics_port: 666
  - host: 172.16.5.138
    tcp_port: 1111
    http_port: 1222
    flash_service_port: 1333
    flash_proxy_port: 1234
    flash_proxy_status_port: 1555
    metrics_port: 1666
`), &topo)
	require.Error(t, err)
	require.Equal(t, "port conflict for '1234' between 'tiflash_servers:172.16.5.138.flash_proxy_port' and 'tiflash_servers:172.16.5.138.flash_proxy_port'", err.Error())

	topo = Specification{}
	// tispark_masters has "omitempty" in its tag value
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 1234
tispark_masters:
  - host: 172.16.5.138
    port: 1234
tikv_servers:
  - host: 172.16.5.138
    status_port: 2345
`), &topo)
	require.Error(t, err)
	require.Equal(t, "port conflict for '1234' between 'tispark_masters:172.16.5.138.port' and 'monitored:172.16.5.138.node_exporter_port'", err.Error())
}

func TestPlatformConflicts(t *testing.T) {
	// aarch64 and arm64 are equal
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  os: "linux"
  arch: "aarch64"
tidb_servers:
  - host: 172.16.5.138
    arch: "arm64"
    os: "linux"
tikv_servers:
  - host: 172.16.5.138
    arch: "arm64"
    os: "linux"
`), &topo)
	require.NoError(t, err)

	// different arch defined for the same host
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  os: "linux"
tidb_servers:
  - host: 172.16.5.138
    arch: "aarch64"
    os: "linux"
tikv_servers:
  - host: 172.16.5.138
    arch: "amd64"
    os: "linux"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "platform mismatch for '172.16.5.138' between 'tidb_servers:linux/arm64' and 'tikv_servers:linux/amd64'", err.Error())

	// different os defined for the same host
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  os: "linux"
  arch: "aarch64"
tidb_servers:
  - host: 172.16.5.138
    os: "darwin"
    arch: "arm64"
tikv_servers:
  - host: 172.16.5.138
    arch: "arm64"
    os: "linux"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "platform mismatch for '172.16.5.138' between 'tidb_servers:darwin/arm64' and 'tikv_servers:linux/arm64'", err.Error())
}

func TestCountDir(t *testing.T) {
	topo := Specification{}

	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
tikv_servers:
  - host: 172.16.5.138
    data_dir: "/test-data/data-1"
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-data/data-2"
`), &topo)
	require.NoError(t, err)
	cnt := topo.CountDir("172.16.5.138", "/home/test1/test-deploy/pd-2379")
	require.Equal(t, 2, cnt)
	cnt = topo.CountDir("172.16.5.138", "") // the default user home
	require.Equal(t, 4, cnt)
	cnt = topo.CountDir("172.16.5.138", "/test-data/data")
	require.Equal(t, 0, cnt) // should not match partial path

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "/test-deploy"
tikv_servers:
  - host: 172.16.5.138
    data_dir: "/test-data/data-1"
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-data/data-2"
`), &topo)
	require.NoError(t, err)
	cnt = topo.CountDir("172.16.5.138", "/test-deploy/pd-2379")
	require.Equal(t, 2, cnt)
	cnt = topo.CountDir("172.16.5.138", "")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.16.5.138", "test-data")
	require.Equal(t, 0, cnt)

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "/test-deploy"
  data_dir: "/test-data"
tikv_servers:
  - host: 172.16.5.138
    data_dir: "data-1"
pd_servers:
  - host: 172.16.5.138
    data_dir: "data-2"
  - host: 172.16.5.139
`), &topo)
	require.NoError(t, err)
	// if per-instance data_dir is set, the global data_dir is ignored, and if it
	// is a relative path, it will be under the instance's deploy_dir
	cnt = topo.CountDir("172.16.5.138", "/test-deploy/pd-2379")
	require.Equal(t, 3, cnt)
	cnt = topo.CountDir("172.16.5.138", "")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.16.5.139", "/test-data")
	require.Equal(t, 1, cnt)
}

func TestTiSparkSpecValidation(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
tispark_masters:
  - host: 172.16.5.138
    port: 1235
tispark_workers:
  - host: 172.16.5.138
    port: 1236
  - host: 172.16.5.139
    port: 1235
`), &topo)
	require.NoError(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
tispark_masters:
  - host: 172.16.5.138
    port: 1235
  - host: 172.16.5.139
    port: 1235
`), &topo)
	require.Error(t, err)
	require.Equal(t, "a TiSpark enabled cluster with more than 1 Spark master node is not supported", err.Error())

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
tispark_workers:
  - host: 172.16.5.138
    port: 1235
  - host: 172.16.5.139
    port: 1235
`), &topo)
	require.Error(t, err)
	require.Equal(t, "there must be a Spark master node if you want to use the TiSpark component", err.Error())

	err = yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
tispark_masters:
  - host: 172.16.5.138
    port: 1236
tispark_workers:
  - host: 172.16.5.138
    port: 1235
  - host: 172.16.5.139
    port: 1235
  - host: 172.16.5.139
    port: 1236
    web_port: 8089
`), &topo)
	require.Error(t, err)
	require.Equal(t, "the host 172.16.5.139 is duplicated: multiple TiSpark workers on the same host is not supported by Spark", err.Error())
}

func TestTLSEnabledValidation(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  enable_tls: true
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
tidb_servers:
  - host: 172.16.5.138
    port: 1235
tikv_servers:
  - host: 172.16.5.138
    port: 1236
`), &topo)
	require.NoError(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  enable_tls: true
tidb_servers:
  - host: 172.16.5.138
    port: 1234
tispark_masters:
  - host: 172.16.5.138
    port: 1235
  - host: 172.16.5.139
    port: 1235
`), &topo)
	require.Error(t, err)
	require.Equal(t, "component tispark is not supported in TLS enabled cluster", err.Error())
}

func TestMonitorAgentValidation(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    port: 1234
  - host: 172.16.5.139
    ignore_exporter: true
`), &topo)
	require.NoError(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    port: 1234
tikv_servers:
  - host: 172.16.5.138
    ignore_exporter: true
`), &topo)
	require.Error(t, err)
	require.Equal(t, "ignore_exporter mismatch for '172.16.5.138' between 'tikv_servers:true' and 'pd_servers:false'", err.Error())
}

func TestCrossClusterPortConflicts(t *testing.T) {
	topo1 := Specification{}
	err := yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    client_port: 1234
    peer_port: 1235
`), &topo1)
	require.NoError(t, err)

	topo2 := Specification{}
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9101
  blackbox_exporter_port: 9116
pd_servers:
  - host: 172.16.5.138
    client_port: 2234
    peer_port: 2235
  - host: 172.16.5.139
    client_port: 2236
    peer_port: 2237
    ignore_exporter: true
`), &topo2)
	require.NoError(t, err)

	clsList := make(map[string]Metadata)

	// no port conflict with empty list
	err = CheckClusterPortConflict(clsList, "topo", &topo2)
	require.NoError(t, err)

	clsList["topo1"] = &ClusterMeta{Topology: &topo1}

	// no port conflict
	err = CheckClusterPortConflict(clsList, "topo", &topo2)
	require.NoError(t, err)

	// add topo2 to the list
	clsList["topo2"] = &ClusterMeta{Topology: &topo2}

	// monitoring agent port conflict
	topo3 := Specification{}
	err = yaml.Unmarshal([]byte(`
tidb_servers:
  - host: 172.16.5.138
`), &topo3)
	require.NoError(t, err)
	err = CheckClusterPortConflict(clsList, "topo", &topo3)
	require.Error(t, err)
	require.Equal(t, "spec.deploy.port_conflict: Deploy port conflicts to an existing cluster", errors.Cause(err).Error())
	suggestion, ok := errorx.ExtractProperty(err, utils.ErrPropSuggestion)
	require.True(t, ok)
	require.Equal(t, `The port you specified in the topology file is:
  Port:      9100
  Component: monitor 172.16.5.138

It conflicts to a port in the existing cluster:
  Existing Cluster Name: topo1
  Existing Port:         9100
  Existing Component:    monitor 172.16.5.138

Please change to use another port or another host.`, suggestion)

	// monitoring agent port conflict but the instance marked as ignore_exporter
	topo3 = Specification{}
	err = yaml.Unmarshal([]byte(`
tidb_servers:
- host: 172.16.5.138
  ignore_exporter: true
`), &topo3)
	require.NoError(t, err)
	err = CheckClusterPortConflict(clsList, "topo", &topo3)
	require.NoError(t, err)

	// monitoring agent port conflict but the existing instance marked as ignore_exporter
	topo3 = Specification{}
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9102
  blackbox_exporter_port: 9116
tidb_servers:
- host: 172.16.5.139
`), &topo3)
	require.NoError(t, err)
	err = CheckClusterPortConflict(clsList, "topo", &topo3)
	require.NoError(t, err)

	// component port conflict
	topo4 := Specification{}
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9102
  blackbox_exporter_port: 9117
pump_servers:
  - host: 172.16.5.138
    port: 2235
`), &topo4)
	require.NoError(t, err)
	err = CheckClusterPortConflict(clsList, "topo", &topo4)
	require.Error(t, err)
	require.Equal(t, "spec.deploy.port_conflict: Deploy port conflicts to an existing cluster", errors.Cause(err).Error())
	suggestion, ok = errorx.ExtractProperty(err, utils.ErrPropSuggestion)
	require.True(t, ok)
	require.Equal(t, `The port you specified in the topology file is:
  Port:      2235
  Component: pump 172.16.5.138

It conflicts to a port in the existing cluster:
  Existing Cluster Name: topo2
  Existing Port:         2235
  Existing Component:    pd 172.16.5.138

Please change to use another port or another host.`, suggestion)
}

func TestCrossClusterDirConflicts(t *testing.T) {
	topo1 := Specification{}
	err := yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    client_port: 1234
    peer_port: 1235
`), &topo1)
	require.NoError(t, err)

	topo2 := Specification{}
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9101
  blackbox_exporter_port: 9116
pd_servers:
  - host: 172.16.5.138
    client_port: 2234
    peer_port: 2235
`), &topo2)
	require.NoError(t, err)

	clsList := make(map[string]Metadata)

	// no port conflict with empty list
	err = CheckClusterDirConflict(clsList, "topo", &topo2)
	require.NoError(t, err)

	clsList["topo1"] = &ClusterMeta{Topology: &topo1}

	// no port conflict
	err = CheckClusterDirConflict(clsList, "topo", &topo2)
	require.NoError(t, err)

	// add topo2 to the list
	clsList["topo2"] = &ClusterMeta{Topology: &topo2}

	// monitoring agent dir conflict
	topo3 := Specification{}
	err = yaml.Unmarshal([]byte(`
tidb_servers:
  - host: 172.16.5.138
`), &topo3)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo3)
	require.Error(t, err)
	require.Equal(t, "spec.deploy.dir_conflict: Deploy directory conflicts to an existing cluster", errors.Cause(err).Error())
	suggestion, ok := errorx.ExtractProperty(err, utils.ErrPropSuggestion)
	require.True(t, ok)
	require.Equal(t, `The directory you specified in the topology file is:
  Directory: monitor deploy directory /home/tidb/deploy/monitor-9100
  Component: tidb 172.16.5.138

It conflicts to a directory in the existing cluster:
  Existing Cluster Name: topo1
  Existing Directory:    monitor deploy directory /home/tidb/deploy/monitor-9100
  Existing Component:    pd 172.16.5.138

Please change to use another directory or another host.`, suggestion)

	// no dir conflict error if one of the instance is marked as ignore_exporter
	err = yaml.Unmarshal([]byte(`
tidb_servers:
  - host: 172.16.5.138
    ignore_exporter: true
`), &topo3)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo3)
	require.NoError(t, err)

	// component with different port has no dir conflict
	topo4 := Specification{}
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9102
  blackbox_exporter_port: 9117
pump_servers:
  - host: 172.16.5.138
    port: 2235
`), &topo4)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo4)
	require.NoError(t, err)

	// component with relative dir has no dir conflic
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9102
  blackbox_exporter_port: 9117
pump_servers:
  - host: 172.16.5.138
    port: 2235
    data_dir: "pd-1234"
`), &topo4)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo4)
	require.NoError(t, err)

	// component with absolute dir conflict
	err = yaml.Unmarshal([]byte(`
monitored:
  node_exporter_port: 9102
  blackbox_exporter_port: 9117
pump_servers:
  - host: 172.16.5.138
    port: 2235
    data_dir: "/home/tidb/deploy/pd-2234"
`), &topo4)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo4)
	require.Error(t, err)
	require.Equal(t, "spec.deploy.dir_conflict: Deploy directory conflicts to an existing cluster", errors.Cause(err).Error())
	suggestion, ok = errorx.ExtractProperty(err, utils.ErrPropSuggestion)
	require.True(t, ok)
	require.Equal(t, `The directory you specified in the topology file is:
  Directory: data directory /home/tidb/deploy/pd-2234
  Component: pump 172.16.5.138

It conflicts to a directory in the existing cluster:
  Existing Cluster Name: topo2
  Existing Directory:    deploy directory /home/tidb/deploy/pd-2234
  Existing Component:    pd 172.16.5.138

Please change to use another directory or another host.`, suggestion)
}

func TestRelativePathDetect(t *testing.T) {
	servers := map[string]string{
		"monitoring_servers":   "rule_dir",
		"grafana_servers":      "dashboard_dir",
		"alertmanager_servers": "config_file",
	}
	paths := map[string]bool{
		"/an/absolute/path":   true,
		"an/relative/path":    false,
		"./an/relative/path":  false,
		"../an/relative/path": false,
	}

	for server, field := range servers {
		for p, expectNil := range paths {
			topo5 := Specification{}
			content := fmt.Sprintf(`
%s:
  - host: 1.1.1.1
    %s: %s
`, server, field, p)
			if expectNil {
				require.Nil(t, yaml.Unmarshal([]byte(content), &topo5))
			} else {
				require.NotNil(t, yaml.Unmarshal([]byte(content), &topo5))
			}
		}
	}
}

func TestPrometheusExternalLabels(t *testing.T) {
	topo := Specification{}
	// Allow user-defined external labels to round-trip into the monitor spec.
	err := yaml.Unmarshal([]byte(`
monitoring_servers:
  - host: 172.16.5.138
    external_labels:
      environment: production
      region: us-east-1
`), &topo)
	require.NoError(t, err)
	require.Equal(t, "production", topo.Monitors[0].ExternalLabels["environment"])
	require.Equal(t, "us-east-1", topo.Monitors[0].ExternalLabels["region"])

	topo = Specification{}
	// Reject labels reserved for the built-in TiUP metadata.
	err = yaml.Unmarshal([]byte(`
monitoring_servers:
  - host: 172.16.5.138
    external_labels:
      cluster: production
`), &topo)
	require.Error(t, err)
	require.Equal(t, "monitoring_servers:172.16.5.138.external_labels contains reserved label 'cluster'", err.Error())

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
monitoring_servers:
  - host: 172.16.5.138
    external_labels:
      monitor: production
`), &topo)
	require.Error(t, err)
	require.Equal(t, "monitoring_servers:172.16.5.138.external_labels contains reserved label 'monitor'", err.Error())

	topo = Specification{}
	// Reject names that violate the Prometheus label naming rules.
	err = yaml.Unmarshal([]byte(`
monitoring_servers:
  - host: 172.16.5.138
    external_labels:
      1region: us-east-1
`), &topo)
	require.Error(t, err)
	require.Equal(t, "monitoring_servers:172.16.5.138.external_labels contains invalid label name '1region'", err.Error())

	topo = Specification{}
	// Reject the Prometheus-reserved __* label namespace.
	err = yaml.Unmarshal([]byte(`
monitoring_servers:
  - host: 172.16.5.138
    external_labels:
      __replica__: tiflash
`), &topo)
	require.Error(t, err)
	require.Equal(t, "monitoring_servers:172.16.5.138.external_labels contains invalid label name '__replica__': labels starting with '__' are reserved", err.Error())
}

func TestTiKVLocationLabelsCheck(t *testing.T) {
	// 2 tikv on different host
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
tikv_servers:
  - host: 172.16.5.140
    port: 20160
    status_port: 20180
  - host: 172.16.5.139
    port: 20160
    status_port: 20180
`), &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels(nil, &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels([]string{}, &topo)
	require.NoError(t, err)

	// 2 tikv on the same host without label
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
tikv_servers:
  - host: 172.16.5.140
    port: 20160
    status_port: 20180
  - host: 172.16.5.140
    port: 20161
    status_port: 20181
`), &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels(nil, &topo)
	require.Error(t, err)

	// 2 tikv on the same host with unacquainted label
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
tikv_servers:
  - host: 172.16.5.140
    port: 20160
    status_port: 20180
    config:
      server.labels: { zone: "zone1", host: "172.16.5.140" }
  - host: 172.16.5.140
    port: 20161
    status_port: 20181
    config:
      server.labels: { zone: "zone1", host: "172.16.5.140" }
`), &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels(nil, &topo)
	require.Error(t, err)

	// 2 tikv on the same host with correct label
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
tikv_servers:
  - host: 172.16.5.140
    port: 20160
    status_port: 20180
    config:
      server.labels: { zone: "zone1", host: "172.16.5.140" }
  - host: 172.16.5.140
    port: 20161
    status_port: 20181
    config:
      server.labels: { zone: "zone1", host: "172.16.5.140" }
`), &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels([]string{"zone", "host"}, &topo)
	require.NoError(t, err)

	// 2 tikv on the same host with different config style
	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
tikv_servers:
  - host: 172.16.5.140
    port: 20160
    status_port: 20180
    config:
      server:
        labels: { zone: "zone1", host: "172.16.5.140" }
  - host: 172.16.5.140
    port: 20161
    status_port: 20181
    config:
      server.labels:
        zone: "zone1"
        host: "172.16.5.140"
`), &topo)
	require.NoError(t, err)
	err = CheckTiKVLabels([]string{"zone", "host"}, &topo)
	require.NoError(t, err)
}

func TestCountDirMultiPath(t *testing.T) {
	topo := Specification{}

	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
tiflash_servers:
  - host: 172.19.0.104
    data_dir: "/home/tidb/birdstorm/data1, /home/tidb/birdstorm/data3"
`), &topo)
	require.NoError(t, err)
	cnt := topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data1")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data2")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data3")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm")
	require.Equal(t, 2, cnt)

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
tiflash_servers:
  - host: 172.19.0.104
    data_dir: "birdstorm/data1,/birdstorm/data3"
`), &topo)
	require.NoError(t, err)
	cnt = topo.CountDir("172.19.0.104", "/home/test1/test-deploy/tiflash-9000/birdstorm/data1")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/birdstorm/data3")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data3")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/test1/test-deploy/tiflash-9000/birdstorm/data3")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.19.0.104", "/birdstorm")
	require.Equal(t, 1, cnt)

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
tiflash_servers:
  - host: 172.19.0.104
    data_dir: /data1 # this is ignored
    config:
      # test with these paths
      storage.main.dir: [ /home/tidb/birdstorm/data1,/home/tidb/birdstorm/data3]
`), &topo)
	require.NoError(t, err)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data1")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data2")
	require.Equal(t, 0, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm/data3")
	require.Equal(t, 1, cnt)
	cnt = topo.CountDir("172.19.0.104", "/home/tidb/birdstorm")
	require.Equal(t, 2, cnt)
}

func TestDirectoryConflictsWithMultiDir(t *testing.T) {
	topo := Specification{}

	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    data_dir: " /test-1, /test-2"
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-2"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "directory conflict for '/test-2' between 'tiflash_servers:172.16.5.138.data_dir' and 'pd_servers:172.16.5.138.data_dir'", err.Error())

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    data_dir: "/test-1,/test-1"
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-2"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "directory conflict for '/test-1' between 'tiflash_servers:172.16.5.138.data_dir' and 'tiflash_servers:172.16.5.138.data_dir'", err.Error())
}

func TestDirectoryConflictsWithTiFlashMultiDir2(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    data_dir: "/test-1" # this will be overwrite by storage.main.dir
    config:
      storage.main.dir: [ /test-1, /test-2]
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-2"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "directory conflict for '/test-2' between 'tiflash_servers:172.16.5.138.data_dir' and 'pd_servers:172.16.5.138.data_dir'", err.Error())

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    # this will be overwrite by storage.main.dir
    data_dir: "/test-1"
    config:
      storage.main.dir: [ /test-2, /test-2 ] # conflict inside
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-1"
`), &topo)
	require.Error(t, err)
	require.Equal(t, "directory conflict for '/test-2' between 'tiflash_servers:172.16.5.138.config.storage.main.dir' and 'tiflash_servers:172.16.5.138.config.storage.main.dir'", err.Error())

	err = yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: "test-deploy"
  data_dir: "test-data"
tiflash_servers:
  - host: 172.16.5.138
    data_dir: "/test-1" # this will be overwrite by storage.main.dir
    config:
      # no conflict between main and latest
      storage.main.dir: [ /test-1, /test-2]
      storage.latest.dir: [ /test-1, /test-2]
pd_servers:
  - host: 172.16.5.138
    data_dir: "/test-3"
`), &topo)
	require.NoError(t, err)
}

func TestPdServerWithSameName(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
    name: name1
  - host: 172.16.5.139
    perr_port: 1234
    name: name2
`), &topo)
	require.NoError(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
  pd_servers:
  - host: 172.16.5.138
    peer_port: 1234
    name: name1
  - host: 172.16.5.139
    perr_port: 1234
    name: name1
`), &topo)
	require.Error(t, err)
	require.Equal(t, "component pd_servers.name is not supported duplicated, the name name1 is duplicated", err.Error())
}

func TestInvalidPort(t *testing.T) {
	topo := Specification{}

	err := yaml.Unmarshal([]byte(`
global:
  ssh_port: 65536
`), &topo)
	require.Error(t, err)
	require.Equal(t, "`global` of ssh_port=65536 is invalid, port should be in the range [1, 65535]", err.Error())

	err = yaml.Unmarshal([]byte(`
global:
  ssh_port: 655
tidb_servers:
  - host: 172.16.5.138
    port: -1
`), &topo)
	require.Error(t, err)
	require.Equal(t, "`tidb_servers` of port=-1 is invalid, port should be in the range [1, 65535]", err.Error())

	err = yaml.Unmarshal([]byte(`
monitored:
    node_exporter_port: 102400
`), &topo)
	require.Error(t, err)
	require.Equal(t, "`monitored` of node_exporter_port=102400 is invalid, port should be in the range [1, 65535]", err.Error())
}

func TestInvalidUserGroup(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  user: helloworldtidb-_-_
  group: wor_l-d
`), &topo)
	require.NoError(t, err)
	require.Equal(t, "helloworldtidb-_-_", topo.GlobalOptions.User)
	require.Equal(t, "wor_l-d", topo.GlobalOptions.Group)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  user: ../hello
`), &topo)
	require.Error(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  user: hel.lo
`), &topo)
	require.Error(t, err)

	topo = Specification{}
	err = yaml.Unmarshal([]byte(`
global:
  group: hello123456789012
`), &topo)
	require.Error(t, err)
}

func TestMissingGroup(t *testing.T) {
	topo := Specification{}
	err := yaml.Unmarshal([]byte(`
global:
  user: tidb
`), &topo)
	require.NoError(t, err)
	require.Equal(t, "tidb", topo.GlobalOptions.User)
	require.Equal(t, "", topo.GlobalOptions.Group)
}

func TestLogDirUnderDataDir(t *testing.T) {
	topo := Specification{}
	clsList := make(map[string]Metadata)

	err := yaml.Unmarshal([]byte(`
global:
  user: "test1"
  ssh_port: 220
  deploy_dir: deploy
  data_dir: data
tikv_servers:
  - host: n1
    port: 32160
    status_port: 32180
    log_dir: "/home/tidb6wu/tidb1-data/tikv-32160/log"
    data_dir: "/home/tidb6wu/tidb1-data/tikv-32160"
`), &topo)
	require.NoError(t, err)
	err = CheckClusterDirConflict(clsList, "topo", &topo)
	require.Error(t, err)
	require.Equal(t, "spec.deploy.dir_overlap: Deploy directory overlaps to another instance", err.Error())
	suggestion, ok := errorx.ExtractProperty(err, utils.ErrPropSuggestion)
	require.True(t, ok)
	require.Equal(t, `The directory you specified in the topology file is:
  Directory: data directory /home/tidb6wu/tidb1-data/tikv-32160
  Component: tikv n1

It overlaps to another instance:
  Other Directory: log directory /home/tidb6wu/tidb1-data/tikv-32160/log
  Other Component: tikv n1

Please modify the topology file and try again.`, suggestion)

	goodTopos := []string{
		`
tikv_servers:
  - host: n1
    log_dir: 'tikv-data/log'
  - host: n2
    data_dir: 'tikv-data'
`,
		`
tikv_servers:
  - host: n1
    port: 32160
    status_port: 32180
    log_dir: "/home/tidb6wu/tidb1-data/tikv-32160-log"
    data_dir: "/home/tidb6wu/tidb1-data/tikv-32160"
`,
		`
monitored:
  node_exporter_port: 9100
  blackbox_exporter_port: 9115
  deploy_dir: /data/deploy/monitor-9100
  data_dir: /data/deploy/monitor-9100
  log_dir: /data/deploy/monitor-9100/log
pd_servers:
  - host: n0
    name: pd0
    imported: true
    deploy_dir: /data/deploy
    data_dir: /data/deploy/data.pd
    log_dir: /data/deploy/log
  - host: n1
    name: pd1
    log_dir: "/data/deploy/pd-2379/log"
    data_dir: "/data/pd-2379"
    deploy_dir: "/data/deploy/pd-2379"
cdc_servers:
  - host: n1
    port: 8300
    deploy_dir: /data/deploy/ticdc-8300
    data_dir: /data1/ticdc-8300
    log_dir: /data/deploy/ticdc-8300/log
`,
	}
	for _, s := range goodTopos {
		err = yaml.Unmarshal([]byte(s), &topo)
		require.NoError(t, err)
		err = CheckClusterDirConflict(clsList, "topo", &topo)
		require.NoError(t, err)
	}

	overlapTopos := []string{
		`
tikv_servers:
  - host: n1
    log_dir: 'tikv-data/log'
    data_dir: 'tikv-data'
`,
		`
tikv_servers:
  - host: n1
    log_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log'
    data_dir: '/home/tidb6wu/tidb1-data/tikv-32160'
`,
		`
tikv_servers:
  - host: n1
    log_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log'
    data_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log/data'
`,
		`
tikv_servers:
  - host: n1
    log_dir: 'tikv-log'
    data_dir: 'tikv-log/data'
`,
		`
tikv_servers:
  - host: n1
    data_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log'

tidb_servers:
  - host: n1
    log_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log/data'
`,
		`
tikv_servers:
  - host: n1
    log_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log'

tidb_servers:
  - host: n1
    log_dir: '/home/tidb6wu/tidb1-data/tikv-32160/log/log'
`,
		`
tikv_servers:
  - host: n1
    data_dir: '/home/tidb6wu/tidb1-data/tikv-32160/data'

pd_servers:
  - host: n1
    data_dir: '/home/tidb6wu/tidb1-data/tikv-32160'
`,
		`
global:
  user: "test1"
  deploy_dir: deploy
  data_dir: data
tikv_servers:
  - host: n1
    log_dir: "/home/test1/deploy/tikv-20160/ddd/log"
    data_dir: "ddd"
`,
		`
global:
  user: "test1"
  deploy_dir: deploy
  data_dir: data
tikv_servers:
  - host: n1
    log_dir: "log"
    data_dir: "/home/test1/deploy/tikv-20160/log/data"
`,
	}

	for _, s := range overlapTopos {
		err = yaml.Unmarshal([]byte(s), &topo)
		require.NoError(t, err)
		err = CheckClusterDirConflict(clsList, "topo", &topo)
		require.Error(t, err)
		require.Equal(t, "spec.deploy.dir_overlap: Deploy directory overlaps to another instance", err.Error())
	}
}
