72 lines
2.7 KiB
Go
72 lines
2.7 KiB
Go
|
// 功能描述：合同申请答复的原气站归属、状态锁与幂等；版本：1.0.0。
|
|||
|
|
package gas
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"git.apinb.com/bsm-sdk/core/types"
|
|||
|
|
"git.apinb.com/heqiapp/platforms/backend/api/internal/impl"
|
|||
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"gorm.io/driver/postgres"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestResolveContractRequestScopeAndState(t *testing.T) {
|
|||
|
|
for _, tc := range []struct {
|
|||
|
|
name string
|
|||
|
|
state int
|
|||
|
|
result string
|
|||
|
|
found, success, update bool
|
|||
|
|
}{
|
|||
|
|
{"待受理答复", 32, "", true, true, true}, {"处理中答复", 11, "", true, true, true},
|
|||
|
|
{"重复答复", 34, "已处理", true, true, false}, {"不得覆盖答复", 34, "原答复", true, false, false},
|
|||
|
|
{"取消后拒绝", 22, "", true, false, false}, {"其他气站", 0, "", false, false, false},
|
|||
|
|
} {
|
|||
|
|
t.Run(tc.name, func(t *testing.T) {
|
|||
|
|
connection, mock, err := sqlmock.New()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer connection.Close()
|
|||
|
|
db, err := gorm.Open(postgres.New(postgres.Config{Conn: connection}), &gorm.Config{})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
previous := impl.DBService
|
|||
|
|
impl.DBService = db
|
|||
|
|
defer func() { impl.DBService = previous }()
|
|||
|
|
mock.ExpectQuery(`SELECT .* FROM "gas_account"`).WillReturnRows(sqlmock.NewRows([]string{"id", "identity", "gas_basic_id"}).AddRow(2, "gas", 9))
|
|||
|
|
mock.ExpectQuery(`SELECT .* FROM "gas_basic"`).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(9))
|
|||
|
|
mock.ExpectBegin()
|
|||
|
|
rows := sqlmock.NewRows([]string{"id", "ticket_status", "result"})
|
|||
|
|
if tc.found {
|
|||
|
|
rows.AddRow(7, tc.state, tc.result)
|
|||
|
|
}
|
|||
|
|
mock.ExpectQuery(`SELECT .* FROM "cs_ticket".*identity = \$1 AND gas_basic_id = \$2 AND gasorder_contract_id <> 0 AND category = \$3 AND status <> \$4.*FOR UPDATE`).WithArgs("request", 9, "contract_change", 3, 1).WillReturnRows(rows)
|
|||
|
|
if tc.update {
|
|||
|
|
mock.ExpectExec(`UPDATE "cs_ticket" SET .*"result"=.*"ticket_status"=`).WillReturnResult(sqlmock.NewResult(0, 1))
|
|||
|
|
}
|
|||
|
|
if tc.success {
|
|||
|
|
mock.ExpectCommit()
|
|||
|
|
} else {
|
|||
|
|
mock.ExpectRollback()
|
|||
|
|
}
|
|||
|
|
response := httptest.NewRecorder()
|
|||
|
|
ctx, _ := gin.CreateTestContext(response)
|
|||
|
|
ctx.Set("Auth", &types.JwtClaims{Client: "gas_admin", Identity: "gas"})
|
|||
|
|
ctx.Params = gin.Params{{Key: "identity", Value: "request"}}
|
|||
|
|
ctx.Request = httptest.NewRequest("POST", "/requests", strings.NewReader(`{"result":"已处理"}`))
|
|||
|
|
ctx.Request.Header.Set("Content-Type", "application/json")
|
|||
|
|
ResolveContractRequest(ctx)
|
|||
|
|
if strings.Contains(response.Body.String(), `"code":0`) != tc.success {
|
|||
|
|
t.Fatal(response.Body.String())
|
|||
|
|
}
|
|||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|