metadapta_test.go

@@ -25,38 +25,7 @@ func TestConnA(test *testing.T) {
 		"When the impostor is sus!",
 	}
 
-	network := "tcp"
-	addr := "localhost:7959"
-
-	// server
-	listener, err := net.Listen(network, addr)
-	if err != nil { test.Fatal(err) }
-	test.Cleanup(func() { listener.Close() })
-	go func() {
-		test.Log("SERVER listening")
-		conn, err := listener.Accept()
-		if err != nil { test.Error("SERVER", err); return }
-		defer conn.Close()
-		test.Cleanup(func() { conn.Close() })
-		a := AdaptA(conn, ServerSide)
-		trans, err := a.OpenTrans()
-		if err != nil { test.Error("SERVER", err); return }
-		test.Cleanup(func() { trans.Close() })
-		for method, payload := range payloads {
-			test.Log("SERVER m:", method, "p:", payload)
-			err := trans.Send(uint16(method), []byte(payload))
-			if err != nil { test.Error("SERVER", err); return }
-		}
-		test.Log("SERVER closing connection")
-	}()
-
-	// client
-	test.Log("CLIENT dialing")
-	conn, err := net.Dial(network, addr)
-	if err != nil { test.Fatal("CLIENT", err) }
-	test.Log("CLIENT dialed")
-	a := AdaptA(conn, ClientSide)
-	test.Cleanup(func() { a.Close() })
+	clientFunc := func(a Conn) {
 		test.Log("CLIENT accepting transaction")
 		trans, err := a.AcceptTrans()
 		if err != nil { test.Fatal("CLIENT", err) }
@@ -75,13 +44,26 @@ func TestConnA(test *testing.T) {
 				test.Errorf("CLIENT payload not equal")
 			}
 		}
-	test.Log("CLIENT waiting for connection close...")
+		test.Log("CLIENT waiting for transaction close...")
 		_, _, err = trans.Receive()
 		if !errors.Is(err, io.EOF) {
 			test.Fatal("CLIENT wrong error:", err)
 		}
-	test.Log("CLIENT done")
-	conn.Close()
+	}
+	
+	serverFunc := func(a Conn) {
+		trans, err := a.OpenTrans()
+		if err != nil { test.Error("SERVER", err); return }
+		test.Cleanup(func() { trans.Close() })
+		for method, payload := range payloads {
+			test.Log("SERVER m:", method, "p:", payload)
+			err := trans.Send(uint16(method), []byte(payload))
+			if err != nil { test.Error("SERVER", err); return }
+		}
+		test.Log("SERVER closing connection")
+	}
+
+	clientServerEnvironment(test, clientFunc, serverFunc)
 }
 
 func TestEncodeMessageA(test *testing.T) {
@@ -144,3 +126,42 @@ func TestDecodeMessageAErr(test *testing.T) {
 		test.Fatalf("wrong error: %v", err)
 	}
 }
+
+func clientServerEnvironment(test *testing.T, clientFunc func(conn Conn), serverFunc func(conn Conn)) {
+	network := "tcp"
+	addr := "localhost:7959"
+
+	// server
+	listener, err := net.Listen(network, addr)
+	if err != nil { test.Fatal(err) }
+	test.Cleanup(func() { listener.Close() })
+	go func() {
+		test.Log("SERVER listening")
+		conn, err := listener.Accept()
+		if err != nil { test.Error("SERVER", err); return }
+		defer conn.Close()
+		test.Cleanup(func() { conn.Close() })
+		a := AdaptA(conn, ServerSide)
+		test.Cleanup(func() { a.Close() })
+
+		serverFunc(a)
+	}()
+
+	// client
+	test.Log("CLIENT dialing")
+	conn, err := net.Dial(network, addr)
+	if err != nil { test.Fatal("CLIENT", err) }
+	test.Log("CLIENT dialed")
+	a := AdaptA(conn, ClientSide)
+	test.Cleanup(func() { a.Close() })
+
+	clientFunc(a)
+	
+	test.Log("CLIENT waiting for connection close...")
+	_, err = a.AcceptTrans()
+	if !errors.Is(err, io.EOF) {
+		test.Fatal("CLIENT wrong error:", err)
+	}
+	test.Log("CLIENT DONE")
+	conn.Close()
+}