148 lines
5.1 KiB
Python
148 lines
5.1 KiB
Python
# tests/test_stock.py
|
|
import pytest
|
|
from models.database_models import Stock
|
|
|
|
|
|
class TestStockAPI:
|
|
def test_set_stock_new(self, client, auth_headers):
|
|
"""Test setting stock for new GTIN"""
|
|
stock_data = {
|
|
"gtin": "1234567890123",
|
|
"location": "WAREHOUSE_A",
|
|
"quantity": 100
|
|
}
|
|
|
|
response = client.post("/api/v1/stock", headers=auth_headers, json=stock_data)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["gtin"] == "1234567890123"
|
|
assert data["location"] == "WAREHOUSE_A"
|
|
assert data["quantity"] == 100
|
|
|
|
def test_set_stock_existing(self, client, auth_headers, db):
|
|
"""Test updating existing stock"""
|
|
# Create initial stock
|
|
stock = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
db.add(stock)
|
|
db.commit()
|
|
|
|
stock_data = {
|
|
"gtin": "1234567890123",
|
|
"location": "WAREHOUSE_A",
|
|
"quantity": 75
|
|
}
|
|
|
|
response = client.post("/api/v1/stock", headers=auth_headers, json=stock_data)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["quantity"] == 75 # Should be replaced, not added
|
|
|
|
def test_add_stock(self, client, auth_headers, db):
|
|
"""Test adding to existing stock"""
|
|
# Create initial stock
|
|
stock = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
db.add(stock)
|
|
db.commit()
|
|
|
|
stock_data = {
|
|
"gtin": "1234567890123",
|
|
"location": "WAREHOUSE_A",
|
|
"quantity": 25
|
|
}
|
|
|
|
response = client.post("/api/v1/stock/add", headers=auth_headers, json=stock_data)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["quantity"] == 75 # 50 + 25
|
|
|
|
def test_remove_stock(self, client, auth_headers, db):
|
|
"""Test removing from existing stock"""
|
|
# Create initial stock
|
|
stock = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
db.add(stock)
|
|
db.commit()
|
|
|
|
stock_data = {
|
|
"gtin": "1234567890123",
|
|
"location": "WAREHOUSE_A",
|
|
"quantity": 15
|
|
}
|
|
|
|
response = client.post("/api/v1/stock/remove", headers=auth_headers, json=stock_data)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["quantity"] == 35 # 50 - 15
|
|
|
|
def test_remove_stock_insufficient(self, client, auth_headers, db):
|
|
"""Test removing more stock than available"""
|
|
# Create initial stock
|
|
stock = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=10)
|
|
db.add(stock)
|
|
db.commit()
|
|
|
|
stock_data = {
|
|
"gtin": "1234567890123",
|
|
"location": "WAREHOUSE_A",
|
|
"quantity": 20
|
|
}
|
|
|
|
response = client.post("/api/v1/stock/remove", headers=auth_headers, json=stock_data)
|
|
|
|
assert response.status_code == 400
|
|
assert "Insufficient stock" in response.json()["detail"]
|
|
|
|
def test_get_stock_by_gtin(self, client, auth_headers, db):
|
|
"""Test getting stock summary for GTIN"""
|
|
# Create stock in multiple locations
|
|
stock1 = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
stock2 = Stock(gtin="1234567890123", location="WAREHOUSE_B", quantity=25)
|
|
db.add_all([stock1, stock2])
|
|
db.commit()
|
|
|
|
response = client.get("/api/v1/stock/1234567890123", headers=auth_headers)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["gtin"] == "1234567890123"
|
|
assert data["total_quantity"] == 75
|
|
assert len(data["locations"]) == 2
|
|
|
|
def test_get_total_stock(self, client, auth_headers, db):
|
|
"""Test getting total stock for GTIN"""
|
|
# Create stock in multiple locations
|
|
stock1 = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
stock2 = Stock(gtin="1234567890123", location="WAREHOUSE_B", quantity=25)
|
|
db.add_all([stock1, stock2])
|
|
db.commit()
|
|
|
|
response = client.get("/api/v1/stock/1234567890123/total", headers=auth_headers)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["gtin"] == "1234567890123"
|
|
assert data["total_quantity"] == 75
|
|
assert data["locations_count"] == 2
|
|
|
|
def test_get_all_stock(self, client, auth_headers, db):
|
|
"""Test getting all stock entries"""
|
|
# Create some stock entries
|
|
stock1 = Stock(gtin="1234567890123", location="WAREHOUSE_A", quantity=50)
|
|
stock2 = Stock(gtin="9876543210987", location="WAREHOUSE_B", quantity=25)
|
|
db.add_all([stock1, stock2])
|
|
db.commit()
|
|
|
|
response = client.get("/api/v1/stock", headers=auth_headers)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert len(data) == 2
|
|
|
|
def test_stock_requires_auth(self, client):
|
|
"""Test that stock endpoints require authentication"""
|
|
response = client.get("/api/v1/stock")
|
|
assert response.status_code == 403
|