tjdify/api/tests/unit_tests/utils/position_helper/test_position_helper.py

from textwrap import dedent

import pytest

from core.helper.position_helper import get_position_map, sort_and_filter_position_map


@pytest.fixture
def prepare_example_positions_yaml(tmp_path, monkeypatch) -> str:
    monkeypatch.chdir(tmp_path)
    tmp_path.joinpath("example_positions.yaml").write_text(dedent(
        """\
        - first
        - second
        # - commented
        - third
        
        - 9999999999999
        - forth
        """))
    return str(tmp_path)


@pytest.fixture
def prepare_empty_commented_positions_yaml(tmp_path, monkeypatch) -> str:
    monkeypatch.chdir(tmp_path)
    tmp_path.joinpath("example_positions_all_commented.yaml").write_text(dedent(
        """\
        # - commented1
        # - commented2
        - 
        -   
        
        """))
    return str(tmp_path)


def test_position_helper(prepare_example_positions_yaml):
    position_map = get_position_map(
        folder_path=prepare_example_positions_yaml,
        file_name='example_positions.yaml')
    assert len(position_map) == 4
    assert position_map == {
        'first': 0,
        'second': 1,
        'third': 2,
        'forth': 3,
    }


def test_position_helper_with_all_commented(prepare_empty_commented_positions_yaml):
    position_map = get_position_map(
        folder_path=prepare_empty_commented_positions_yaml,
        file_name='example_positions_all_commented.yaml')
    assert position_map == {}


def test_excluded_position_map(prepare_example_positions_yaml):
    position_map = get_position_map(
        folder_path=prepare_example_positions_yaml,
        file_name='example_positions.yaml'
    )
    pin_list = ['forth', 'first']
    include_list = []
    exclude_list = ['9999999999999']
    sorted_filtered_position_map = sort_and_filter_position_map(
        original_position_map=position_map,
        pin_list=pin_list,
        include_list=include_list,
        exclude_list=exclude_list
    )
    assert sorted_filtered_position_map == {
        'forth': 0,
        'first': 1,
        'second': 2,
        'third': 3,
    }


def test_included_position_map(prepare_example_positions_yaml):
    position_map = get_position_map(
        folder_path=prepare_example_positions_yaml,
        file_name='example_positions.yaml'
    )
    pin_list = ['second', 'first']
    include_list = ['first', 'second', 'third', 'forth']
    exclude_list = []
    sorted_filtered_position_map = sort_and_filter_position_map(
        original_position_map=position_map,
        pin_list=pin_list,
        include_list=include_list,
        exclude_list=exclude_list
    )
    assert sorted_filtered_position_map == {
        'second': 0,
        'first': 1,
        'third': 2,
        'forth': 3,
    }