Initial commit

waitfor_test.go

@@ -0,0 +1,58 @@
+package waitfor
+
+import (
+	"net"
+	"testing"
+	"time"
+
+	"git.fuyu.moe/Fuyu/assert"
+)
+
+func TestDeadline(t *testing.T) {
+	assert := assert.New(t)
+
+	start := time.Now()
+	ok := Try(`tcp`, `127.0.0.1:65398`, time.Microsecond*100, time.Millisecond)
+
+	assert.False(ok)
+	assert.Eq(int64(1), time.Since(start).Milliseconds())
+
+	start = time.Now()
+	ok = Try(`tcp`, `127.0.0.1:65398`, time.Microsecond*100, 0)
+
+	assert.False(ok)
+	assert.Eq(int64(0), time.Since(start).Milliseconds())
+}
+
+func TestSuccess(t *testing.T) {
+	assert := assert.New(t)
+
+	conn, err := net.Listen(`tcp`, `127.0.0.1:65398`)
+	assert.NoError(err)
+	defer conn.Close()
+
+	start := time.Now()
+	ok := Try(`tcp`, `127.0.0.1:65398`, time.Microsecond*100, time.Millisecond*10)
+
+	assert.True(ok)
+	assert.True(time.Since(start).Milliseconds() < 10)
+}
+
+func TestInfiniteLimit(t *testing.T) {
+	assert := assert.New(t)
+
+	var conn net.Listener
+	go func() {
+		time.Sleep(time.Millisecond)
+		var err error
+		conn, err = net.Listen(`tcp`, `127.0.0.1:65398`)
+		assert.NoError(err)
+	}()
+
+	start := time.Now()
+	ok := Try(`tcp`, `127.0.0.1:65398`, time.Microsecond*100, time.Millisecond*-1)
+
+	assert.True(ok)
+	assert.True(time.Since(start).Milliseconds() < 2)
+	conn.Close()
+}