mirror of
https://github.com/shouptech/tempgopher.git
synced 2026-02-03 16:49:42 +00:00
39 lines
849 B
Go
39 lines
849 B
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_ReadInput(t *testing.T) {
|
|
// Test basic input
|
|
buf := bytes.NewBufferString("foobar\n")
|
|
reader := bufio.NewReader(buf)
|
|
resp := ReadInput(reader, "")
|
|
assert.Equal(t, "foobar", resp)
|
|
|
|
// Test default input
|
|
buf = bytes.NewBufferString("\n")
|
|
reader = bufio.NewReader(buf)
|
|
resp = ReadInput(reader, "default")
|
|
assert.Equal(t, "default", resp)
|
|
|
|
// Test that a panic occurred
|
|
buf = bytes.NewBufferString("")
|
|
reader = bufio.NewReader(buf)
|
|
assert.Panics(t, func() { ReadInput(reader, "") })
|
|
}
|
|
|
|
func Test_ParsePort(t *testing.T) {
|
|
// Test that 600 is parsed succesfully
|
|
port, err := ParsePort(":600")
|
|
assert.Equal(t, uint16(600), port)
|
|
assert.Equal(t, nil, err)
|
|
|
|
// Test failure
|
|
port, err = ParsePort("")
|
|
assert.NotEqual(t, nil, err)
|
|
}
|