Update tests

Author: MDUYN

investing_algorithm_framework/configuration/setup/default_template_creators.py

@@ -93,12 +93,14 @@ def create(self) -> None:
                 if filename in [
                     'manage.py-template',
                     'settings.py-template',
-                    'context.py-template'
+                    'context.py-template',
+                    'extensions.py-template',
+                    'data_providers.py-template',
+                    'strategies.py-template',
+                    'order_executors.py-template'
                 ]:
-
                     # Read the file
                     with open(destination_path, 'r') as file:
-
                         file_data = file.read()
 
                     # Replace the placeholder with the

investing_algorithm_framework/management/commands/run_algorithm.py

@@ -4,7 +4,6 @@
 from investing_algorithm_framework.core.context import Context
 from investing_algorithm_framework.configuration.config_constants import \
     SETTINGS_CONTEXT_CONFIGURATION
-from investing_algorithm_framework.configuration import settings
 
 
 class RunAlgorithmCommand(BaseCommand):
@@ -17,14 +16,14 @@ def add_arguments(self, parser) -> None:
         pass
 
     def handle(self, *args, **options) -> Any:
+        # Call the context
+        context = Context()
 
-        # configure settings
-        settings.configure()
+        # configure the configuration
+        context.config.configure()
 
         # Load the context configuration
-        __import__(settings[SETTINGS_CONTEXT_CONFIGURATION])
+        __import__(context.config[SETTINGS_CONTEXT_CONFIGURATION])
 
-        # Create an investing_algorithm_framework context of the
-        # investing_algorithm_framework and run it
-        context = Context()
+        # run the context
         context.start()

requirements.txt

@@ -1,4 +1,3 @@
-requests==2.22.0
 wrapt==1.11.2
 colorama==0.4.3
 SQLAlchemy==1.3.13

tests/configuration/__init__.py

@@ -0,0 +1,29 @@
+import pytest
+from investing_algorithm_framework.configuration import ContextConfiguration
+from investing_algorithm_framework.configuration.config_constants import \
+    SETTINGS_CONTEXT_CONFIGURATION, SETTINGS_LOGGING_CONFIG, \
+    SETTINGS_MAX_CONCURRENT_WORKERS, SETTINGS_PROJECT_NAME, BASE_DIR
+from tests.resources.utils import random_string
+
+
+def test() -> None:
+    config = ContextConfiguration()
+    config.configure('tests.resources.standard_settings')
+    assert config.configured
+
+    assert config[BASE_DIR] is not None
+    assert config[SETTINGS_CONTEXT_CONFIGURATION] is not None
+    assert config[SETTINGS_LOGGING_CONFIG] is not None
+    assert config[SETTINGS_MAX_CONCURRENT_WORKERS] is not None
+    assert config[SETTINGS_PROJECT_NAME] is not None
+
+    new_attribute = random_string(10)
+    new_attribute_value = random_string(10)
+
+    config.set(new_attribute, new_attribute_value)
+
+    assert config[new_attribute] is not None
+    assert config[new_attribute] == new_attribute_value
+
+    with pytest.raises(Exception):
+        config.set(new_attribute, random_string(10))

tests/configuration/test_context_configuration.py

@@ -88,3 +88,30 @@ def test() -> None:
     assert test_executor_observer.update_count == 3
     assert test_worker_observer.update_count == 7
 
+    executor = Executor(
+        workers=[test_worker_one]
+    )
+    executor.add_observer(test_executor_observer)
+
+    # Start the executor
+    executor.start()
+
+    sleep(2)
+
+    # Observers must have been updated by the executor
+    assert test_executor_observer.update_count == 4
+    assert test_worker_observer.update_count == 8
+
+
+class TestExceptionWorker(Worker):
+
+    def work(self, **kwargs: Dict[str, Any]) -> None:
+        raise Exception("hello")
+
+
+def test_exception() -> None:
+    test_worker = TestExceptionWorker()
+    executor = Executor(workers=[test_worker])
+    test_executor_observer = TestObserver()
+    executor.add_observer(test_executor_observer)
+    executor.start()

tests/core/executors/test_executor.py

@@ -3,7 +3,7 @@
 
 from tests.resources import BaseTestMixin
 from tests.resources import utils
-from investing_algorithm_framework.configuration import settings
+from investing_algorithm_framework.configuration import ContextConfiguration
 from investing_algorithm_framework.core.extensions import db
 
 
@@ -17,8 +17,9 @@ class TestDatabaseResolverConfiguration(BaseTestMixin):
     def setup_method(self) -> None:
         self.initialize_environment()
 
-    def test_configuration(self):
-        settings.configure()
+    def test_configuration(self) -> None:
+        config_config = ContextConfiguration()
+        config_config.configure()
         db.configure()
 
         # Check if all properties are configured
@@ -29,16 +30,17 @@ def test_configuration(self):
         assert os.path.isfile(db.database_path) == True
 
     def teardown_method(self) -> None:
-
-        if os.path.isfile(db.database_path):
-            os.remove(db.database_path)
+        pass
+        # if os.path.isfile(db.database_path):
+        #     os.remove(db.database_path)
 
 
 class TestDatabaseResolverModel(BaseTestMixin):
 
     def setup_method(self) -> None:
         self.initialize_environment()
