import pytest
import json
import jsonpickle
from ldclient.evaluation import FeatureFlagsState

def test_can_get_flag_value():
    state = FeatureFlagsState(True)
    flag_state = { 'key': 'key', 'version': 100, 'value': 'value', 'variation': 1, 'reason': None }
    state.add_flag(flag_state, False, False)
    assert state.get_flag_value('key') == 'value'

def test_returns_none_for_unknown_flag():
    state = FeatureFlagsState(True)
    assert state.get_flag_value('key') is None

def test_can_convert_to_values_map():
    state = FeatureFlagsState(True)
    flag_state1 = { 'key': 'key1', 'version': 100, 'value': 'value1', 'variation': 0, 'reason': None }
    flag_state2 = { 'key': 'key2', 'version': 200, 'value': 'value2', 'variation': 1, 'reason': None }
    state.add_flag(flag_state1, False, False)
    state.add_flag(flag_state2, False, False)
    assert state.to_values_map() == { 'key1': 'value1', 'key2': 'value2' }

def test_can_convert_to_json_dict():
    state = FeatureFlagsState(True)
    flag_state1 = { 'key': 'key1', 'version': 100, 'trackEvents': False, 'value': 'value1', 'variation': 0, 'reason': None }
    flag_state2 = { 'key': 'key2', 'version': 200, 'trackEvents': True, 'debugEventsUntilDate': 1000, 'value': 'value2', 'variation': 1, 'reason': None }
    state.add_flag(flag_state1, False, False)
    state.add_flag(flag_state2, False, False)

    result = state.to_json_dict()
    assert result == {
        'key1': 'value1',
        'key2': 'value2',
        '$flagsState': {
            'key1': {
                'variation': 0,
                'version': 100
            },
            'key2': {
                'variation': 1,
                'version': 200,
                'trackEvents': True,
                'debugEventsUntilDate': 1000
            }
        },
        '$valid': True
    }

def test_can_convert_to_json_string():
    state = FeatureFlagsState(True)
    flag_state1 = { 'key': 'key1', 'version': 100, 'trackEvents': False, 'value': 'value1', 'variation': 0, 'reason': None }
    flag_state2 = { 'key': 'key2', 'version': 200, 'trackEvents': True, 'debugEventsUntilDate': 1000, 'value': 'value2', 'variation': 1, 'reason': None }
    state.add_flag(flag_state1, False, False)
    state.add_flag(flag_state2, False, False)

    obj = state.to_json_dict()
    str = state.to_json_string()
    assert json.loads(str) == obj

# We don't actually use jsonpickle in the SDK, but FeatureFlagsState has a magic method that makes it
# behave correctly in case the application uses jsonpickle to serialize it.
def test_can_serialize_with_jsonpickle():
    state = FeatureFlagsState(True)
    flag_state1 = { 'key': 'key1', 'version': 100, 'trackEvents': False, 'value': 'value1', 'variation': 0, 'reason': None }
    flag_state2 = { 'key': 'key2', 'version': 200, 'trackEvents': True, 'debugEventsUntilDate': 1000, 'value': 'value2', 'variation': 1, 'reason': None }
    state.add_flag(flag_state1, False, False)
    state.add_flag(flag_state2, False, False)

    obj = state.to_json_dict()
    str = jsonpickle.encode(state, unpicklable=False)
    assert json.loads(str) == obj