57 lines
774 B
Go
57 lines
774 B
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSocket(t *testing.T) {
|
||
|
|
fileName := "./test_server.log"
|
||
|
|
|
||
|
|
os.Remove(fileName)
|
||
|
|
|
||
|
|
s, err := NewServer(fileName, "tcp", "127.0.0.1:11183")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
go s.Run()
|
||
|
|
defer s.Close()
|
||
|
|
|
||
|
|
var h *SocketHandler
|
||
|
|
h, err = NewSocketHandler("tcp", "127.0.0.1:11183")
|
||
|
|
|
||
|
|
_, err = h.Write([]byte("hello world"))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
time.Sleep(1 * time.Second)
|
||
|
|
|
||
|
|
s.Close()
|
||
|
|
|
||
|
|
var f *os.File
|
||
|
|
f, err = os.Open(fileName)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
defer f.Close()
|
||
|
|
|
||
|
|
buf := make([]byte, 64)
|
||
|
|
var n int
|
||
|
|
n, err = f.Read(buf)
|
||
|
|
if err != nil && err != io.EOF {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
buf = buf[0:n]
|
||
|
|
|
||
|
|
if string(buf) != "hello world\n" {
|
||
|
|
t.Fatal(string(buf))
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Remove(fileName)
|
||
|
|
}
|