backend/tests/test_query_parser.py

import pytest
from backend.app.engine.query_parser import QueryParser, ParsedQuery


@pytest.fixture
def parser():
    return QueryParser()


class TestCategoryParsing:
    def test_dress(self, parser):
        result = parser.parse("black midi dress")
        assert result.category_filter == "Dresses"

    def test_jacket(self, parser):
        result = parser.parse("leather jacket")
        assert result.category_filter == "Coats & Jackets"

    def test_jeans(self, parser):
        result = parser.parse("blue jeans")
        assert result.category_filter == "Jeans"

    def test_hoodie(self, parser):
        result = parser.parse("oversized hoodie")
        assert result.category_filter == "Hoodies & Sweatshirts"

    def test_trainers(self, parser):
        result = parser.parse("white trainers")
        assert result.category_filter == "Shoes"

    def test_bag(self, parser):
        result = parser.parse("leather bag")
        assert result.category_filter == "Bags"

    def test_no_category(self, parser):
        result = parser.parse("something nice")
        assert result.category_filter is None

    def test_multi_word_category(self, parser):
        result = parser.parse("puffer jacket warm")
        assert result.category_filter == "Coats & Jackets"


class TestColorParsing:
    def test_basic_color(self, parser):
        result = parser.parse("black dress")
        assert result.color_filter == "black"

    def test_synonym_color(self, parser):
        result = parser.parse("scarlet top")
        assert result.color_filter == "red"

    def test_navy(self, parser):
        result = parser.parse("navy blazer")
        assert result.color_filter == "navy"

    def test_multi_word_color(self, parser):
        result = parser.parse("sky blue dress")
        assert result.color_filter == "blue"

    def test_no_color(self, parser):
        result = parser.parse("casual hoodie")
        assert result.color_filter is None


class TestPriceParsing:
    def test_under(self, parser):
        result = parser.parse("dress under £40")
        assert result.price_max == 40.0
        assert result.price_min is None

    def test_over(self, parser):
        result = parser.parse("jacket over £100")
        assert result.price_min == 100.0

    def test_range(self, parser):
        result = parser.parse("shoes £20-£50")
        assert result.price_min == 20.0
        assert result.price_max == 50.0

    def test_budget(self, parser):
        result = parser.parse("budget dress")
        assert result.price_max == 30.0

    def test_luxury(self, parser):
        result = parser.parse("luxury jacket")
        assert result.price_min == 100.0

    def test_no_price(self, parser):
        result = parser.parse("blue dress")
        assert result.price_min is None
        assert result.price_max is None


class TestGenderParsing:
    def test_mens(self, parser):
        result = parser.parse("mens hoodie")
        assert result.gender_filter == "Men"

    def test_womens(self, parser):
        result = parser.parse("for women dress")
        assert result.gender_filter == "Women"

    def test_ladies(self, parser):
        result = parser.parse("ladies dress elegant")
        assert result.gender_filter == "Women"

    def test_no_gender(self, parser):
        result = parser.parse("casual hoodie")
        assert result.gender_filter is None


class TestMaterialParsing:
    def test_silk(self, parser):
        result = parser.parse("silk midi dress")
        assert result.material_filter == "silk"

    def test_leather(self, parser):
        result = parser.parse("leather jacket")
        assert result.material_filter == "leather"

    def test_denim(self, parser):
        result = parser.parse("denim jacket")
        assert result.material_filter == "denim"

    def test_no_material(self, parser):
        result = parser.parse("black dress")
        assert result.material_filter is None


class TestSizeParsing:
    def test_named_size(self, parser):
        result = parser.parse("size small hoodie")
        assert result.size_filter == "S"

    def test_numeric_size(self, parser):
        result = parser.parse("size 10 dress")
        assert result.size_filter == "10"

    def test_xl(self, parser):
        result = parser.parse("XL casual shirt")
        assert result.size_filter == "XL"

    def test_no_size(self, parser):
        result = parser.parse("black dress")
        assert result.size_filter is None


class TestExclusions:
    def test_not(self, parser):
        result = parser.parse("black dress not floral")
        assert "floral" in result.exclusions

    def test_without(self, parser):
        result = parser.parse("jacket without leather")
        assert "leather" in result.exclusions

    def test_no_keyword(self, parser):
        result = parser.parse("summer top no black")
        assert "black" in result.exclusions

    def test_material_exclusion_conflict(self, parser):
        result = parser.parse("jacket not cotton")
        assert result.material_filter is None
        assert "cotton" in result.exclusions

    def test_no_exclusions(self, parser):
        result = parser.parse("black dress")
        assert result.exclusions == []


class TestStyleTags:
    def test_single_tag(self, parser):
        result = parser.parse("casual hoodie")
        assert "casual" in result.style_tags

    def test_multiple_tags(self, parser):
        result = parser.parse("vintage boho dress")
        assert "vintage" in result.style_tags
        assert "boho" in result.style_tags

    def test_no_tags(self, parser):
        result = parser.parse("blue jeans")
        assert "casual" not in result.style_tags


class TestVibeText:
    def test_preserves_raw_query(self, parser):
        result = parser.parse("black dress under £40")
        assert result.raw_query == "black dress under £40"

    def test_vibe_not_empty(self, parser):
        result = parser.parse("black dress")
        assert len(result.vibe_text) > 0