-        settings.configure()
+        config_config = ContextConfiguration()
+        config_config.configure()
         db.configure()
         db.initialize_tables()
 

tests/core/resolvers/test_database_resolver.py

@@ -60,6 +60,13 @@ def test(self):
                     'configuration',
                     'settings.py')
             )
+            assert os.path.isfile(
+                os.path.join(
+                    tempdir,
+                    self.project_name,
+                    'configuration',
+                    'extensions.py')
+            )
 
             # Check if all data provider files are present
             assert os.path.isfile(

tests/management/test_create_standard_algo.py

@@ -1,3 +1,6 @@
+from typing import Dict, Any
+
+import pytest
 from time import sleep
 from investing_algorithm_framework.templates.states.data_providing_state \
     import DataProvidingState
@@ -6,6 +9,7 @@
 from investing_algorithm_framework.core.utils import TimeUnit
 from investing_algorithm_framework.core.context import Context
 from investing_algorithm_framework.core.events import Observer
+from investing_algorithm_framework.core.workers import Worker
 
 
 class TestObserver(Observer):
@@ -45,6 +49,21 @@ def provide_data(self) -> None:
         pass
 
 
+class WrongDataProviderOne:
+
+    def provide_data(self) -> None:
+        pass
+
+
+class WrongDataProviderTwo(Worker):
+
+    def work(self, **kwargs: Dict[str, Any]) -> None:
+        pass
+
+    def provide_data(self) -> None:
+        pass
+
+
 def test() -> None:
     observer = TestObserver()
     data_provider_one = TestDataProviderOne()
@@ -73,3 +92,13 @@ def test() -> None:
 
     assert len(data_providing_state.registered_data_providers) == 3
     assert observer.updated == 5
+
+    with pytest.raises(Exception):
+        data_providing_state.register_data_providers([
+            WrongDataProviderOne()
+        ])
+
+    with pytest.raises(Exception):
+        data_providing_state.register_data_providers([
+            WrongDataProviderTwo()
+        ])

tests/templates/states/test_data_providing_state.py

@@ -0,0 +1,105 @@
+import pytest
+from time import sleep
+from investing_algorithm_framework.templates.states.ordering_state \
+    import OrderingState
+from investing_algorithm_framework.templates.order_executors \
+    import OrderExecutor, ScheduledOrderExecutor, RelationalOrderExecutor
+from investing_algorithm_framework.core.utils import TimeUnit
+from investing_algorithm_framework.core.context import Context
+from investing_algorithm_framework.core.events import Observer
+from investing_algorithm_framework.core.workers import Worker
+
+
+class TestObserver(Observer):
+    updated: int = 0
+
+    def update(self, observable, **kwargs) -> None:
+        TestObserver.updated += 1
+
+
+class CustomOrderingState(OrderingState):
+
+    def get_transition_state_class(self):
+        return None
+
+
+class TestOrderExecutor(OrderExecutor):
+
+    id = 'TestOrderExecutorOne'
+
+    def execute_orders(self) -> None:
+        pass
+
+
+class TestScheduledOrderExecutor(ScheduledOrderExecutor):
+
+    id = 'TestOrderExecutorTwo'
+    time_interval = 1
+    time_unit = TimeUnit.SECOND
+
+    def execute_orders(self) -> None:
+        pass
+
+
+class TestRelationalOrderExecutor(RelationalOrderExecutor):
+    id = 'TestOrderExecutorThree'
+    run_after = TestScheduledOrderExecutor
+
+    def execute_orders(self) -> None:
+        pass
+
+
+class WrongOrderExecutorOne:
+
+    def execute_orders(self) -> None:
+        pass
+
+
+class WrongOrderExecutorTwo(Worker):
+
+    def work(self, **kwargs) -> None:
+        pass
+
+    def execute_orders(self) -> None:
+        pass
+
+
+def test() -> None:
+    observer = TestObserver()
+    order_executor_one = TestOrderExecutor()
+    order_executor_one.add_observer(observer)
+    order_executor_two = TestScheduledOrderExecutor()
+    order_executor_two.add_observer(observer)
+    order_executor_three = TestRelationalOrderExecutor()
+    order_executor_three.add_observer(observer)
+
+    context = Context()
+    CustomOrderingState.register_order_executors(
+        [order_executor_one, order_executor_two]
+    )
+    ordering_state = CustomOrderingState(context)
+    ordering_state.start()
+
+    assert len(ordering_state.registered_order_executors) == 2
+    assert observer.updated == 2
+
+    sleep(1)
+
+    CustomOrderingState.register_order_executors(
+        [order_executor_one, order_executor_two, order_executor_three]
+    )
+    ordering_state = CustomOrderingState(context)
+    ordering_state.start()
+
+    assert len(ordering_state.registered_order_executors) == 3
+    assert observer.updated == 5
+
+    with pytest.raises(Exception):
+        ordering_state.register_order_executors([
+            WrongOrderExecutorOne()
+        ])
+
+    with pytest.raises(Exception):
+        ordering_state.register_order_executors([
+            WrongOrderExecutorTwo()
+        ])