made test_metadata_extension a bit more reliable
This commit is contained in:
parent
a75699eff1
commit
f2455c8689
|
@ -77,6 +77,7 @@ struct test_swarm_config : swarm_config
|
||||||
: swarm_config()
|
: swarm_config()
|
||||||
, m_flags(flags)
|
, m_flags(flags)
|
||||||
, m_plugin(plugin)
|
, m_plugin(plugin)
|
||||||
|
, m_metadata_alerts(0)
|
||||||
{}
|
{}
|
||||||
|
|
||||||
// called for every session that's added
|
// called for every session that's added
|
||||||
|
@ -146,6 +147,11 @@ struct test_swarm_config : swarm_config
|
||||||
, std::vector<libtorrent::torrent_handle> const& handles
|
, std::vector<libtorrent::torrent_handle> const& handles
|
||||||
, libtorrent::session& ses) override
|
, libtorrent::session& ses) override
|
||||||
{
|
{
|
||||||
|
if (alert_cast<metadata_received_alert>(alert))
|
||||||
|
{
|
||||||
|
m_metadata_alerts += 1;
|
||||||
|
}
|
||||||
|
|
||||||
// make sure this function can be called on
|
// make sure this function can be called on
|
||||||
// torrents without metadata
|
// torrents without metadata
|
||||||
if ((m_flags & disconnect) == 0)
|
if ((m_flags & disconnect) == 0)
|
||||||
|
@ -175,6 +181,7 @@ struct test_swarm_config : swarm_config
|
||||||
|
|
||||||
virtual void on_exit(std::vector<torrent_handle> const& torrents) override
|
virtual void on_exit(std::vector<torrent_handle> const& torrents) override
|
||||||
{
|
{
|
||||||
|
TEST_EQUAL(m_metadata_alerts, 1);
|
||||||
// in this case we should have completed without downloading anything
|
// in this case we should have completed without downloading anything
|
||||||
// because the downloader had upload only set
|
// because the downloader had upload only set
|
||||||
if (m_flags & upload_only) return;
|
if (m_flags & upload_only) return;
|
||||||
|
@ -185,6 +192,7 @@ struct test_swarm_config : swarm_config
|
||||||
private:
|
private:
|
||||||
int m_flags;
|
int m_flags;
|
||||||
boost::shared_ptr<torrent_plugin> (*m_plugin)(torrent_handle const&, void*);
|
boost::shared_ptr<torrent_plugin> (*m_plugin)(torrent_handle const&, void*);
|
||||||
|
int m_metadata_alerts;
|
||||||
};
|
};
|
||||||
|
|
||||||
TORRENT_TEST(ut_metadata_encryption_reverse)
|
TORRENT_TEST(ut_metadata_encryption_reverse)
|
||||||
|
|
Loading…
Reference in New Issue