Comment tester efficacement la transmission d'arguments dans les fonctions Golang ?

Test de la transmission d'arguments dans Golang

Dans Golang, transmettre des arguments à une fonction est un moyen courant de fournir des entrées. Tester l'exactitude de la transmission des arguments est crucial pour garantir la fiabilité de votre code.

Le défi

Considérez l'extrait de code suivant :

package main

import (
    "flag"
    "fmt"
)

func main() {
    passArguments()
}

func passArguments() string {
    username := flag.String("user", "root", "Username for this server")
    flag.Parse()
    fmt.Printf("Your username is %q.", *username)

    usernameToString := *username
    return usernameToString
}

Lorsque vous passez l'argument -user=bla à ce programme, il affiche correctement "Votre nom d'utilisateur est 'bla'." Le défi est de créer un test qui vérifie ce comportement sans avoir à créer et exécuter manuellement le code à chaque fois.

La tentative

On pourrait尝试使用 os.Args pour fournir l'argument dans un test :

package main

import (
    "os"
    "testing"
)

func TestArgs(t *testing.T) {
    expected := "bla"
    os.Args = []string{"-user=bla"}

    actual := passArguments()

    if actual != expected {
        t.Errorf("Test failed, expected: '%s', got:  '%s'", expected, actual)
    }
}

Cependant, lors de l'exécution de ce test, le résultat est :

Your username is "root".
Your username is "root".
--- FAIL: TestArgs (0.00s)
    args_test.go:15: Test failed, expected: 'bla', got:  'root'
FAIL
coverage: 87.5% of statements
FAIL    tool    0.008s

La solution

Le problème est que os.Args est une variable globale, donc la valeur initiale de ["args"] est écrasée par le test, affectant d’autres parties du programme. Pour résoudre cela, nous pouvons conserver la valeur d'origine et la restaurer après le test :

oldArgs := os.Args
defer func() { os.Args = oldArgs }()
os.Args = []string{"cmd", "-user=bla"}

// ...

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!