def test_company_dict():
"""Tests company_dict."""
obj = mock.Mock(
id=123,
name='Name',
alias='Trading',
spec_set=('id', 'name', 'alias'),
)
res = dict_utils.company_dict(obj)
assert res == {
'id': str(obj.id),
'name': obj.name,
'trading_name': obj.alias,
}
def test_company_dict(obj, expected_dict):
"""Tests for the company_dict function."""
company = dict_utils.company_dict(obj)
assert company == expected_dict
def _companies_list(companies):
return [dict_utils.company_dict(company) for company in companies.all()]
def _get_company_list(col):
return [dict_utils.company_dict(c) for c in col.all()]