DebTik/cmd/debtik-cli/main_test.go

package main

import (
	"bufio"
	"fmt"
	"net"
	"testing"
)

func TestAutomaticLoginWaitsForSuccessAndPreservesBufferedData(t *testing.T) {
	for _, accepted := range []bool{true, false} {
		t.Run(fmt.Sprint(accepted), func(t *testing.T) {
			client, server := net.Pipe()
			defer client.Close()
			defer server.Close()
			done := make(chan error, 1)
			go func() {
				r := bufio.NewReader(server)
				fmt.Fprint(server, "DebTik login: ")
				user, err := r.ReadString('\n')
				if err != nil {
					done <- err
					return
				}
				fmt.Fprint(server, "Password: ")
				password, err := r.ReadString('\n')
				if err != nil {
					done <- err
					return
				}
				if user != "admin\n" || password != "test-password\n" {
					done <- fmt.Errorf("wrong credentials sent")
					return
				}
				if accepted {
					_, err = fmt.Fprint(server, "Welcome\n[admin@router] > buffered")
				} else {
					_, err = fmt.Fprint(server, "Login failed: invalid credentials\n")
				}
				done <- err
			}()
			r := bufio.NewReader(client)
			err := autoLogin(client, r, "admin", "test-password")
			if (err == nil) != accepted {
				t.Fatalf("accepted=%v err=%v", accepted, err)
			}
			if accepted {
				buffered := make([]byte, 8)
				if _, err := r.Read(buffered); err != nil || string(buffered) != "buffered" {
					t.Fatal("buffered data was lost")
				}
			}
			if err := <-done; err != nil {
				t.Fatal(err)
			}
		})
	}
}