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
|
/*
* Copyright (c) 2012 Remko Tronçon
* Licensed under the GNU General Public License v3.
* See Documentation/Licenses/GPLv3.txt for more information.
*/
#include <QA/Checker/IO.h>
#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <Swiften/Network/DomainNameServiceQuery.h>
#include <Swiften/Base/RandomGenerator.h>
using namespace Swift;
namespace {
struct RandomGenerator1 : public RandomGenerator {
virtual int generateWeighedRandomNumber(std::vector<double>::const_iterator, std::vector<double>::const_iterator) {
return 0;
}
};
struct RandomGenerator2 : public RandomGenerator {
virtual int generateWeighedRandomNumber(std::vector<double>::const_iterator probabilities_begin, std::vector<double>::const_iterator probabilities_end) {
return std::max_element(probabilities_begin, probabilities_end) - probabilities_begin;
}
};
}
class DomainNameServiceQueryTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(DomainNameServiceQueryTest);
CPPUNIT_TEST(testSortResults_Random1);
CPPUNIT_TEST(testSortResults_Random2);
CPPUNIT_TEST_SUITE_END();
public:
void testSortResults_Random1() {
std::vector<DomainNameServiceQuery::Result> results;
results.push_back(DomainNameServiceQuery::Result("server1.com", 5222, 5, 1));
results.push_back(DomainNameServiceQuery::Result("server2.com", 5222, 3, 10));
results.push_back(DomainNameServiceQuery::Result("server3.com", 5222, 6, 1));
results.push_back(DomainNameServiceQuery::Result("server4.com", 5222, 3, 20));
results.push_back(DomainNameServiceQuery::Result("server5.com", 5222, 2, 1));
results.push_back(DomainNameServiceQuery::Result("server6.com", 5222, 3, 10));
RandomGenerator1 generator;
DomainNameServiceQuery::sortResults(results, generator);
CPPUNIT_ASSERT_EQUAL(std::string("server5.com"), results[0].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server2.com"), results[1].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server4.com"), results[2].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server6.com"), results[3].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server1.com"), results[4].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server3.com"), results[5].hostname);
}
void testSortResults_Random2() {
std::vector<DomainNameServiceQuery::Result> results;
results.push_back(DomainNameServiceQuery::Result("server1.com", 5222, 5, 1));
results.push_back(DomainNameServiceQuery::Result("server2.com", 5222, 3, 10));
results.push_back(DomainNameServiceQuery::Result("server3.com", 5222, 6, 1));
results.push_back(DomainNameServiceQuery::Result("server4.com", 5222, 3, 20));
results.push_back(DomainNameServiceQuery::Result("server5.com", 5222, 2, 1));
results.push_back(DomainNameServiceQuery::Result("server6.com", 5222, 3, 10));
results.push_back(DomainNameServiceQuery::Result("server7.com", 5222, 3, 40));
RandomGenerator2 generator;
DomainNameServiceQuery::sortResults(results, generator);
CPPUNIT_ASSERT_EQUAL(std::string("server5.com"), results[0].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server7.com"), results[1].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server4.com"), results[2].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server6.com"), results[3].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server2.com"), results[4].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server1.com"), results[5].hostname);
CPPUNIT_ASSERT_EQUAL(std::string("server3.com"), results[6].hostname);
}
};
CPPUNIT_TEST_SUITE_REGISTRATION(DomainNameServiceQueryTest);
|
Swift