blob: c4f39784d43efd7b3d1ea43530d64a94f90a7ab2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#ifndef _D_MOCK_PEER_STORAGE_H_
#define _D_MOCK_PEER_STORAGE_H_
#include "PeerStorage.h"
#include <algorithm>
#include "Peer.h"
namespace aria2 {
class MockPeerStorage : public PeerStorage {
private:
TransferStat stat;
std::deque<SharedHandle<Peer> > peers;
std::deque<SharedHandle<Peer> > activePeers;
int _numChokeExecuted;
public:
MockPeerStorage():_numChokeExecuted(0) {}
virtual ~MockPeerStorage() {}
virtual bool addPeer(const SharedHandle<Peer>& peer) {
peers.push_back(peer);
return true;
}
virtual void addPeer(const std::deque<SharedHandle<Peer> >& peers) {
std::copy(peers.begin(), peers.end(), back_inserter(this->peers));
}
virtual const std::deque<SharedHandle<Peer> >& getPeers() {
return peers;
}
virtual SharedHandle<Peer> getUnusedPeer() {
return SharedHandle<Peer>();
}
virtual bool isPeerAvailable() {
return false;
}
void setActivePeers(const std::deque<SharedHandle<Peer> >& activePeers)
{
this->activePeers = activePeers;
}
virtual void getActivePeers(std::deque<SharedHandle<Peer> >& peers) {
peers.insert(peers.end(), activePeers.begin(), activePeers.end());
}
virtual TransferStat calculateStat() {
return stat;
}
void setStat(const TransferStat& stat) {
this->stat = stat;
}
virtual void returnPeer(const SharedHandle<Peer>& peer)
{
}
virtual bool chokeRoundIntervalElapsed()
{
return false;
}
virtual void executeChoke()
{
++_numChokeExecuted;
}
virtual void updateTransferStatFor(const SharedHandle<Peer>& peer) {}
int getNumChokeExecuted() const
{
return _numChokeExecuted;
}
};
#endif // _D_MOCK_PEER_STORAGE_H_
} // namespace aria2
|
