from dapd_transformation_service.entities.dimension_meta import TrackPosition, TrackPositions


def test_to_plain_list():
    track_positions = TrackPositions()
    for idx in range(0, 3):
        track_positions.add(position=idx, track_id=f'track_id_{idx}')

    plain_list = track_positions.to_plain_list()

    assert len(plain_list) == 3
    for idx in range(0, 3):
        assert plain_list[idx] == {
            'position': idx,
            'track_id': f'track_id_{idx}',
        }


def test_from_plain_list():
    plain_list = [{
        'position': idx,
        'track_id': f'track_id_{idx}',
    } for idx in range(0, 3)]

    track_positions = TrackPositions()
    track_positions.from_plain_list(plain_list)

    for idx, track_position in enumerate(track_positions):
        assert isinstance(track_position, TrackPosition)
        assert track_position.position == idx
        assert track_position.track_id == f'track_id_{idx}'