import contextlib
import os
import re

import pytest

from .common import (app, exec_command, pod)
from .test_marathon import (_list_tasks)

_PODS_ENABLED = 'DCOS_PODS_ENABLED' in os.environ
list_regex = '/stuck-(?:sleep|pod)\W+[^Z]+Z\W+9\W+(?:True|False)' \
             '\W+\d\W+\d\W+[^Z]+Z\W+[^Z]+Z'


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_list():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'list'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'WAITING' in decoded
        assert '/stuck-sleep' in decoded
        """A line in the output looks like
        /stuck-sleep $since 9 True 4 3 $last_unsued_offer $last_used_offer
        Therefore `list_regex` tests this
        """
        assert re.search(list_regex, decoded) is not None


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_list_json():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'list', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'overdue' in decoded
        assert '/stuck-sleep' in decoded
        assert '"reason": "UnfulfilledConstraint"' in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_list_pod():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'list'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'WAITING' in decoded
        assert '/stuck-pod' in decoded
        """A line in the output looks like
        /stuck-sleep $since 9 True 4 3 $last_unsued_offer $last_used_offer
        Therefore `list_regex` tests this
        """
        assert re.search(list_regex, decoded) is not None


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_list_pod_json():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'list', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'overdue' in decoded
        assert '/stuck-pod' in decoded
        assert '"reason": "UnfulfilledConstraint"' in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_summary():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'summary', '/stuck-sleep'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'CONSTRAINTS' in decoded
        assert "[['hostname', 'UNIQUE']]" in decoded
        assert '0.00%' in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_summary_json():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'summary', '/stuck-sleep', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode().replace(' ', '').replace('\n', '')
        assert '"reason":"UnfulfilledConstraint"' in decoded
        assert '{"declined":0,"processed":0,"reason":"InsufficientCpus"}' \
               in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_summary_pod():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'summary', '/stuck-pod'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        assert 'CONSTRAINTS' in decoded
        assert "'operator': 'UNIQUE'" in decoded
        assert "'fieldName': 'hostname'" in decoded
        assert '0.00%' in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_summary_pod_json():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'summary', '/stuck-pod', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode().replace(' ', '').replace('\n', '')
        assert '"reason":"UnfulfilledConstraint"' in decoded
        assert '{"declined":0,"processed":0,"reason":"InsufficientCpus"}' \
               in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_details():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'details', '/stuck-sleep'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        """The public agent has all resources to launch this task,
        but not the matching role, therefore the output should be:
        ok        -        ok    ok   ok    ok
        To avoid formatting issues, whitspaces are ignored.
        """
        assert 'ok-okokokok' in decoded.replace(' ', '')
        """We do have 3 lines. The headline and two lines for the agents.
        If we split the decoded output by line break, there should be four
        entries in the array. The additional entry is empty.
        """
        assert len(decoded.split('\n')) == 4


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_details_json():
    with _stuck_app():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'details', '/stuck-sleep', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode().replace(' ', '').replace('\n', '')
        assert '"reason":"UnfulfilledConstraint"' in decoded
        assert '{"declined":0,"processed":0,"reason":"InsufficientCpus"}' \
               in decoded


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_details_pod():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'details', '/stuck-pod'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode()
        """The public agent has all resources to launch this task,
        but not the matching role, therefore the output should be:
        ok        -        ok    ok   ok    ok
        To avoid formatting issues, whitspaces are ignored.
        """
        assert 'ok-okokokok' in decoded.replace(' ', '')
        """We do have 3 lines. The headline and two lines for the agents.
        If we split the decoded output by line break, there should be four
        entries in the array. The additional entry is empty.
        """
        assert len(decoded.split('\n')) == 4


@pytest.mark.skipif(not _PODS_ENABLED, reason="Requires pods")
def test_debug_details_pod_json():
    with _stuck_pod():
        returncode, stdout, stderr = exec_command(
            ['dcos', 'marathon', 'debug', 'details', '/stuck-pod', '--json'])

        assert returncode == 0
        assert stderr == b''

        decoded = stdout.decode().replace(' ', '').replace('\n', '')
        assert '"reason":"UnfulfilledConstraint"' in decoded
        assert '{"declined":0,"processed":0,"reason":"InsufficientCpus"}' \
               in decoded


@contextlib.contextmanager
def _stuck_app(max_count=300):
    with app('tests/data/marathon/apps/stuck_sleep.json',
             'stuck-sleep', False):
        count = 0
        while count < max_count:
            tasks = _list_tasks(app_id='stuck-sleep')
            if (len(tasks) == 1):
                break
        yield


@contextlib.contextmanager
def _stuck_pod(max_count=300):
    with pod('tests/data/marathon/pods/stuck_sleep.json',
             '/stuck-pod', False):
        count = 0
        while count < max_count:
            returncode, stdout, stderr = exec_command(
                ['dcos', 'marathon', 'pod', 'list'])
            if '/stuck-pod1' in str(stdout).replace(' ', ''):
                break
        yield