Refactored and corrected client's network data reading

[goircd.git] / client_test.go

diff --git a/client_test.go b/client_test.go
index f9e3887b5b0fc782d99b7df5ddb020161bea830a..3735e1f71034f8c3ebcd03f7efb63c522eff4c1a 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -1,6 +1,6 @@
 /*
 goircd -- minimalistic simple Internet Relay Chat (IRC) server
-Copyright (C) 2014 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2014-2015 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -15,6 +15,7 @@ GNU General Public License for more details.
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
+
 package main
 
 import (
@@ -46,10 +47,10 @@ func (conn *TestingConn) Read(b []byte) (n int, err error) {
        if msg == "" {
                return 0, conn
        }
-       for n, bt := range []byte(msg + CRLF) {
+       for n, bt := range append([]byte(msg), CRLF...) {
                b[n] = bt
        }
-       return len(msg), nil
+       return len(msg)+2, nil
 }
 
 type MyAddr struct{}
@@ -96,7 +97,8 @@ func (conn TestingConn) SetWriteDeadline(t time.Time) error {
 func TestNewClient(t *testing.T) {
        conn := NewTestingConn()
        sink := make(chan ClientEvent)
-       client := NewClient("foohost", conn)
+       host := "foohost"
+       client := NewClient(&host, conn)
        go client.Processor(sink)
 
        event := <-sink
@@ -123,7 +125,8 @@ func TestNewClient(t *testing.T) {
 // Test replies formatting
 func TestClientReplies(t *testing.T) {
        conn := NewTestingConn()
-       client := NewClient("foohost", conn)
+       host := "foohost"
+       client := NewClient(&host, conn)
        client.nickname = "мойник"
 
        client.Reply("hello")