finance_test.py

   1 import unittest
   2 import finance
   3 import finance.service
   4 import os
   5 import getpass
   6
   7 def make_client():
   8     if not os.environ.has_key('GOOGLE_USER'):
   9         raise Exception('"GOOGLE_USER not set')
  10     else:
  11         user = os.environ['GOOGLE_USER']
  12     if not os.environ.has_key('GOOGLE_PASSWORD'):
  13         password = getpass.getpass('Enter password for %s: ' %
  14                                    user)
  15     else:
  16         password = os.environ['GOOGLE_PASSWORD']
  17
  18     client = finance.service.FinanceService()
  19     client.ClientLogin(user, password)
  20     return client
  21
  22
  23 class TestSequenceFunctions(unittest.TestCase):
  24     user = None
  25     password = None
  26
  27     def setUp(self):
  28         if not self.user:
  29             if not os.environ.has_key('GOOGLE_USER'):
  30                 raise Exception('"GOOGLE_USER not set')
  31             else:
  32                 self.user = os.environ['GOOGLE_USER']
  33         if not self.password:
  34             if not os.environ.has_key('GOOGLE_PASSWORD'):
  35                 self.password = getpass.getpass('Enter password for %s: ' %
  36                                                 self.user)
  37             else:
  38                 self.password = os.environ['GOOGLE_PASSWORD']
  39
  40         self.client = finance.service.FinanceService()
  41         self.client.ClientLogin(self.user, self.password)
  42
  43     def testGetPortfolioFeed(self):
  44         feed = self.client.GetPortfolioListFeed()
  45         print feed
  46
  47 if __name__ == '__main__':
  48     unittest.main()