github-forks
/
naza
mirror of https://github.com/q191201771/naza
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

rm pkg mockserver

Browse Source
pull/2/head
q191201771 5 years ago
parent 34957541d3
commit ac12458266
4 changed files with 1 additions and 136 deletions
Show Stats Download Patch File Download Diff File

3
README.md
Unescape Escape View File

@ -43,8 +43,7 @@ pkg/ ...... 源码包
|-- unique/ ...... 对象唯一 ID
|-- nazamd5/ ...... md5 操作
|-- ic/ ...... 将整型切片压缩成二进制字节切片
|-- mockserver/ ...... 模拟一些服务端，用于快速测试其它代码
|-- mockwriter/ ...... 模拟 Writer 接口，用于快速测试其它代码
|-- fake/ ...... 实现一些常用的接口，辅助测试其它代码
demo/ ...... 示例相关的代码
bin/ ...... 可执行文件编译输出目录
```

44
pkg/mockserver/acceptserver.go
Unescape Escape View File

@ -1,44 +0,0 @@
// Copyright 2019, Chef. All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)
package mockserver
import (
"net"
"sync"
)
type MockAcceptServer struct {
l net.Listener
conns []net.Conn
m sync.Mutex
}
func (s *MockAcceptServer) Run(addr string) (err error) {
s.m.Lock()
s.l, err = net.Listen("tcp", addr)
s.m.Unlock()
if err != nil {
return
}
c, err := s.l.Accept()
if err != nil {
return
}
s.conns = append(s.conns, c)
return
}
func (s *MockAcceptServer) Dispose() {
s.m.Lock()
if s.l != nil {
s.l.Close()
}
s.m.Unlock()
}

41
pkg/mockserver/listenserver.go
Unescape Escape View File

@ -1,41 +0,0 @@
// Copyright 2019, Chef. All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)
package mockserver
import (
"net"
"time"
"github.com/q191201771/naza/pkg/nazalog"
)
// 建立一个server端的监听，在内部创建n个连接快速消耗掉listen队列，达到对外模拟不处理连接的情况
type MockListenServer struct {
l net.Listener
}
func (s *MockListenServer) Run(addr string) (err error) {
s.l, err = net.Listen("tcp", addr)
if err != nil {
return
}
for i := 0; ; i++ {
if _, err := net.DialTimeout("tcp", addr, time.Duration(200)*time.Millisecond); err != nil {
nazalog.Infof("Dial failed. i=%d, err=%+v", i, err)
break
}
}
return
}
func (s *MockListenServer) Dispose() {
_ = s.l.Close()
}

49
pkg/mockserver/mockserver_test.go
Unescape Escape View File

@ -1,49 +0,0 @@
// Copyright 2019, Chef. All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)
package mockserver
import (
"net"
"testing"
"time"
"github.com/q191201771/naza/pkg/assert"
)
var addr = ":10027"
func TestMockListenServer(t *testing.T) {
//var s MockListenServer
//err := s.Run(addr)
//assert.Equal(t, nil, err)
//defer s.Dispose()
//_, err = net.DialTimeout("tcp", addr, time.Duration(1000) * time.Millisecond)
//assert.IsNotNil(t, err)
}
func TestMockAcceptServer(t *testing.T) {
var s MockAcceptServer
var conns []net.Conn
go s.Run(addr)
for i := 0; i < 16; i++ {
c, err := net.DialTimeout("tcp", addr, time.Duration(1000)*time.Millisecond)
if err != nil {
break
}
//assert.Equal(t, nil, err)
conns = append(conns, c)
}
s.Dispose()
}
func TestCorner(t *testing.T) {
var s MockListenServer
err := s.Run("wrong addr")
assert.IsNotNil(t, err)
}
Write Preview
Loading…
Cancel
Save