Persist, renew and delete sessions, refactor storage package, move reusable packages to pkg

pkg/irc/conn_test.go

@@ -0,0 +1,233 @@
+package irc
+
+import (
+	"bufio"
+	"bytes"
+	"crypto/tls"
+	"log"
+	"net"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+var ircd *mockIrcd
+
+func init() {
+	initTestServer()
+}
+
+func initTestServer() {
+	ircd = &mockIrcd{
+		conn:       make(chan bool, 1),
+		connClosed: make(chan bool, 1),
+	}
+	ircd.start()
+}
+
+type mockIrcd struct {
+	conn       chan bool
+	connClosed chan bool
+}
+
+func (i *mockIrcd) start() {
+	ln, err := net.Listen("tcp", "127.0.0.1:45678")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	cert, err := tls.X509KeyPair(testCert, testKey)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	tlsConfig := &tls.Config{
+		Certificates: []tls.Certificate{cert},
+	}
+
+	lnTLS, err := tls.Listen("tcp", "127.0.0.1:45679", tlsConfig)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	go i.accept(ln)
+	go i.accept(lnTLS)
+}
+
+func (i *mockIrcd) accept(ln net.Listener) {
+	for {
+		conn, err := ln.Accept()
+		if err != nil {
+			return
+		}
+		go i.handle(conn)
+		i.conn <- true
+	}
+}
+
+func (i *mockIrcd) handle(conn net.Conn) {
+	buf := make([]byte, 1024)
+	for {
+		_, err := conn.Read(buf)
+		if err != nil {
+			i.connClosed <- true
+			return
+		}
+	}
+}
+
+func TestConnect(t *testing.T) {
+	c := testClient()
+	c.Connect("127.0.0.1:45678")
+	assert.Equal(t, c.Host, "127.0.0.1")
+	assert.Equal(t, c.Server, "127.0.0.1:45678")
+	waitConnAndClose(t, c)
+}
+
+func TestConnectTLS(t *testing.T) {
+	c := testClient()
+	c.TLS = true
+	c.TLSConfig = &tls.Config{
+		InsecureSkipVerify: true,
+	}
+	c.Connect("127.0.0.1:45679")
+	assert.Equal(t, c.Host, "127.0.0.1")
+	assert.Equal(t, c.Server, "127.0.0.1:45679")
+	waitConnAndClose(t, c)
+}
+
+func TestConnectDefaultPorts(t *testing.T) {
+	c := testClient()
+	c.Connect("127.0.0.1")
+	assert.Equal(t, "127.0.0.1:6667", c.Server)
+
+	c = testClient()
+	c.TLS = true
+	c.Connect("127.0.0.1")
+	assert.Equal(t, "127.0.0.1:6697", c.Server)
+}
+
+func TestWrite(t *testing.T) {
+	c, out := testClientSend()
+	c.write("test")
+	assert.Equal(t, "test\r\n", <-out)
+	c.Write("test")
+	assert.Equal(t, "test\r\n", <-out)
+	c.writef("test %d", 2)
+	assert.Equal(t, "test 2\r\n", <-out)
+	c.Writef("test %d", 2)
+	assert.Equal(t, "test 2\r\n", <-out)
+}
+
+func TestRecv(t *testing.T) {
+	c := testClient()
+	conn := &mockConn{hook: make(chan string, 16)}
+	c.conn = conn
+
+	buf := &bytes.Buffer{}
+	buf.WriteString("CMD\r\n")
+	buf.WriteString("PING :test\r\n")
+	buf.WriteString("001 foo\r\n")
+	c.reader = bufio.NewReader(buf)
+
+	c.sendRecv.Add(1)
+	go c.recv()
+
+	assert.Equal(t, "PONG :test\r\n", <-conn.hook)
+	assert.Equal(t, &Message{Command: "CMD"}, <-c.Messages)
+}
+
+func TestRecvTriggersReconnect(t *testing.T) {
+	c := testClient()
+	c.conn = &mockConn{}
+	c.reader = bufio.NewReader(&bytes.Buffer{})
+	done := make(chan struct{})
+	ok := false
+	go func() {
+		c.sendRecv.Add(1)
+		c.recv()
+		_, ok = <-c.reconnect
+		close(done)
+	}()
+
+	select {
+	case <-done:
+		assert.False(t, ok)
+		return
+
+	case <-time.After(100 * time.Millisecond):
+		t.Error("Reconnect not triggered")
+	}
+}
+
+func TestClose(t *testing.T) {
+	c := testClient()
+	close(c.quit)
+	ok := false
+	done := make(chan struct{})
+	go func() {
+		_, ok = <-c.Messages
+		close(done)
+	}()
+
+	c.run()
+
+	select {
+	case <-done:
+		assert.False(t, ok)
+		return
+
+	case <-time.After(100 * time.Millisecond):
+		t.Error("Channels not closed")
+	}
+}
+
+func waitConnAndClose(t *testing.T, c *Client) {
+	done := make(chan struct{})
+	quit := make(chan struct{})
+	go func() {
+		<-ircd.conn
+		quit <- struct{}{}
+		<-ircd.connClosed
+		close(done)
+	}()
+
+	for {
+		select {
+		case <-done:
+			return
+
+		case <-quit:
+			assert.True(t, c.Connected())
+			c.Quit()
+
+		case <-time.After(500 * time.Millisecond):
+			t.Error("Took too long")
+			return
+		}
+	}
+}
+
+var testCert = []byte(`-----BEGIN CERTIFICATE-----
+MIIB0zCCAX2gAwIBAgIJAI/M7BYjwB+uMA0GCSqGSIb3DQEBBQUAMEUxCzAJBgNV
+BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYDVQQKDBhJbnRlcm5ldCBX
+aWRnaXRzIFB0eSBMdGQwHhcNMTIwOTEyMjE1MjAyWhcNMTUwOTEyMjE1MjAyWjBF
+MQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1TdGF0ZTEhMB8GA1UECgwYSW50
+ZXJuZXQgV2lkZ2l0cyBQdHkgTHRkMFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBANLJ
+hPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wok/4xIA+ui35/MmNa
+rtNuC+BdZ1tMuVCPFZcCAwEAAaNQME4wHQYDVR0OBBYEFJvKs8RfJaXTH08W+SGv
+zQyKn0H8MB8GA1UdIwQYMBaAFJvKs8RfJaXTH08W+SGvzQyKn0H8MAwGA1UdEwQF
+MAMBAf8wDQYJKoZIhvcNAQEFBQADQQBJlffJHybjDGxRMqaRmDhX0+6v02TUKZsW
+r5QuVbpQhH6u+0UgcW0jp9QwpxoPTLTWGXEWBBBurxFwiCBhkQ+V
+-----END CERTIFICATE-----`)
+
+var testKey = []byte(`-----BEGIN RSA PRIVATE KEY-----
+MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo
+k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G
+6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N
+MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW
+SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T
+xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi
+D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g==
+-----END RSA PRIVATE KEY-----`)