26 lines
621 B
Go
26 lines
621 B
Go
|
package controller
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/QuantumNous/new-api/setting/system_setting"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPaymentReturnPathUsesDefaultDashboardRoutes(t *testing.T) {
|
||
|
|
previousAddress := system_setting.ServerAddress
|
||
|
|
system_setting.ServerAddress = "https://dashboard.example.com/"
|
||
|
|
t.Cleanup(func() { system_setting.ServerAddress = previousAddress })
|
||
|
|
|
||
|
|
assert.Equal(
|
||
|
|
t,
|
||
|
|
"https://dashboard.example.com/wallet?pay=success",
|
||
|
|
paymentReturnPath("/wallet?pay=success"),
|
||
|
|
)
|
||
|
|
assert.Equal(
|
||
|
|
t,
|
||
|
|
"https://dashboard.example.com/usage-logs",
|
||
|
|
paymentReturnPath("/usage-logs"),
|
||
|
|
)
|
||
|
|
}
|