mirror of
https://github.com/juanfont/headscale.git
synced 2025-07-19 21:48:57 -04:00
05996a5048
We are already being punished by github actions, there seem to be little value in running all the tests for both databases, so only run a few key tests to check postgres isnt broken. Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
package main
|
|
|
|
//go:generate go run ./gh-action-integration-generator.go
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
func findTests() []string {
|
|
rgBin, err := exec.LookPath("rg")
|
|
if err != nil {
|
|
log.Fatalf("failed to find rg (ripgrep) binary")
|
|
}
|
|
|
|
args := []string{
|
|
"--regexp", "func (Test.+)\\(.*",
|
|
"../../integration/",
|
|
"--replace", "$1",
|
|
"--sort", "path",
|
|
"--no-line-number",
|
|
"--no-filename",
|
|
"--no-heading",
|
|
}
|
|
|
|
cmd := exec.Command(rgBin, args...)
|
|
var out bytes.Buffer
|
|
cmd.Stdout = &out
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
log.Fatalf("failed to run command: %s", err)
|
|
}
|
|
|
|
tests := strings.Split(strings.TrimSpace(out.String()), "\n")
|
|
return tests
|
|
}
|
|
|
|
func updateYAML(tests []string, jobName string, testPath string) {
|
|
testsForYq := fmt.Sprintf("[%s]", strings.Join(tests, ", "))
|
|
|
|
yqCommand := fmt.Sprintf(
|
|
"yq eval '.jobs.%s.strategy.matrix.test = %s' %s -i",
|
|
jobName,
|
|
testsForYq,
|
|
testPath,
|
|
)
|
|
cmd := exec.Command("bash", "-c", yqCommand)
|
|
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Printf("stdout: %s", stdout.String())
|
|
log.Printf("stderr: %s", stderr.String())
|
|
log.Fatalf("failed to run yq command: %s", err)
|
|
}
|
|
|
|
fmt.Printf("YAML file (%s) job %s updated successfully\n", testPath, jobName)
|
|
}
|
|
|
|
func main() {
|
|
tests := findTests()
|
|
|
|
quotedTests := make([]string, len(tests))
|
|
for i, test := range tests {
|
|
quotedTests[i] = fmt.Sprintf("\"%s\"", test)
|
|
}
|
|
|
|
// Define selected tests for PostgreSQL
|
|
postgresTestNames := []string{
|
|
"TestACLAllowUserDst",
|
|
"TestPingAllByIP",
|
|
"TestEphemeral2006DeletedTooQuickly",
|
|
"TestPingAllByIPManyUpDown",
|
|
"TestSubnetRouterMultiNetwork",
|
|
}
|
|
|
|
quotedPostgresTests := make([]string, len(postgresTestNames))
|
|
for i, test := range postgresTestNames {
|
|
quotedPostgresTests[i] = fmt.Sprintf("\"%s\"", test)
|
|
}
|
|
|
|
// Update both SQLite and PostgreSQL job matrices
|
|
updateYAML(quotedTests, "sqlite", "./test-integration.yaml")
|
|
updateYAML(quotedPostgresTests, "postgres", "./test-integration.yaml")
|
|
}
|