from splitio.models.fallback_treatment import FallbackTreatment
from splitio.models.fallback_config import FallbackTreatmentsConfiguration, FallbackTreatmentCalculator

class FallbackTreatmentModelTests(object):
    """Fallback treatment model tests."""

    def test_working(self):
        fallback_treatment = FallbackTreatment("on", '{"prop": "val"}')
        assert fallback_treatment.config == '{"prop": "val"}'
        assert fallback_treatment.treatment == 'on'
        
        fallback_treatment = FallbackTreatment("off")
        assert fallback_treatment.config == None
        assert fallback_treatment.treatment == 'off'
        
class FallbackTreatmentsConfigModelTests(object):
    """Fallback treatment configuration model tests."""

    def test_working(self):
        global_fb = FallbackTreatment("on")
        flag_fb = FallbackTreatment("off")
        fallback_config = FallbackTreatmentsConfiguration(global_fb, {"flag1": flag_fb})
        assert fallback_config.global_fallback_treatment == global_fb
        assert fallback_config.by_flag_fallback_treatment == {"flag1": flag_fb}

        fallback_config.global_fallback_treatment = None
        assert fallback_config.global_fallback_treatment == None
        
        fallback_config.by_flag_fallback_treatment["flag2"] = flag_fb
        assert fallback_config.by_flag_fallback_treatment == {"flag1": flag_fb, "flag2": flag_fb}

        fallback_config = FallbackTreatmentsConfiguration("on", {"flag1": "off"})
        assert isinstance(fallback_config.global_fallback_treatment, FallbackTreatment)
        assert fallback_config.global_fallback_treatment.treatment == "on"
        
        assert isinstance(fallback_config.by_flag_fallback_treatment["flag1"], FallbackTreatment)
        assert fallback_config.by_flag_fallback_treatment["flag1"].treatment == "off"
        

class FallbackTreatmentCalculatorTests(object):
    """Fallback treatment calculator model tests."""

    def test_working(self):
        fallback_config = FallbackTreatmentsConfiguration(FallbackTreatment("on" ,"{}"), None)
        fallback_calculator = FallbackTreatmentCalculator(fallback_config)
        assert fallback_calculator.fallback_treatments_configuration == fallback_config
        assert fallback_calculator._label_prefix == "fallback - "
        
        fallback_treatment = fallback_calculator.resolve("feature", "not ready")
        assert fallback_treatment.treatment == "on"
        assert fallback_treatment.label == "fallback - not ready"
        assert fallback_treatment.config == "{}"
        
        fallback_calculator._fallback_treatments_configuration = FallbackTreatmentsConfiguration(FallbackTreatment("on" ,"{}"), {'feature': FallbackTreatment("off" , '{"prop": "val"}')})
        fallback_treatment = fallback_calculator.resolve("feature", "not ready")
        assert fallback_treatment.treatment == "off"
        assert fallback_treatment.label == "fallback - not ready"
        assert fallback_treatment.config == '{"prop": "val"}'
        
        fallback_treatment = fallback_calculator.resolve("feature2", "not ready")
        assert fallback_treatment.treatment == "on"
        assert fallback_treatment.label == "fallback - not ready"
        assert fallback_treatment.config == "{}"