good morning!!!!

Skip to content
Snippets Groups Projects
Unverified Commit fb30712a authored by Toyam Cox's avatar Toyam Cox Committed by Anmol Sethi
Browse files

netconn: Fix SetDeadline methods

Closes #111
Closes #112
parent 3e63f826
Branches
Tags
No related merge requests found
......@@ -134,11 +134,19 @@ func (c *netConn) SetDeadline(t time.Time) error {
}
func (c *netConn) SetWriteDeadline(t time.Time) error {
if t.IsZero() {
c.writeTimer.Stop()
} else {
c.writeTimer.Reset(t.Sub(time.Now()))
}
return nil
}
func (c *netConn) SetReadDeadline(t time.Time) error {
if t.IsZero() {
c.readTimer.Stop()
} else {
c.readTimer.Reset(t.Sub(time.Now()))
}
return nil
}
......@@ -130,6 +130,8 @@ func TestHandshake(t *testing.T) {
nc := websocket.NetConn(c)
defer nc.Close()
nc.SetWriteDeadline(time.Time{})
time.Sleep(1)
nc.SetWriteDeadline(time.Now().Add(time.Second * 15))
for i := 0; i < 3; i++ {
......@@ -153,6 +155,8 @@ func TestHandshake(t *testing.T) {
nc := websocket.NetConn(c)
defer nc.Close()
nc.SetReadDeadline(time.Time{})
time.Sleep(1)
nc.SetReadDeadline(time.Now().Add(time.Second * 15))
read := func() error {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment