"""Tests for output formatting module."""

from cmdparse.formatters import format_csv, format_data, format_json, format_raw, format_yaml


class TestFormatJson:
    def test_format_simple_list(self):
        data = [{'name': 'John', 'age': 30}]
        result = format_json(data)
        assert 'John' in result
        assert '30' in result

    def test_format_empty_list(self):
        result = format_json([])
        assert result == '[]'

    def test_pretty_format(self):
        data = [{'name': 'John'}]
        result = format_json(data, pretty=True)
        assert '\n' in result

    def test_compact_format(self):
        data = [{'name': 'John'}]
        result = format_json(data, pretty=False)
        assert '\n' not in result


class TestFormatYaml:
    def test_format_simple_list(self):
        data = [{'name': 'John', 'age': 30}]
        result = format_yaml(data)
        assert 'John' in result
        assert 'age' in result

    def test_format_empty_list(self):
        result = format_yaml([])
        assert result == '[]\n'

    def test_format_nested_structure(self):
        data = [{'user': {'name': 'John', 'tags': ['a', 'b']}}]
        result = format_yaml(data)
        assert 'John' in result


class TestFormatCsv:
    def test_format_simple_list(self):
        data = [{'name': 'John', 'age': 30}]
        result = format_csv(data)
        assert 'name' in result
        assert 'John' in result
        assert 'age' in result

    def test_format_empty_list(self):
        result = format_csv([])
        assert result == ''

    def test_format_multiple_rows(self):
        data = [
            {'name': 'John', 'age': 30},
            {'name': 'Jane', 'age': 25}
        ]
        result = format_csv(data)
        assert 'John' in result
        assert 'Jane' in result


class TestFormatRaw:
    def test_format_simple_list(self):
        data = [{'name': 'John', 'age': 30}]
        result = format_raw(data)
        assert 'name: John' in result
        assert 'age: 30' in result

    def test_format_empty_list(self):
        result = format_raw([])
        assert result == ''


class TestFormatData:
    def test_json_format(self):
        data = [{'name': 'John'}]
        result = format_data(data, 'json')
        assert 'John' in result

    def test_yaml_format(self):
        data = [{'name': 'John'}]
        result = format_data(data, 'yaml')
        assert 'John' in result

    def test_csv_format(self):
        data = [{'name': 'John'}]
        result = format_data(data, 'csv')
        assert 'name' in result

    def test_raw_format(self):
        data = [{'name': 'John'}]
        result = format_data(data, 'raw')
        assert 'John' in result

    def test_default_to_json(self):
        data = [{'name': 'John'}]
        result = format_data(data, '')
        assert 'John' in result

    def test_unknown_format_defaults_to_json(self):
        data = [{'name': 'John'}]
        result = format_data(data, 'unknown')
        assert 'John' in result