LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

middleware/torrentapproval : make use of ErrTorrentUnapproved

Browse source
This commit is contained in:
Pierre-Alain TORET 2018-09-20 18:05:12 +02:00
parent c9d51e8e68
commit 757ebf1241
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
middleware/torrentapproval/torrentapproval_test.go
View file

@ -6,8 +6,9 @@ import (
"fmt" "fmt"
"testing" "testing"
"github.com/chihaya/chihaya/bittorrent"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"github.com/chihaya/chihaya/bittorrent"
) )
var cases = []struct { var cases = []struct {
@ -69,9 +70,9 @@ func TestHandleAnnounce(t *testing.T) {
nctx, err := h.HandleAnnounce(ctx, req, resp) nctx, err := h.HandleAnnounce(ctx, req, resp)
require.Equal(t, ctx, nctx) require.Equal(t, ctx, nctx)
if tt.approved == true { if tt.approved == true {
require.Nil(t, err) require.NotEqual(t, err, ErrTorrentUnapproved)
} else { } else {
require.NotNil(t, err) require.Equal(t, err, ErrTorrentUnapproved)
} }
}) })
} }