import pytest
from django.test import Client
from strawberry.django.test import GraphQLTestClient

from main.api.mutations import create_order
from main.models import Product, Order
from main.testing import MutationTest


@pytest.fixture(name="gql_client")
def gql_client_fixture(client: Client):
    yield GraphQLTestClient(client)


@pytest.fixture(name='product')
def product_fix(db):
    return Product.objects.create(name='borg', sale_price_ext=35, sale_price_int=30)


@pytest.fixture(name='create_order_mutation')
def create_order_mutation_fixture(product):
    fields = """
    ... on CreateOrderSuccess {
      order {
        id
        orderSum
        orderlines {
          id
          product {
            id
          }
        }
      }
    }"""
    # FIXME: maybe drop this helper and just use GraphQLTestClient
    return MutationTest(create_order, fields)


def test_create_order(product, create_order_mutation):
    assert Order.objects.count() == 0
    res = create_order_mutation.mutate(customerId="external", isExternal=True, orderLines=[{"productId": product.id, "amount": 1}])
    assert res
    assert not res.errors
    assert res.data
    order_lines = res.data.get('createOrder', {}).get('order', {}).get('orderlines', [])
    assert order_lines
    assert order_lines[0]['product']['id'] == str(product.id)
    assert Order.objects.count() == 1
    o = Order.objects.first()
    assert o.orderlines.first().product.id == product.id