optimize add ut for message

pkg/protocol/message/message_apis_test.go

@@ -0,0 +1,32 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package message
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMergedWarpMessage_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_SeataMerge, MergedWarpMessage{}.GetTypeCode())
+}
+
+func TestMergedResultMessage_GetTYpeCode(t *testing.T) {
+	assert.Equal(t, MessageType_SeataMergeResult, MergeResultMessage{}.GetTypeCode())
+}

pkg/protocol/message/other_message_test.go

@@ -0,0 +1,38 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package message
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewMessageFuture(t *testing.T) {
+	rpcMessage := RpcMessage{ID: 0}
+	assert.Equal(t, 0, NewMessageFuture(rpcMessage).ID)
+}
+
+func TestHeartBeatMessage_ToString(t *testing.T) {
+	assert.Equal(t, "services ping", HeartBeatMessagePing.ToString())
+	assert.Equal(t, "services pong", HeartBeatMessagePong.ToString())
+}
+
+func TestHeartBeatMessage_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_HeartbeatMsg, HeartBeatMessage{}.GetTypeCode())
+}

pkg/protocol/message/request_message.go

@@ -153,11 +153,3 @@ type RegisterRMRequest struct {
 func (req RegisterRMRequest) GetTypeCode() MessageType {
 	return MessageType_RegRm
 }
-
-type RegisterRMResponse struct {
-	AbstractIdentifyResponse
-}
-
-func (resp RegisterRMResponse) GetTypeCode() MessageType {
-	return MessageType_RegRmResult
-}

pkg/protocol/message/request_message_test.go

@@ -0,0 +1,76 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package message
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBranchRegisterRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchRegister, BranchRegisterRequest{}.GetTypeCode())
+}
+
+func TestBranchReportRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchStatusReport, BranchReportRequest{}.GetTypeCode())
+}
+
+func TestBranchCommitRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchCommit, BranchCommitRequest{}.GetTypeCode())
+}
+
+func TestBranchRollbackRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchRollback, BranchRollbackRequest{}.GetTypeCode())
+}
+
+func TestGlobalBeginRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalBegin, GlobalBeginRequest{}.GetTypeCode())
+}
+
+func TestGlobalCommitRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalCommit, GlobalCommitRequest{}.GetTypeCode())
+}
+
+func TestGlobalRollbackRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalRollback, GlobalRollbackRequest{}.GetTypeCode())
+}
+
+func TestGlobalLockQueryRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalLockQuery, GlobalLockQueryRequest{}.GetTypeCode())
+}
+
+func TestGlobalReportRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalReport, GlobalReportRequest{}.GetTypeCode())
+}
+
+func TestUndoLogDeleteRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_RmDeleteUndolog, UndoLogDeleteRequest{}.GetTypeCode())
+}
+
+func TestRegisterTMRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_RegClt, RegisterTMRequest{}.GetTypeCode())
+}
+
+func TestRegisterRMRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_RegRm, RegisterRMRequest{}.GetTypeCode())
+}
+
+func TestGlobalStatusRequest_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalStatusResult, GlobalStatusResponse{}.GetTypeCode())
+}

pkg/protocol/message/response_message.go

@@ -132,3 +132,11 @@ type GlobalRollbackResponse struct {
 func (resp GlobalRollbackResponse) GetTypeCode() MessageType {
 	return MessageType_GlobalRollbackResult
 }
+
+type RegisterRMResponse struct {
+	AbstractIdentifyResponse
+}
+
+func (resp RegisterRMResponse) GetTypeCode() MessageType {
+	return MessageType_RegRmResult
+}

pkg/protocol/message/response_message_test.go

@@ -0,0 +1,72 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package message
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRegisterRMResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_RegRmResult, RegisterRMResponse{}.GetTypeCode())
+}
+
+func TestRegisterTMResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_RegCltResult, RegisterTMResponse{}.GetTypeCode())
+}
+
+func TestGlobalReportResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalReportResult, GlobalReportResponse{}.GetTypeCode())
+}
+
+func TestGlobalLockQueryResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalLockQueryResult, GlobalLockQueryResponse{}.GetTypeCode())
+}
+
+func TestGlobalRollbackResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalRollbackResult, GlobalRollbackResponse{}.GetTypeCode())
+}
+
+func TestGlobalCommitResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalCommitResult, GlobalCommitResponse{}.GetTypeCode())
+}
+
+func TestGlobalBeginResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalBeginResult, GlobalBeginResponse{}.GetTypeCode())
+}
+
+func TestBranchRollbackResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchRollbackResult, BranchRollbackResponse{}.GetTypeCode())
+}
+
+func TestBranchCommitResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchCommitResult, BranchCommitResponse{}.GetTypeCode())
+}
+
+func TestBranchRegisterResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchRegisterResult, BranchRegisterResponse{}.GetTypeCode())
+}
+
+func TestBranchReportResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_BranchStatusReportResult, BranchReportResponse{}.GetTypeCode())
+}
+
+func TestGlobalStatusResponse_GetTypeCode(t *testing.T) {
+	assert.Equal(t, MessageType_GlobalStatusResult, GlobalStatusResponse{}.GetTypeCode())
+}