kraken

1
// Copyright (c) 2016-2019 Uber Technologies, Inc.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14
package storage
15

16
import (
17
	"fmt"
18
	"testing"
19

20
	"github.com/uber/kraken/core"
21
	"github.com/uber/kraken/utils/bitsetutil"
22

23
	"github.com/stretchr/testify/require"
24
	"github.com/willf/bitset"
25
)
26

27
func TestTorrentInfoPercentDownloaded(t *testing.T) {
28
	mi := core.SizedBlobFixture(100, 25).MetaInfo
29
	tests := []struct {
30
		bitfield *bitset.BitSet
31
		expected int
32
	}{
33
		{bitsetutil.FromBools(true, true, true, true), 100},
34
		{bitsetutil.FromBools(true, false, true, false), 50},
35
		{bitsetutil.FromBools(false, false, false, false), 0},
36
	}
37
	for _, test := range tests {
38
		t.Run(fmt.Sprintf("%d%%", test.expected), func(t *testing.T) {
39
			require := require.New(t)
40

41
			info := NewTorrentInfo(mi, test.bitfield)
42
			require.Equal(test.expected, info.PercentDownloaded())
43
			require.Equal(test.bitfield, info.Bitfield())
44
			require.Equal(int64(25), info.MaxPieceLength())
45
			require.Equal(mi.InfoHash(), info.InfoHash())
46
			require.Equal(mi.Digest(), info.Digest())
47
			require.Equal(mi.InfoHash().Hex(), info.String())
48
		})
49
	}
50
}
51