from io import BytesIO

import pytest
from openpyxl import load_workbook

from copanier.models import Delivery, Order, ProductOrder

pytestmark = pytest.mark.asyncio


async def test_empty_home(client):
    resp = await client.get("/")
    assert resp.status == 200


async def test_home_should_list_active_delivery(client, delivery):
    delivery.persist()
    resp = await client.get("/")
    assert resp.status == 200
    assert delivery.producer in resp.body.decode()


async def test_home_should_redirect_to_login_if_not_logged(client):
    client.logout()
    resp = await client.get("/")
    assert resp.status == 302
    assert resp.headers["Location"] == "/sésame?next=/"


async def test_create_delivery(client):
    assert not list(Delivery.all())
    body = {
        "producer": "Andines",
        "where": "Marché de la Briche",
        "date": "2019-02-23",
        "from_time": "18:30:00",
        "to_time": "20:00:00",
        "order_before": "2019-02-12",
    }
    resp = await client.post("/livraison", body=body)
    assert resp.status == 302
    assert len(list(Delivery.all())) == 1
    delivery = list(Delivery.all())[0]
    assert delivery.producer == "Andines"
    assert delivery.from_date.year == 2019
    assert delivery.from_date.hour == 18
    assert delivery.from_date.minute == 30
    assert delivery.to_date.year == 2019
    assert delivery.to_date.hour == 20
    assert delivery.to_date.minute == 0


async def test_place_order_with_session(client, delivery):
    delivery.persist()
    body = {"123": "3"}
    resp = await client.post(f"/livraison/{delivery.id}/commander", body=body)
    assert resp.status == 302
    delivery = Delivery.load(id=delivery.id)
    assert delivery.orders["foo@bar.org"]
    assert delivery.orders["foo@bar.org"].products["123"].wanted == 3


async def test_place_empty_order(client, delivery):
    delivery.persist()
    resp = await client.post(f"/livraison/{delivery.id}/commander", body={})
    assert resp.status == 302
    delivery = Delivery.load(id=delivery.id)
    assert not delivery.orders


async def test_place_empty_order_should_delete_previous(client, delivery):
    delivery.orders["foo@bar.org"] = Order(products={"123": ProductOrder(wanted=1)})
    delivery.persist()
    resp = await client.post(f"/livraison/{delivery.id}/commander", body={})
    assert resp.status == 302
    delivery = Delivery.load(delivery.id)
    assert not delivery.orders


async def test_place_order_with_empty_string(client, delivery):
    delivery.persist()
    body = {"123": ""}  # User deleted the field value.
    resp = await client.post(f"/livraison/{delivery.id}/commander", body=body)
    assert resp.status == 302
    delivery = Delivery.load(id=delivery.id)
    assert not delivery.orders


async def test_change_paid_status_when_placing_order(client, delivery):
    delivery.persist()
    body = {"123": "3", "paid": 1}
    resp = await client.post(f"/livraison/{delivery.id}/commander", body=body)
    assert resp.status == 302
    delivery = Delivery.load(id=delivery.id)
    assert delivery.orders["foo@bar.org"]
    assert delivery.orders["foo@bar.org"].paid is True


async def test_export_products(client, delivery):
    delivery.persist()
    resp = await client.get(f"/livraison/{delivery.id}/exporter/produits")
    wb = load_workbook(filename=BytesIO(resp.body))
    assert list(wb.active.values) == [
        ("name", "ref", "price", "weight", "description", "url", "img"),
        ("Lait", "123", 1.5, None, None, None, None),
    ]