# -*- coding: utf-8 -*-
"""Unit tests for Python API"""

from __future__ import unicode_literals
from __future__ import print_function


# Standard imports
import pytest
import tempfile

@pytest.fixture()
def feed_model_with_SCC():
    """Fixture for a Cadbiom model with 4 Strongly Connected Components

    :return: A list of SCC, and the corrected model with start nodes inserted.
    :rtype: <tuple <list>, <str>>
    """

    scc = [['I', 'K', 'J', 'L'], ['Y', 'X', 'Z']]
    model = """<?xml version = "1.0" encoding="ASCII" standalone="yes" ?>
<model xmlns="http://cadbiom" name="">
  <CSimpleNode name="A"/>
  <CSimpleNode name="B"/>
  <CSimpleNode name="E"/>
  <CSimpleNode name="D"/>
  <CSimpleNode name="C"/>
  <CSimpleNode name="Y"/>
  <CSimpleNode name="X"/>
  <CSimpleNode name="Z"/>
  <CSimpleNode name="I"/>
  <CSimpleNode name="K"/>
  <CSimpleNode name="J"/>
  <CSimpleNode name="L"/>
  <CStartNode name="__start__0"/>
  <CStartNode name="__start__1"/>
  <transition ori="I" ext="J" event="" condition=""/>
  <transition ori="L" ext="I" event="" condition=""/>
  <transition ori="K" ext="L" event="" condition=""/>
  <transition ori="X" ext="Z" event="" condition=""/>
  <transition ori="__start__1" ext="X" event="" condition=""/>
  <transition ori="J" ext="K" event="" condition=""/>
  <transition ori="B" ext="D" event="" condition=""/>
  <transition ori="A" ext="B" event="" condition="C"/>
  <transition ori="X" ext="C" event="" condition=""/>
  <transition ori="C" ext="B" event="" condition="A"/>
  <transition ori="I" ext="K" event="" condition=""/>
  <transition ori="D" ext="C" event="" condition=""/>
  <transition ori="J" ext="B" event="" condition=""/>
  <transition ori="__start__0" ext="I" event="" condition=""/>
  <transition ori="Y" ext="X" event="" condition=""/>
  <transition ori="Z" ext="Y" event="" condition=""/>
  <transition ori="D" ext="E" event="" condition=""/>
</model>
"""

    return scc, model


def test_SCC_search(feed_model_with_SCC):
    """Test the correction of a model by removing Strongly Connected Components

    - We add a start node for every SCC
    - We add a transition between the start node and the smallest node of every
    SCC (sorted in lexicogrpahic order)

    - Thus, we add xml preamble in the model

    Keep in mind that by testing the result of the export of a model, we also
    test its content...
    """

    import cadbiom.models.guard_transitions.analyser.model_corrections as mc

    model = feed_model_with_SCC[1]

    # Create the file model in /tmp/
    # Note: prevent the deletion of the file after the close() call
    fd_model = tempfile.NamedTemporaryFile(suffix='.bcx', delete=False)
    fd_model.write(
        """<model xmlns="http://cadbiom" name="">
    <CSimpleNode name="A"/>
    <CSimpleNode name="B"/>
    <CSimpleNode name="E"/>
    <CSimpleNode name="D"/>
    <CSimpleNode name="C"/>
    <CSimpleNode name="Y"/>
    <CSimpleNode name="X"/>
    <CSimpleNode name="Z"/>
    <CSimpleNode name="I"/>
    <CSimpleNode name="K"/>
    <CSimpleNode name="J"/>
    <CSimpleNode name="L"/>
    <transition name="" ori="A" ext="B" event="" condition="C"/>
    <transition name="" ori="B" ext="D" event="" condition=""/>
    <transition name="" ori="C" ext="B" event="" condition="A"/>
    <transition name="" ori="D" ext="C" event="" condition=""/>
    <transition name="" ori="D" ext="E" event="" condition=""/>
    <transition name="" ori="Z" ext="Y" event="" condition=""/>
    <transition name="" ori="Y" ext="X" event="" condition=""/>
    <transition name="" ori="X" ext="Z" event="" condition=""/>
    <transition name="" ori="X" ext="C" event="" condition=""/>
    <transition name="" ori="I" ext="J" event="" condition=""/>
    <transition name="" ori="L" ext="I" event="" condition=""/>
    <transition name="" ori="K" ext="L" event="" condition=""/>
    <transition name="" ori="J" ext="K" event="" condition=""/>
    <transition name="" ori="J" ext="B" event="" condition=""/>
    <transition name="" ori="I" ext="K" event="" condition=""/>
    </model>"""
    )
    fd_model.close()


    # Make a new model file (with "_without_scc" suffix in filename)
    mc.add_start_nodes(fd_model.name) # Filename + path

    expected_lines = set(model.split("\n"))

    with open(fd_model.name[:-4] + "_without_scc.bcx", 'r') as file:
        found_lines = set(file.read().split("\n"))
        assert found_lines == expected_lines