initial

1 files changed, 30 insertions, 0 deletions

diff --git a/invest-applet/invest/test.py b/invest-applet/invest/test.py
new file mode 100755
index 00000000..aea5f5f3
--- /dev/null
+++ b/invest-applet/invest/test.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+
+import unittest
+from os.path import *
+import sys
+
+# Make sure we run the local version
+sys.path.insert(0, abspath(dirname(__file__) + "/.."))
+print sys.path
+
+import quotes
+import invest
+
+def null_function (*args):
+    pass
+
+class TestQuotes (unittest.TestCase):
+    def testQuoteUpdater_populate (self):
+        qu = quotes.QuoteUpdater (null_function, null_function)
+        invest.STOCKS = {'GOGO': {'label': "Google Inc.", 'purchases': [{'amount' : 1, 'comission' : 0.0, 'bought': 0.0}]}, 'JAVA': {'label':"Sun Microsystems Inc.", 'purchases': [{'amount' : 1, 'comission' : 0.0, 'bought': 0.0}]}}
+        quote = { 'GOGO': { "ticker": 'GOGO', "trade": 386.91, "time": "10/3/2008", "date": "4.00pm", "variation": -3.58, "open": 397.14, "variation_pct": 10 }}
+        qu.populate (quote)
+        self.assertEqual (qu.quotes_valid, True)
+        # In response to bug 554425, try a stock that isn't in our database
+        quote = { "clearlyFake": { "ticker": "clearlyFake", "trade": 386.91, "time": "10/3/2008", "date": "4.00pm", "variation": -3.58, "open": 397.14, "variation_pct": 10 }}
+        qu.populate (quote)
+        self.assertEqual (qu.quotes_valid, False)
+
+if __name__ == '__main__':
+    unittest.main ()