from dcoscli import tables

from ..fixtures.marathon import (app_fixture, app_task_fixture,
                                 deployment_fixture, group_fixture)
from ..fixtures.node import slave_fixture
from ..fixtures.package import package_fixture, search_result_fixture
from ..fixtures.service import framework_fixture
from ..fixtures.task import task_fixture


def test_task_table():
    _test_table(tables.task_table,
                task_fixture,
                'tests/unit/data/task.txt')


def test_app_table():
    apps = [app_fixture()]
    deployments = []
    table = tables.app_table(apps, deployments)
    with open('tests/unit/data/app.txt') as f:
        assert str(table) == f.read()


def test_deployment_table():
    _test_table(tables.deployment_table,
                deployment_fixture,
                'tests/unit/data/deployment.txt')


def test_app_task_table():
    _test_table(tables.app_task_table,
                app_task_fixture,
                'tests/unit/data/app_task.txt')


def test_service_table():
    _test_table(tables.service_table,
                framework_fixture,
                'tests/unit/data/service.txt')


def test_group_table():
    _test_table(tables.group_table,
                group_fixture,
                'tests/unit/data/group.txt')


def test_package_table():
    _test_table(tables.package_table,
                package_fixture,
                'tests/unit/data/package.txt')


def test_package_search_table():
    _test_table(tables.package_search_table,
                search_result_fixture,
                'tests/unit/data/package_search.txt')


def test_node_table():
    _test_table(tables.slave_table,
                slave_fixture,
                'tests/unit/data/node.txt')


def _test_table(table_fn, fixture_fn, path):
    table = table_fn([fixture_fn()])
    with open(path) as f:
        assert str(table) == f.read()