Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-07-23 08:18:11 -07:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #184 from mrd0ll4r/tests-smallfix

Browse Source 
store: fix various small test issues
This commit is contained in:
Jimmy Zelinskie
2016-07-09 21:58:54 -04:00
committed by GitHub
parent 6a47110c76 b3261a584e
commit 40d0236596
3 changed files with 32 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/store/store_bench.go
+3 -1
View File
@@ -856,7 +856,9 @@ func generatePeers() (a [num1KElements]chihaya.Peer) {
// benchmarks.
func PreparePeerStoreBenchmarker(driver PeerStoreDriver) PeerStoreBenchmarker {
return peerStoreBench{
driver: driver,
driver: driver,
infohashes: generateInfohashes(),
peers: generatePeers(),
}
}
server/store/store_tests.go
+19 -8
View File
@@ -359,26 +359,37 @@ func (s *ipStoreTester) TestHasAllHasAnyNetworks(t *testing.T, cfg *DriverConfig
// Every benchmark expects a new, clean storage. Every benchmark should be
// called with a DriverConfig that ensures this.
type PeerStoreTester interface {
// CompareEndpoints sets the function used to compare peers to a
// comparison that only compares endpoints and omits PeerIDs.
CompareEndpoints()
TestPeerStore(*testing.T, *DriverConfig)
}
var _ PeerStoreTester = &peerStoreTester{}
type peerStoreTester struct {
driver PeerStoreDriver
driver PeerStoreDriver
equalityFunc func(a, b chihaya.Peer) bool
}
// PreparePeerStoreTester prepares a reusable suite for PeerStore driver
// tests.
// The tester will use PeerIDs and endpoints to compare peers.
func PreparePeerStoreTester(driver PeerStoreDriver) PeerStoreTester {
return &peerStoreTester{
driver: driver,
driver: driver,
equalityFunc: func(a, b chihaya.Peer) bool { return a.Equal(b) },
}
}
func peerInSlice(peer chihaya.Peer, peers []chihaya.Peer) bool {
func (pt *peerStoreTester) CompareEndpoints() {
pt.equalityFunc = func(a, b chihaya.Peer) bool { return a.EqualEndpoint(b) }
}
func (pt *peerStoreTester) peerInSlice(peer chihaya.Peer, peers []chihaya.Peer) bool {
for _, v := range peers {
if v.Equal(peer) {
if pt.equalityFunc(peer, v) {
return true
}
}
@@ -453,9 +464,9 @@ func (pt *peerStoreTester) TestPeerStore(t *testing.T, cfg *DriverConfig) {
}
if p.seeder {
require.True(t, peerInSlice(peer, seeders))
require.True(t, pt.peerInSlice(peer, seeders))
} else {
require.True(t, peerInSlice(peer, leechers))
require.True(t, pt.peerInSlice(peer, leechers))
}
if p.seeder {
@@ -506,8 +517,8 @@ func (pt *peerStoreTester) TestPeerStore(t *testing.T, cfg *DriverConfig) {
err = s.CollectGarbage(time.Now())
require.Nil(t, err)
require.Equal(t, s.NumLeechers(hash), 0)
require.Equal(t, s.NumSeeders(hash), 0)
require.Equal(t, 0, s.NumLeechers(hash))
require.Equal(t, 0, s.NumSeeders(hash))
errChan := s.Stop()
err = <-errChan