added test for IB source

Author: Idan Yael

tests/integration/test_ib_source.py

@@ -0,0 +1,29 @@
+from datetime import datetime, timedelta
+from unittest import TestCase
+
+from entities.timespan import TimeSpan
+from pipeline.sources.ib_history import IBHistorySource
+from providers.ib.interactive_brokers_connector import InteractiveBrokersConnector
+
+
+class TestIBSource(TestCase):
+    def setUp(self) -> None:
+        super().setUp()
+        self.ib_connector = InteractiveBrokersConnector()
+
+    def tearDown(self) -> None:
+        super().tearDown()
+        self.ib_connector.kill()
+
+    def test(self):
+        symbol = 'AAPL'
+        from_time = datetime.now() - timedelta(days=30)
+        source = IBHistorySource(self.ib_connector, [symbol], TimeSpan.Day, from_time)
+
+        candles = list(source.read())
+        self.assertTrue(len(candles) > 10)
+
+        for candle in candles:
+            self.assertEqual(symbol, candle.symbol)
+            self.assertEqual(TimeSpan.Day, candle.time_span)
+            self.assertTrue(candle.timestamp < datetime.now())