Lib/test/test_netrc.py

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