Lib/test/test_netrc.py

   1
   2 import netrc, os, unittest
   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         fp = open(temp_filename, 'wt')
  26         fp.write(TEST_NETRC)
  27         fp.close()
  28         self.netrc = netrc.netrc(temp_filename)
  29
  30     def tearDown (self):
  31         del self.netrc
  32         os.unlink(temp_filename)
  33
  34     def test_case_1(self):
  35         self.assert_(self.netrc.macros == {'macro1':['line1\n', 'line2\n'],
  36                                            'macro2':['line3\n', 'line4\n']}
  37                                            )
  38         self.assert_(self.netrc.hosts['foo'] == ('log1', 'acct1', 'pass1'))
  39         self.assert_(self.netrc.hosts['default'] == ('log2', None, 'pass2'))
  40
  41 def test_main():
  42     test_support.run_unittest(NetrcTestCase)
  43
  44 if __name__ == "__main__":
  45     test_main()