From 21cbcd6c307ca472c2bcd15c13410e7d8a38b1c2 Mon Sep 17 00:00:00 2001
From: Benedikt Bartscher <bb@senbax.de>
Date: Wed, 30 Oct 2024 14:45:56 +0100
Subject: [PATCH] showcase objectvar issues

---
 tests/units/vars/test_object.py | 87 +++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 tests/units/vars/test_object.py

diff --git a/tests/units/vars/test_object.py b/tests/units/vars/test_object.py
new file mode 100644
index 000000000..ed678038e
--- /dev/null
+++ b/tests/units/vars/test_object.py
@@ -0,0 +1,87 @@
+import pytest
+
+import reflex as rx
+from reflex.utils.types import GenericType
+from reflex.vars.base import Var
+from reflex.vars.object import LiteralObjectVar, ObjectVar
+
+
+class Bare:
+    """A bare class with a single attribute."""
+
+    quantity: int = 0
+
+
+@rx.serializer
+def serialize_bare(obj: Bare) -> dict:
+    """A serializer for the bare class.
+
+    Args:
+        obj: The object to serialize.
+
+    Returns:
+        A dictionary with the quantity attribute.
+    """
+    return {"quantity": obj.quantity}
+
+
+class Base(rx.Base):
+    """A reflex base class with a single attribute."""
+
+    quantity: int = 0
+
+
+class ObjectState(rx.State):
+    """A reflex state with bare and base objects."""
+
+    bare: rx.Field[Bare] = rx.field(Bare())
+    base: rx.Field[Base] = rx.field(Base())
+
+
+@pytest.mark.parametrize("type_", [Base, Bare])
+def test_var_create(type_: GenericType) -> None:
+    my_object = type_()
+    var = Var.create(my_object)
+    assert var._var_type is type_
+
+    quantity = var.quantity
+    assert quantity._var_type is int
+
+
+@pytest.mark.parametrize("type_", [Base, Bare])
+def test_literal_create(type_: GenericType) -> None:
+    my_object = type_()
+    var = LiteralObjectVar.create(my_object)
+    assert var._var_type is type_
+
+    quantity = var.quantity
+    assert quantity._var_type is int
+
+
+@pytest.mark.parametrize("type_", [Base, Bare])
+def test_guess(type_: GenericType) -> None:
+    my_object = type_()
+    var = Var.create(my_object)
+    var = var.guess_type()
+    assert var._var_type is type_
+
+    quantity = var.quantity
+    assert quantity._var_type is int
+
+
+@pytest.mark.parametrize("type_", [Base, Bare])
+def test_state(type_: GenericType) -> None:
+    var = ObjectState.bare
+    assert var._var_type is type_
+
+    quantity = var.quantity
+    assert quantity._var_type is int
+
+
+@pytest.mark.parametrize("type_", [Base, Bare])
+def test_state_to_operation(type_: GenericType) -> None:
+    var = ObjectState.bare.to(ObjectVar, type_)
+    assert var._var_type is type_
+
+    var = ObjectState.bare.to(ObjectVar)
+    assert var._var_type is type_