From eadee97e5e67ad5e09e81b2a0c37b2944faf46ff Mon Sep 17 00:00:00 2001
From: Arsen Musayelyan <moussaelianarsen@gmail.com>
Date: Thu, 26 May 2022 13:32:19 -0700
Subject: [PATCH] Add tests

---
 lrpc_test.go | 134 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100644 lrpc_test.go

diff --git a/lrpc_test.go b/lrpc_test.go
new file mode 100644
index 0000000..22da4ee
--- /dev/null
+++ b/lrpc_test.go
@@ -0,0 +1,134 @@
+package lrpc_test
+
+import (
+	"context"
+	"encoding/gob"
+	"net"
+	"testing"
+
+	"go.arsenm.dev/lrpc/client"
+	"go.arsenm.dev/lrpc/codec"
+	"go.arsenm.dev/lrpc/server"
+)
+
+type Arith struct{}
+
+func (Arith) Add(ctx *server.Context, in [2]int) int {
+	return in[0] + in[1]
+}
+
+func (Arith) Mul(ctx *server.Context, in [2]int) int {
+	return in[0] * in[1]
+}
+
+func (Arith) Div(ctx *server.Context, in [2]int) int {
+	return in[0] / in[1]
+}
+
+func (Arith) Sub(ctx *server.Context, in [2]int) int {
+	return in[0] - in[1]
+}
+
+func TestCalls(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// Create new network pipe
+	sConn, cConn := net.Pipe()
+
+	s := server.New()
+	defer s.Close()
+	// Register Arith for RPC
+	s.Register(Arith{})
+	// Serve the pipe connection using default codec
+	go s.ServeConn(ctx, sConn, codec.Default)
+
+	// Create new client using default codec
+	c := client.New(cConn, codec.Default)
+	defer c.Close()
+
+	// Call Arith.Add()
+	var add int
+	err := c.Call(ctx, "Arith", "Add", [2]int{5, 5}, &add)
+	if err != nil {
+		t.Error(err)
+	}
+
+	// Call Arith.Sub()
+	var sub int
+	err = c.Call(ctx, "Arith", "Sub", [2]int{5, 5}, &sub)
+	if err != nil {
+		t.Error(err)
+	}
+
+	// Call Arith.Mul()
+	var mul int
+	err = c.Call(ctx, "Arith", "Mul", [2]int{5, 5}, &mul)
+	if err != nil {
+		t.Error(err)
+	}
+
+	// Call Arith.Div()
+	var div int
+	err = c.Call(ctx, "Arith", "Div", [2]int{5, 5}, &div)
+	if err != nil {
+		t.Error(err)
+	}
+
+	if add != 10 {
+		t.Errorf("add: expected 10, got %d", add)
+	}
+
+	if sub != 0 {
+		t.Errorf("sub: expected 0, got %d", sub)
+	}
+
+	if mul != 25 {
+		t.Errorf("mul: expected 25, got %d", mul)
+	}
+
+	if div != 1 {
+		t.Errorf("div: expected 1, got %d", div)
+	}
+}
+
+func TestCodecs(t *testing.T) {
+	// Register the 2-integer array for gob
+	gob.Register([2]int{})
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// Create function to test each codec
+	testCodec := func(cf codec.CodecFunc, name string) {
+		// Create network pipe
+		sConn, cConn := net.Pipe()
+
+		s := server.New()
+		defer s.Close()
+		// Register Arith for RPC
+		s.Register(Arith{})
+		// Serve the pipe connection using provided codec
+		go s.ServeConn(ctx, sConn, cf)
+
+		// Create new client using provided codec
+		c := client.New(cConn, cf)
+		defer c.Close()
+
+		// Call Arith.Add()
+		var add int
+		err := c.Call(ctx, "Arith", "Add", [2]int{2, 2}, &add)
+		if err != nil {
+			t.Errorf("codec/%s: %v", name, err)
+		}
+
+		if add != 4 {
+			t.Errorf("codec/%s: add: expected 4, got %d", name, add)
+		}
+	}
+
+	// Test all codecs
+	testCodec(codec.Msgpack, "msgpack")
+	testCodec(codec.JSON, "json")
+	testCodec(codec.Gob, "gob")
+}