import pytest


@pytest.mark.django_db
def test_graph_empty(api_client):
    response = api_client.get('/api/v1/graph/')
    assert response.status_code == 200
    assert response.data == {'nodes': [], 'edges': []}


@pytest.mark.django_db
def test_graph_with_data(api_client, sample_relation):
    response = api_client.get('/api/v1/graph/')
    assert response.status_code == 200
    assert len(response.data['nodes']) == 2
    assert len(response.data['edges']) == 1


@pytest.mark.django_db
def test_legacy_api_path(api_client):
    response = api_client.get('/api/graph/')
    assert response.status_code == 200