graphene-django/examples/starwars/tests/test_connections.py

import pytest

from ..data import initialize
from ..schema import schema

pytestmark = pytest.mark.django_db


def test_correct_fetch_first_ship_rebels():
    initialize()
    query = '''
    query RebelsShipsQuery {
      rebels {
        name,
        hero {
          name
        }
        ships(first: 1) {
          edges {
            node {
              name
            }
          }
        }
      }
    }
    '''
    expected = {
        'rebels': {
            'name': 'Alliance to Restore the Republic',
            'hero': {
                'name': 'Human'
            },
            'ships': {
                'edges': [
                    {
                        'node': {
                            'name': 'X-Wing'
                        }
                    }
                ]
            }
        }
    }
    result = schema.execute(query)
    assert not result.errors
    assert result.data == expected


def test_correct_list_characters():
    initialize()
    query = '''
    query RebelsShipsQuery {
      node(id: "U2hpcDox") {
        ... on Ship {
          name
          characters {
            name
          }
        }
      }
    }
    '''
    expected = {
        'node': {
            'name': 'X-Wing',
            'characters': [{
                'name': 'Human'
            }],
        }
    }
    result = schema.execute(query)
    assert not result.errors
    assert result.data == expected