Streamer: BadEnvironment exception test

ai-nick · Feb 29, 2016 · 3acfc9d · 3acfc9d
1 parent 1f5fc9e
commit 3acfc9d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_streaming_rates_and_events.py b/tests/test_streaming_rates_and_events.py
@@ -91,6 +91,12 @@ def test__Events(self, count, ignore_heartbeat=False):
         r.events(ignore_heartbeat=False)
         self.assertEqual(count, r.reccnt + r.hbcnt)
 
+    def test__Environment(self):
+        """ test BadEnvironment exception """
+        with self.assertRaises(oandapy.BadEnvironment) as envErr:
+            s = Stream(access_token=access_token,
+                       environment="nothing", count=10)
+        self.assertTrue(isinstance(envErr.exception, oandapy.BadEnvironment))
 
 if __name__ == "__main__":