#include "Netrc.h"
#include "Exception.h"
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class NetrcTest : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(NetrcTest);
  CPPUNIT_TEST(testFindAuthenticatable);
  CPPUNIT_TEST(testParse);
  CPPUNIT_TEST(testParse_fileNotFound);
  CPPUNIT_TEST(testParse_emptyfile);
  CPPUNIT_TEST_SUITE_END();
private:

public:
  void setUp() {
  }

  void testFindAuthenticatable();
  void testParse();
  void testParse_fileNotFound();
  void testParse_emptyfile();
};


CPPUNIT_TEST_SUITE_REGISTRATION( NetrcTest );

void NetrcTest::testFindAuthenticatable()
{
  Netrc netrc;
  netrc.addAuthenticatable(new Authenticator("host1", "tujikawa", "tujikawapasswd", "tujikawaaccount"));
  netrc.addAuthenticatable(new Authenticator("host2", "aria2", "aria2password", "aria2account"));
  netrc.addAuthenticatable(new DefaultAuthenticator("default", "defaultpassword", "defaultaccount"));

  AuthenticatorHandle aria2auth = netrc.findAuthenticatable("host2");
  CPPUNIT_ASSERT(!aria2auth.isNull());
  CPPUNIT_ASSERT_EQUAL(string("aria2"), aria2auth->getLogin());
  CPPUNIT_ASSERT_EQUAL(string("aria2password"), aria2auth->getPassword());
  CPPUNIT_ASSERT_EQUAL(string("aria2account"), aria2auth->getAccount());

  AuthenticatorHandle defaultauth = netrc.findAuthenticatable("host3");
  CPPUNIT_ASSERT(!defaultauth.isNull());
  CPPUNIT_ASSERT_EQUAL(string("default"), defaultauth->getLogin());
  CPPUNIT_ASSERT_EQUAL(string("defaultpassword"), defaultauth->getPassword());
  CPPUNIT_ASSERT_EQUAL(string("defaultaccount"), defaultauth->getAccount());
}

void NetrcTest::testParse()
{
  Netrc netrc;
  netrc.parse("sample.netrc");
  Authenticatables::const_iterator itr = netrc.getAuthenticatables().begin();

  AuthenticatorHandle tujikawaauth = *itr;
  CPPUNIT_ASSERT(!tujikawaauth.isNull());
  CPPUNIT_ASSERT_EQUAL(string("host1"), tujikawaauth->getMachine());
  CPPUNIT_ASSERT_EQUAL(string("tujikawa"), tujikawaauth->getLogin());
  CPPUNIT_ASSERT_EQUAL(string("tujikawapassword"), tujikawaauth->getPassword());
  CPPUNIT_ASSERT_EQUAL(string("tujikawaaccount"), tujikawaauth->getAccount());
  ++itr;
  AuthenticatorHandle aria2auth = *itr;
  CPPUNIT_ASSERT(!aria2auth.isNull());
  CPPUNIT_ASSERT_EQUAL(string("host2"), aria2auth->getMachine());
  CPPUNIT_ASSERT_EQUAL(string("aria2"), aria2auth->getLogin());
  CPPUNIT_ASSERT_EQUAL(string("aria2password"), aria2auth->getPassword());
  CPPUNIT_ASSERT_EQUAL(string("aria2account"), aria2auth->getAccount());
  ++itr;
  DefaultAuthenticatorHandle defaultauth = *itr;
  CPPUNIT_ASSERT(!defaultauth.isNull());
  CPPUNIT_ASSERT_EQUAL(string("anonymous"), defaultauth->getLogin());
  CPPUNIT_ASSERT_EQUAL(string("ARIA2@USER"), defaultauth->getPassword());
  CPPUNIT_ASSERT_EQUAL(string("ARIA2@ACCT"), defaultauth->getAccount());
}

void NetrcTest::testParse_fileNotFound()
{
  Netrc netrc;
  try {
    netrc.parse("");
    CPPUNIT_FAIL("exception must be threw.");
  } catch(Exception* e) {
    cerr << e->getMsg() << endl;
    delete e;
  }
}

void NetrcTest::testParse_emptyfile()
{
  Netrc netrc;
  netrc.parse("emptyfile");

  CPPUNIT_ASSERT_EQUAL((size_t)0, netrc.getAuthenticatables().size());
}