Skip to content

Commit

Permalink
Handle errors from flag setting and tests
Browse files Browse the repository at this point in the history
Signed-off-by: carter.fendley <[email protected]>
  • Loading branch information
CarterFendley committed Oct 17, 2024
1 parent 5a94260 commit 264d809
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 5 deletions.
7 changes: 5 additions & 2 deletions backend/src/v2/cmd/driver/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -77,9 +77,12 @@ func main() {
flag.Parse()

glog.Infof("Setting log level to: '%s'", *logLevel)
flag.Set("v", *logLevel)
err := flag.Set("v", *logLevel)
if err != nil {
glog.Warningf("Failed to set log level: %s", err.Error())
}

err := drive()
err = drive()
if err != nil {
glog.Exitf("%v", err)
}
Expand Down
5 changes: 4 additions & 1 deletion backend/src/v2/cmd/launcher-v2/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,10 @@ func run() error {
ctx := context.Background()

glog.Infof("Setting log level to: '%s'", *logLevel)
flag.Set("v", *logLevel)
err := flag.Set("v", *logLevel)
if err != nil {
glog.Warningf("Failed to set log level: %s", err.Error())
}

if *copy != "" {
// copy is used to copy this binary to a shared volume
Expand Down
12 changes: 10 additions & 2 deletions backend/src/v2/compiler/argocompiler/argo_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -79,10 +79,18 @@ func Test_argo_compiler(t *testing.T) {
if tt.envVars != nil {
for _, envVar := range tt.envVars {
parts := strings.Split(strings.ReplaceAll(envVar, " ", ""), "=")
os.Setenv(parts[0], parts[1])
err := os.Setenv(parts[0], parts[1])
if err != nil {
t.Fatalf("Failed to set environment variable '%s' with error: %s", parts[0], err.Error())
}

// Unset after test cases has ended
defer os.Unsetenv(parts[0])
defer func() {
err := os.Unsetenv(parts[0])
if err != nil {
t.Fatalf("Failed to unset env variable '%s' with error: %s", parts[0], err.Error())
}
}()
}
}
if *update {
Expand Down

0 comments on commit 264d809

Please sign in to comment.