Jak przetestować io.writer w golangu?

Question

Ostatnio mam nadzieję napisać test jednostkowy dla golanga. Funkcja jest jak poniżej.

func (s *containerStats) Display(w io.Writer) error { 
    fmt.Fprintf(w, "%s %s\n", "hello", "world") 
    return nil 
} 

Jak mogę przetestować wynik "Func Display" to "witaj świecie"?

Answer 1

Możesz po prostu przekazać własne io.Writer i sprawdzić, co zostanie zapisane w nim odpowiada oczekiwaniom. bytes.Buffer jest dobrym wyborem dla takiego io.Writer, ponieważ po prostu przechowuje dane wyjściowe w swoim buforze.

func TestDisplay(t *testing.T) { 
    s := newContainerStats() // Replace this the appropriate constructor 
    var b bytes.Buffer 
    if err := s.Display(&b); err != nil { 
     t.Fatalf("s.Display() gave error: %s", err) 
    } 
    got := b.String() 
    want := "hello world\n" 
    if got != want { 
     t.Errorf("s.Display() = %q, want %q", got, want) 
    } 
}