/Lib/test/test_netrc.py

http://unladen-swallow.googlecode.com/ · Python · 48 lines · 39 code · 9 blank · 0 comment · 2 complexity · a764a02f4f18e5963fae96389ce16920 MD5 · raw file

  1. import netrc, os, unittest, sys
  2. from test import test_support
  3. TEST_NETRC = """
  4. machine foo login log1 password pass1 account acct1
  5. macdef macro1
  6. line1
  7. line2
  8. macdef macro2
  9. line3
  10. line4
  11. default login log2 password pass2
  12. """
  13. temp_filename = test_support.TESTFN
  14. class NetrcTestCase(unittest.TestCase):
  15. def setUp (self):
  16. mode = 'w'
  17. if sys.platform not in ['cygwin']:
  18. mode += 't'
  19. fp = open(temp_filename, mode)
  20. fp.write(TEST_NETRC)
  21. fp.close()
  22. self.netrc = netrc.netrc(temp_filename)
  23. def tearDown (self):
  24. del self.netrc
  25. os.unlink(temp_filename)
  26. def test_case_1(self):
  27. self.assert_(self.netrc.macros == {'macro1':['line1\n', 'line2\n'],
  28. 'macro2':['line3\n', 'line4\n']}
  29. )
  30. self.assert_(self.netrc.hosts['foo'] == ('log1', 'acct1', 'pass1'))
  31. self.assert_(self.netrc.hosts['default'] == ('log2', None, 'pass2'))
  32. def test_main():
  33. test_support.run_unittest(NetrcTestCase)
  34. if __name__ == "__main__":
  35. test_main()