plump infra data into tests

This commit is contained in:
William Banfield
2022-12-01 11:17:03 -05:00
parent de20e3d224
commit 1a2afa62c5
4 changed files with 40 additions and 5 deletions
+19 -2
View File
@@ -73,7 +73,14 @@ func loadTestnet(t *testing.T) e2e.Testnet {
if !filepath.IsAbs(manifestFile) {
manifestFile = filepath.Join("..", manifestFile)
}
ifdType := os.Getenv("INFRASTRUCTURE_DATA")
ifdFile := os.Getenv("INFRASTRUCTURE_FILE")
if ifdFile == "" && ifdType != "docker" {
t.Skip("INFRASTRUCTURE_DATA not set and INFRASTRUCTURE_TYPE is not docker")
}
if !filepath.IsAbs(ifdFile) {
manifestFile = filepath.Join("..", manifestFile)
}
testnetCacheMtx.Lock()
defer testnetCacheMtx.Unlock()
if testnet, ok := testnetCache[manifestFile]; ok {
@@ -81,7 +88,17 @@ func loadTestnet(t *testing.T) e2e.Testnet {
}
m, err := e2e.LoadManifest(manifestFile)
require.NoError(t, err)
ifd, err := e2e.NewDockerInfrastructureData(m)
var ifd e2e.InfrastructureData
switch ifdType {
case "docker":
ifd, err = e2e.NewDockerInfrastructureData(m)
require.NoError(t, err)
case "digital-ocean":
ifd, err = e2e.InfrastructureDataFromFile(manifestFile)
require.NoError(t, err)
default:
}
require.NoError(t, err)
testnet, err := e2e.LoadTestnet(m, manifestFile, ifd)