Lib/test/test_netrc.py

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