mirror of
https://github.com/sot-tech/mochi.git
synced 2026-04-26 07:30:00 -07:00
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package varinterval
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/sot-tech/mochi/bittorrent"
|
|
"github.com/sot-tech/mochi/pkg/conf"
|
|
)
|
|
|
|
var configTests = []struct {
|
|
cfg Config
|
|
expected error
|
|
}{
|
|
{
|
|
cfg: Config{0.5, 60, true},
|
|
expected: nil,
|
|
}, {
|
|
cfg: Config{1.0, 60, true},
|
|
expected: nil,
|
|
}, {
|
|
cfg: Config{0.0, 60, true},
|
|
expected: ErrInvalidModifyResponseProbability,
|
|
}, {
|
|
cfg: Config{1.1, 60, true},
|
|
expected: ErrInvalidModifyResponseProbability,
|
|
}, {
|
|
cfg: Config{0.5, 0, true},
|
|
expected: ErrInvalidMaxIncreaseDelta,
|
|
}, {
|
|
cfg: Config{0.5, -10, true},
|
|
expected: ErrInvalidMaxIncreaseDelta,
|
|
},
|
|
}
|
|
|
|
func TestCheckConfig(t *testing.T) {
|
|
for _, tt := range configTests {
|
|
t.Run(fmt.Sprintf("%#v", tt.cfg), func(t *testing.T) {
|
|
got := checkConfig(tt.cfg)
|
|
require.Equal(t, tt.expected, got, "", tt.cfg)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHandleAnnounce(t *testing.T) {
|
|
c := conf.MapConfig{"modify_response_probability": 1.0, "max_increase_delta": 10, "modify_min_interval": true}
|
|
h, err := build(c, nil)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, h)
|
|
|
|
ctx := context.Background()
|
|
req := &bittorrent.AnnounceRequest{InfoHash: "1234567890ABCDEF0000"}
|
|
resp := &bittorrent.AnnounceResponse{}
|
|
|
|
nCtx, err := h.HandleAnnounce(ctx, req, resp)
|
|
require.Nil(t, err)
|
|
require.Equal(t, ctx, nCtx)
|
|
require.True(t, resp.Interval > 0, "interval should have been increased")
|
|
require.True(t, resp.MinInterval > 0, "min_interval should have been increased")
|
|
}
|