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
85
86
87
88
|
#include "NetrcAuthResolver.h"
#include <cppunit/extensions/HelperMacros.h>
#include "prefs.h"
#include "Netrc.h"
#include "AuthConfig.h"
#include "a2functional.h"
namespace aria2 {
class NetrcAuthResolverTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(NetrcAuthResolverTest);
CPPUNIT_TEST(testResolveAuthConfig_without_userDefined);
CPPUNIT_TEST(testResolveAuthConfig_with_userDefined);
CPPUNIT_TEST(testResolveAuthConfig_ignoreDefault);
CPPUNIT_TEST_SUITE_END();
private:
std::unique_ptr<Netrc> netrc_;
std::unique_ptr<NetrcAuthResolver> resolver_;
public:
void setUp()
{
netrc_.reset(new Netrc());
netrc_->addAuthenticator(
make_unique<Authenticator>("localhost", "name", "passwd", "account"));
netrc_->addAuthenticator(make_unique<DefaultAuthenticator>(
"default", "defaultpasswd", "defaultaccount"));
resolver_.reset(new NetrcAuthResolver());
resolver_->setNetrc(netrc_.get());
resolver_->setDefaultCred("foo", "bar");
}
void testResolveAuthConfig_without_userDefined();
void testResolveAuthConfig_with_userDefined();
void testResolveAuthConfig_ignoreDefault();
};
CPPUNIT_TEST_SUITE_REGISTRATION(NetrcAuthResolverTest);
void NetrcAuthResolverTest::testResolveAuthConfig_without_userDefined()
{
auto authConfig = resolver_->resolveAuthConfig("localhost");
CPPUNIT_ASSERT_EQUAL(std::string("name:passwd"), authConfig->getAuthText());
authConfig = resolver_->resolveAuthConfig("mymachine");
CPPUNIT_ASSERT_EQUAL(std::string("default:defaultpasswd"),
authConfig->getAuthText());
resolver_->setNetrc(nullptr);
authConfig = resolver_->resolveAuthConfig("localhost");
CPPUNIT_ASSERT_EQUAL(std::string("foo:bar"), authConfig->getAuthText());
}
void NetrcAuthResolverTest::testResolveAuthConfig_with_userDefined()
{
resolver_->setUserDefinedCred("myname", "mypasswd");
auto authConfig = resolver_->resolveAuthConfig("localhost");
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypasswd"),
authConfig->getAuthText());
authConfig = resolver_->resolveAuthConfig("mymachine");
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypasswd"),
authConfig->getAuthText());
resolver_->setNetrc(nullptr);
authConfig = resolver_->resolveAuthConfig("mymachine");
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypasswd"),
authConfig->getAuthText());
}
void NetrcAuthResolverTest::testResolveAuthConfig_ignoreDefault()
{
resolver_->ignoreDefault();
auto authConfig = resolver_->resolveAuthConfig("mirror");
CPPUNIT_ASSERT_EQUAL(std::string("foo:bar"), authConfig->getAuthText());
resolver_->useDefault();
auto defAuthConfig = resolver_->resolveAuthConfig("mirror");
CPPUNIT_ASSERT_EQUAL(std::string("default:defaultpasswd"),
defAuthConfig->getAuthText());
}
} // namespace aria2
|
