import pytest
from src.fmt.company_formatter import CompanyFormatter

def test_company_format():
    company = {'name': 'Pizza company'}

    formatter = CompanyFormatter()

    company_fmt = formatter.format(company)

    assert '- Pizza company\n' == company_fmt

def test_format_companies():
    companies = {'data': [
        {'name': 'Pizza company'},
        {'name': 'Donuts company'}
    ]}

    expected_format = "".join([
        "# Компанії\n\n",
        "- Pizza company\n",
        "- Donuts company\n",
    ])

    formatter = CompanyFormatter()

    companies_fmt = formatter.format_all(companies)

    assert expected_format == companies_fmt