test-infra-definitions/resources/gcp/environmentDefaults.go at c7492899b2a1b1ae71c91ae4c9d7ebc3119b2258 · DataDog/test-infra-definitions · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package gcp

const (
	agentSandboxEnv = "gcp/agent-sandbox"
	agentQaEnv      = "gcp/agent-qa"
)

type environmentDefault struct {
	gcp     gcpProvider
	ddInfra ddInfra
}

type gcpProvider struct {
	project string
	region  string
	zone    string
}

type ddInfra struct {
	defaultInstanceType     string
	defaultNetworkName      string
	defaultSubnetName       string
	defaultVMServiceAccount string
	gke                     ddInfraGKE
	openshift               ddInfraOpenShift
}

type ddInfraGKE struct {
	autopilot bool
}

type ddInfraOpenShift struct {
	nestedVirtualization bool
}

func getEnvironmentDefault(envName string) environmentDefault {
	switch envName {
	case agentSandboxEnv:
		return agentSandboxDefault()
	case agentQaEnv:
		return agentQaDefault()
	default:
		panic("Unknown environment: " + envName)
	}
}

func agentSandboxDefault() environmentDefault {
	return environmentDefault{
		gcp: gcpProvider{
			project: "datadog-agent-sandbox",
			region:  "us-central1",
			zone:    "us-central1-a",
		},
		ddInfra: ddInfra{
			defaultInstanceType:     "e2-standard-2",
			defaultNetworkName:      "datadog-agent-sandbox-us-central1",
			defaultSubnetName:       "datadog-agent-sandbox-us-central1-private",
			defaultVMServiceAccount: "vmserviceaccount@datadog-agent-sandbox.iam.gserviceaccount.com",
			gke:                     ddInfraGKE{autopilot: false},
			openshift:               ddInfraOpenShift{nestedVirtualization: false},
		},
	}
}

func agentQaDefault() environmentDefault {
	return environmentDefault{
		gcp: gcpProvider{
			project: "datadog-agent-qa",
			region:  "us-central1",
			zone:    "us-central1-a",
		},
		ddInfra: ddInfra{
			defaultInstanceType:     "e2-standard-2",
			defaultNetworkName:      "datadog-agent-qa-us-central1",
			defaultSubnetName:       "datadog-agent-qa-us-central1-private",
			defaultVMServiceAccount: "vmserviceaccount@datadog-agent-qa.iam.gserviceaccount.com",
			gke:                     ddInfraGKE{autopilot: false},
			openshift:               ddInfraOpenShift{nestedVirtualization: false},
		},
	}
}