Go에서 단위 테스트 명령줄 플래그 유효성 검사를 수행하는 방법은 무엇입니까?-Golang-php.cn

Go에서 단위 테스트 명령줄 플래그 유효성 검사를 수행하는 방법은 무엇입니까?

Mary-Kate Olsen

풀어 주다： 2024-11-05 21:29:02

원래의

873명이 탐색했습니다.

How to Unit Test Command Line Flag Validation in Go?

Go에서 명령줄 플래그를 단위 테스트하는 방법

소개:

단위 테스트는 명령줄 플래그 처리를 포함하여 코드의 신뢰성과 정확성을 보장하는 데 매우 중요합니다. 플래그 구문 분석은 Go 애플리케이션의 일반적인 작업이며 플래그가 올바르게 검증되었는지 확인하는 것이 중요합니다. 이 기사에서는 Go에서 명령줄 플래그를 단위 테스트하는 방법을 살펴보겠습니다.

단위 테스트 설정:

단위 테스트 플래그 유효성 검사를 위해 사용자 정의 플래그 값이 예상 열거형 내에 있는지 확인하는 플래그 유형입니다. 다음 예제 코드를 고려하십시오.

<code class="go">var formatType string

const (
    text = "text"
    json = "json"
    hash = "hash"
)

func init() {
    const (
        defaultFormat = "text"
        formatUsage   = "desired output format"
    )

    flag.StringVar(&amp;formatType, "format", defaultFormat, formatUsage)
    flag.StringVar(&amp;formatType, "f", defaultFormat, formatUsage+" (shorthand)")
}</code>

로그인 후 복사

사용자 정의 플래그 유효성 검사:

사용자 정의 플래그를 정의할 수 있습니다. 값 유형을 정의하여 형식 플래그를 유효성 검사할 수 있습니다.

<code class="go">type formatType string

func (f *formatType) String() string {
    return fmt.Sprint(*f)
}

func (f *formatType) Set(value string) error {
    if len(*f) > 0 &amp;&amp; *f != "text" {
        return errors.New("format flag already set")
    }
    if value != "text" &amp;&amp; value != "json" &amp;&amp; value != "hash" {
        return errors.New("Invalid Format Type")
    }
    *f = formatType(value)
    return nil
}</code>

로그인 후 복사

단위 테스트 생성:

이제 사용자 정의 플래그 유효성 검사를 확인하는 단위 테스트를 작성해 보겠습니다.

<code class="go">func TestValidateFormatFlag(t *testing.T) {
    testCases := []struct {
        input    string
        expected error
    }{
        {"text", nil},
        {"json", nil},
        {"hash", nil},
        {"invalid", errors.New("Invalid Format Type")},
    }

    for _, tc := range testCases {
        args := []string{"-format", tc.input}
        os.Args = args

        typeFlag = "text"
        flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
        flag.Var(&amp;typeFlag, "format", "Format type. Must be \"text\", \"json\" or \"hash\". Defaults to \"text\".")
        if err := flag.CommandLine.Parse(args[1:]); err != nil {
            t.Errorf("Error parsing flags: %v", err)
        }

        if tc.expected != nil && err == nil {
            t.Errorf("Expected error: %v, got nil", tc.expected)
        } else if tc.expected == nil && err != nil {
            t.Errorf("Unexpected error: %v", err)
        }
    }
}</code>

로그인 후 복사

이 테스트에서는 다양한 입력과 예상 오류가 포함된 일련의 테스트 사례를 설정합니다. os.Args를 사용하여 명령줄 인수를 모의하고 사용자 정의 FlagSet을 사용하여 플래그를 구문 분석합니다. 실제 오류와 각 테스트 사례의 예상 오류를 비교하여 플래그 유효성 검사가 의도한 대로 작동하는지 확인합니다.

플래그 테스트 실행 팁: