mirror of
https://github.com/golang/go
synced 2024-10-07 11:21:22 -06:00
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
|
// Copyright 2012 The Go Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package net
|
||
|
|
||
|
import (
|
||
|
"runtime"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestWriteToUDP(t *testing.T) {
|
||
|
if runtime.GOOS == "plan9" {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
l, err := ListenPacket("udp", "127.0.0.1:0")
|
||
|
if err != nil {
|
||
|
t.Fatalf("Listen failed: %v", err)
|
||
|
}
|
||
|
defer l.Close()
|
||
|
|
||
|
testWriteToConn(t, l.LocalAddr().String())
|
||
|
testWriteToPacketConn(t, l.LocalAddr().String())
|
||
|
}
|
||
|
|
||
|
func testWriteToConn(t *testing.T, raddr string) {
|
||
|
c, err := Dial("udp", raddr)
|
||
|
if err != nil {
|
||
|
t.Fatalf("Dial failed: %v", err)
|
||
|
}
|
||
|
defer c.Close()
|
||
|
|
||
|
ra, err := ResolveUDPAddr("udp", raddr)
|
||
|
if err != nil {
|
||
|
t.Fatalf("ResolveUDPAddr failed: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.(*UDPConn).WriteToUDP([]byte("Connection-oriented mode socket"), ra)
|
||
|
if err == nil {
|
||
|
t.Fatal("WriteToUDP should be failed")
|
||
|
}
|
||
|
if err != nil && err.(*OpError).Err != ErrWriteToConnected {
|
||
|
t.Fatalf("WriteToUDP should be failed as ErrWriteToConnected: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.(*UDPConn).WriteTo([]byte("Connection-oriented mode socket"), ra)
|
||
|
if err == nil {
|
||
|
t.Fatal("WriteTo should be failed")
|
||
|
}
|
||
|
if err != nil && err.(*OpError).Err != ErrWriteToConnected {
|
||
|
t.Fatalf("WriteTo should be failed as ErrWriteToConnected: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.Write([]byte("Connection-oriented mode socket"))
|
||
|
if err != nil {
|
||
|
t.Fatalf("Write failed: %v", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func testWriteToPacketConn(t *testing.T, raddr string) {
|
||
|
c, err := ListenPacket("udp", "127.0.0.1:0")
|
||
|
if err != nil {
|
||
|
t.Fatalf("ListenPacket failed: %v", err)
|
||
|
}
|
||
|
defer c.Close()
|
||
|
|
||
|
ra, err := ResolveUDPAddr("udp", raddr)
|
||
|
if err != nil {
|
||
|
t.Fatalf("ResolveUDPAddr failed: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.(*UDPConn).WriteToUDP([]byte("Connection-less mode socket"), ra)
|
||
|
if err != nil {
|
||
|
t.Fatalf("WriteToUDP failed: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.WriteTo([]byte("Connection-less mode socket"), ra)
|
||
|
if err != nil {
|
||
|
t.Fatalf("WriteTo failed: %v", err)
|
||
|
}
|
||
|
|
||
|
_, err = c.(*UDPConn).Write([]byte("Connection-less mode socket"))
|
||
|
if err == nil {
|
||
|
t.Fatal("Write should be failed")
|
||
|
}
|
||
|
}
|