"""Test get_complement function."""
import json

from territories.db import db_init


def test_get_complement_continent(
        db_fixture, orch_standard, fixture_client, header_fixture):
    """Test that complement results include the expected properties."""
    db_init.data_backfill()
    complement_response = fixture_client.get(
        '/complement/{}?territory_code_a2={}'.format(orch_standard, 'AF'),
        headers=header_fixture)
    response_json = json.loads(complement_response.data.decode('utf-8'))
    response_items = response_json.get('items')

    assert len(response_items) > 0
    for item in response_items:
        assert item.get('continent') is not None