refactor ciims proxy transport and handlers

2026-07-08 15:17:07 +08:00
parent 04bafe01e7
commit dd3f6769f6
8 changed files with 689 additions and 154 deletions
@@ -0,0 +1,174 @@
+package internal
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"net"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestClientSend_SOAPFault(t *testing.T) {
+	ciims := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "text/xml; charset=UTF-8")
+		fmt.Fprint(w, ErrMsg)
+	}))
+	defer ciims.Close()
+
+	client, err := NewClient(10)
+	require.NoError(t, err)
+	resp, err := client.Send(ciims.URL+"/services/ExchangeService", "<dummy>")
+	assert.NoError(t, err)
+	require.NotNil(t, resp)
+
+	assert.True(t, resp.IsFault())
+	assert.Equal(t, "Can not find the event [FLOP-ESTT-ATC-ALL1]", resp.ErrorMessage())
+}
+
+func TestClientSend_Success(t *testing.T) {
+	ciims := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "text/xml; charset=UTF-8")
+		fmt.Fprint(w, SendOk)
+	}))
+	defer ciims.Close()
+
+	client, err := NewClient(10)
+	require.NoError(t, err)
+	resp, err := client.Send(ciims.URL+"/services/ExchangeService", "<dummy>")
+	assert.NoError(t, err)
+	require.NotNil(t, resp)
+
+	assert.Equal(t, "", resp.ErrorMessage())
+}
+
+func TestClientSend_ReceiveMessages(t *testing.T) {
+	ciims := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "text/xml; charset=UTF-8")
+		fmt.Fprint(w, ReceiveResp)
+	}))
+	defer ciims.Close()
+
+	client, err := NewClient(10)
+	require.NoError(t, err)
+	resp, err := client.Send(ciims.URL+"/services/ExchangeService", "<dummy>")
+	assert.NoError(t, err)
+	require.NotNil(t, resp)
+
+	msgs := resp.Messages()
+	assert.Equal(t, 2, len(msgs))
+	assert.Equal(t, Msg, msgs[0])
+	assert.Equal(t, Msg, msgs[1])
+}
+
+func TestClientSend_NetworkTimeout(t *testing.T) {
+	listener, err := net.Listen("tcp", "127.0.0.1:0")
+	require.NoError(t, err)
+	addr := listener.Addr().String()
+
+	go func() {
+		conn, _ := listener.Accept()
+		if conn != nil {
+			io.Copy(io.Discard, conn)
+			conn.Close()
+		}
+	}()
+
+	client, err := NewClient(1)
+	require.NoError(t, err)
+	_, err = client.Send("http://"+addr+"/services/ExchangeService", "<dummy>")
+	listener.Close()
+	assert.Error(t, err)
+}
+
+func TestClientSend_ServerError(t *testing.T) {
+	client, err := NewClient(1)
+	require.NoError(t, err)
+	_, err = client.Send("http://127.0.0.1:1/nonexistent", "<dummy>")
+	assert.Error(t, err)
+}
+
+func TestClientSend_HTTPStatusError(t *testing.T) {
+	ciims := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		http.Error(w, "backend failed", http.StatusInternalServerError)
+	}))
+	defer ciims.Close()
+
+	client, err := NewClient(10)
+	require.NoError(t, err)
+	resp, err := client.Send(ciims.URL+"/services/ExchangeService", "<dummy>")
+	require.Error(t, err)
+	require.NotNil(t, resp)
+
+	var statusErr *HTTPStatusError
+	require.True(t, errors.As(err, &statusErr))
+	assert.Equal(t, http.StatusInternalServerError, statusErr.StatusCode)
+	assert.Contains(t, statusErr.Body, "backend failed")
+}
+
+func TestGetMsgs_Empty(t *testing.T) {
+	msgs := GetMsgs("<soap><body></body></soap>")
+	assert.Equal(t, 0, len(msgs))
+}
+
+func TestGetMsgs_DifferentNamespacePrefix(t *testing.T) {
+	resp := `<soap:Envelope><soap:Body><ns2:receiveResponse xmlns:ns2="http://ciims.bhia.itdcl.com/ExchangeService">
+		<ns2:out><ns2:string>&lt;MSG/&gt;</ns2:string></ns2:out>
+	</ns2:receiveResponse></soap:Body></soap:Envelope>`
+	msgs := GetMsgs(resp)
+	assert.Equal(t, 1, len(msgs))
+	assert.Equal(t, "<MSG/>", msgs[0])
+}
+
+func TestGetMsgs_AttributesWhitespaceAndMultiline(t *testing.T) {
+	resp := `<soap:Envelope><soap:Body><ns2:out>
+		<ns2:string id="1">
+			&lt;MSG&gt;
+			  &lt;A&gt;1&lt;/A&gt;
+			&lt;/MSG&gt;
+		</ns2:string>
+	</ns2:out></soap:Body></soap:Envelope>`
+	msgs := GetMsgs(resp)
+	require.Len(t, msgs, 1)
+	assert.Contains(t, msgs[0], "<MSG>")
+	assert.Contains(t, msgs[0], "<A>1</A>")
+}
+
+func TestGetErrMsg_NoError(t *testing.T) {
+	errMsg := GetErrMsg(SendOk)
+	assert.Equal(t, "", errMsg)
+}
+
+func TestGetErrMsg_Empty(t *testing.T) {
+	errMsg := GetErrMsg("<soap><body></body></soap>")
+	assert.Equal(t, "", errMsg)
+}
+
+func TestGetErrMsg_AttributesWhitespaceAndMultiline(t *testing.T) {
+	resp := `<soap:Envelope><soap:Body><soap:Fault>
+		<errorMessage xmlns="http://msg.ciims.bhia.itdcl.com" code="20">
+			Can not find
+			the event
+		</errorMessage>
+	</soap:Fault></soap:Body></soap:Envelope>`
+	errMsg := GetErrMsg(resp)
+	assert.Contains(t, errMsg, "Can not find")
+	assert.Contains(t, errMsg, "the event")
+}
+
+func TestCreateSend_XMLEscapes(t *testing.T) {
+	result := CreateSend("user<>&\"'", "pass", 0, "event", false, "<MSG/>")
+	assert.Contains(t, result, "user&lt;&gt;&amp;&#34;&#39;")
+	assert.Contains(t, result, "pass")
+	assert.Contains(t, result, "&lt;MSG/&gt;")
+}
+
+func TestCreateReceive_XMLEscapes(t *testing.T) {
+	result := CreateReceive("user<>&\"'", "pass>", 2)
+	assert.Contains(t, result, "user&lt;&gt;&amp;&#34;&#39;")
+	assert.Contains(t, result, "pass&gt;")
+}