import pytest

from pygenomeviz.track import FeatureTrack


def test_add_subtrack():
    """Test add subtrack"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    track.add_subtrack()
    track.add_subtrack(name="subtrack2", ratio=0.7)

    assert len(track.subtracks) == 2
    assert track.subtracks[0].name == f"{track_name}_subtrack1"
    assert track.subtracks[0].size == track_size
    assert track.subtracks[0].ratio == 1.0
    assert track.subtracks[1].name == "subtrack2"
    assert track.subtracks[1].size == track_size
    assert track.subtracks[1].ratio == 0.7


def test_add_subtrack_name_dup_error():
    """Test add subtrack name duplication error"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    track.add_subtrack(name="dup_subtrack")
    with pytest.raises(ValueError):
        track.add_subtrack(name="dup_subtrack")


def test_get_subtrack():
    """Test get subtrack"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    track.add_subtrack(name="subtrack1")
    track.add_subtrack(name="subtrack2")

    target_subtrack_name = "subtrack2"
    subtrack = track.get_subtrack(target_subtrack_name)
    assert subtrack.name == target_subtrack_name
    assert subtrack.size == track_size
    assert subtrack.ratio == 1.0


def test_set_sublabel():
    """Test set sublabel"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    # Default text
    track.set_sublabel()
    assert track._sublabel_text == "0 - 1,000 bp"
    # User defined text
    sublabel_text = "sublabel"
    track.set_sublabel(text=sublabel_text)
    assert track._sublabel_text == sublabel_text


def test_set_sublabel_position_error():
    """Test set_sublabel position error"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    vpos_types, hpos_types = ("top", "bottom"), ("left", "center", "right")
    # Valid position
    for vpos in vpos_types:
        for hpos in hpos_types:
            track.set_sublabel(position=f"{vpos}-{hpos}")
    # Invalid position
    with pytest.raises(ValueError):
        track.set_sublabel(position="top left")
    with pytest.raises(ValueError):
        track.set_sublabel(position="invalid-position")


def test_get_subtrack_no_exists_error():
    """Test get subtrack no exists error"""
    track_name, track_size = "feature track", 1000
    track = FeatureTrack(track_name, track_size)
    track.add_subtrack(name="subtrack1")
    track.add_subtrack(name="subtrack2")

    with pytest.raises(ValueError):
        no_exists_name = "subtrack3"
        track.get_subtrack(no_exists_name)


def test_add_feature_range_error():
    """Test add feature range check"""
    track_name, track_size = "test", 1000
    under_track_size, over_track_size = -100, track_size + 100
    track = FeatureTrack(track_name, track_size)
    # Case1. start < 0
    with pytest.raises(ValueError):
        track.add_feature(under_track_size, 100)
    # Case2: end > track_size
    with pytest.raises(ValueError):
        track.add_feature(100, over_track_size)
    # Case3: end > start
    with pytest.raises(ValueError):
        track.add_feature(500, 400)
    # Case4: start < 0 and end > track_size
    with pytest.raises(ValueError):
        track.add_feature(under_track_size, over_track_size)


def test_add_exon_feature_range_error():
    """Test add exon feature range check"""
    track_name, track_size = "test", 1000
    under_track_size, over_track_size = -100, track_size + 100
    track = FeatureTrack(track_name, track_size)
    # Case1. start < 0
    with pytest.raises(ValueError):
        exon_regions = [(under_track_size, 200), (300, 600), (650, 900)]
        track.add_exon_feature(exon_regions)
    # Case2: end > track_size
    with pytest.raises(ValueError):
        exon_regions = [(100, 200), (300, 600), (650, over_track_size)]
        track.add_exon_feature(exon_regions)
    # Case3: end > start
    with pytest.raises(ValueError):
        exon_regions = [(100, 200), (600, 300), (650, 900)]
        track.add_exon_feature(exon_regions)
    # Case4: start < 0 and end > track_size
    with pytest.raises(ValueError):
        exon_regions = [(under_track_size, 200), (600, 300), (650, over_track_size)]
        track.add_exon_feature(exon_regions)
