import pytest
import re
from semver import VersionInfo

from ldclient.impl.model import *

from testing.builders import *


def test_flag_targets_are_stored_as_sets():
    flag = FlagBuilder("key") \
        .target(0, "a", "b") \
        .context_target("kind1", 0, "c", "d") \
        .build()
    assert flag.targets[0].values == {"a", "b"}
    assert flag.context_targets[0].values == {"c", "d"}

def test_segment_targets_are_stored_as_sets():
    segment = SegmentBuilder("key") \
        .included("a", "b") \
        .excluded("c", "d") \
        .included_contexts("kind1", "e", "f") \
        .excluded_contexts("kind2", "g", "h") \
        .build()
    assert segment.included == {"a", "b"}
    assert segment.excluded == {"c", "d"}
    assert segment.included_contexts[0].values == {"e", "f"}
    assert segment.excluded_contexts[0].values == {"g", "h"}

def test_clause_values_preprocessed_with_regex_operator():
    pattern_str = "^[a-z]*$"
    pattern = re.compile(pattern_str)
    flag = make_boolean_flag_with_clauses(make_clause(None, "attr", "matches", pattern_str, "?", True))
    assert flag.rules[0].clauses[0]._values == [pattern_str, "?", True]
    assert list(x.as_regex for x in flag.rules[0].clauses[0]._values_preprocessed) == [pattern, None, None]

@pytest.mark.parametrize('op', ['semVerEqual', 'semVerGreaterThan', 'semVerLessThan'])
def test_clause_values_preprocessed_with_semver_operator(op):
    flag = make_boolean_flag_with_clauses(make_clause(None, "attr", op, "1.2.3", 1, True))
    assert flag.rules[0].clauses[0]._values == ["1.2.3", 1, True]
    assert list(x.as_semver for x in flag.rules[0].clauses[0]._values_preprocessed) == [VersionInfo(1, 2, 3), None, None]

@pytest.mark.parametrize('op', ['before', 'after'])
def test_clause_values_preprocessed_with_time_operator(op):
    flag = make_boolean_flag_with_clauses(make_clause(None, "attr", op, 1000, "1970-01-01T00:00:02Z", True))
    assert flag.rules[0].clauses[0]._values == [1000, "1970-01-01T00:00:02Z", True]
    assert list(x.as_time for x in flag.rules[0].clauses[0]._values_preprocessed) == [1000, 2000, None]