test_app.py

from datetime import datetime

import dateutil.parser
import pytest

from ex_back.types import OrderType


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order(client):
    # Test data
    order_data = {
        "type": "limit",
        "side": "buy",
        "instrument": "abcdefghijkl",
        "limit_price": 150.00,
        "quantity": 10,
    }
    # Make a request to the server
    response = client.post("/v1/orders/", json=order_data)

    # Print full response when test fails
    if response.status_code != 201:
        print(f"Response: {response.status_code}, {response.headers}, {response.text}")

    # Assert that the response status code is 201 (Created)
    assert response.status_code == 201

    # Assert that the response data matches the test data
    response_data = response.json()
    assert response_data["order"]["type"] == order_data["type"]
    assert response_data["order"]["side"] == order_data["side"]
    assert response_data["order"]["instrument"] == order_data["instrument"]
    assert response_data["order"]["limit_price"] == order_data["limit_price"]
    assert response_data["order"]["quantity"] == order_data["quantity"]
    assert "job_id" in response_data

    # Parse string to datetime object
    created_at = dateutil.parser.parse(response_data["order"]["created_at"])
    assert isinstance(created_at, datetime)


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order_invalid_limit_price_market_order(client):
    data = {
        "type": "market",
        "side": "buy",
        "instrument": "ABCDEF123456",
        "limit_price": 100.0,
        "quantity": 10,
    }
    response = client.post("/v1/orders", json=data)
    assert response.status_code == 422
    assert "Providing a `limit_price` is prohibited for type `market`" in str(
        response.content
    )


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order_missing_limit_price_limit_order(client):
    data = {
        "type": "limit",
        "side": "buy",
        "instrument": "ABCDEF123456",
        "quantity": 10,
    }
    response = client.post("/v1/orders", json=data)
    assert response.status_code == 422
    assert "Attribute `limit_price` is required for type `limit`" in str(
        response.content
    )


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order_invalid_instrument_short(client):
    data = {
        "type": "market",
        "side": "buy",
        "instrument": "ABC",  # invalid instrument
        "quantity": 10,
    }
    response = client.post("/v1/orders", json=data)
    assert response.status_code == 422
    assert "ensure this value has at least 12 characters" in str(response.content)


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order_invalid_instrument_long(client):
    data = {
        "type": "market",
        "side": OrderType.MARKET.value,
        "instrument": "ABCnthaoeurcg324shau",  # invalid instrument
        "quantity": 10,
    }
    response = client.post("/v1/orders", json=data)
    assert response.status_code == 422
    assert "ensure this value has at most 12 characters" in str(response.content)


@pytest.mark.skip(reason="Functionality not yet updated")
def test_create_order_invalid_quantity(client):
    data = {
        "type": "market",
        "side": "buy",
        "instrument": "ABCDEF123456",
        "quantity": 0,  # invalid quantity
    }
    response = client.post("/v1/orders", json=data)
    assert response.status_code == 422
    assert "ensure this value is greater than 0" in str(response.content)


@pytest.mark.skip(reason="Functionality not yet updated")
def test_list_orders(client, order_stub):
    # Given
    # Add an order to the database
    order_response = client.post("/v1/orders/", json=order_stub)
    assert order_response.status_code == 201
    expected_order = order_response.json()["order"]

    # When
    # Fetch orders from the '/orders' endpoint
    response = client.get("/v1/orders")

    # Then
    assert response.status_code == 200
    orders = response.json()

    # Check the returned orders contain the order we added
    assert len(orders) > 0
    assert expected_order in orders