fix: the len() method of Variables should return the schema length.

barakalon · Jan 29, 2025 · 5a5160a · 5a5160a
1 parent 3ef51d3
commit 5a5160a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mysql_mimic/variables.py b/mysql_mimic/variables.py
@@ -82,7 +82,7 @@ def __iter__(self) -> Iterator[str]:
         return self._values.__iter__()
 
     def __len__(self) -> int:
-        return len(self._values)
+        return len(self.schema)
 
     def get_schema(self, name: str) -> VariableSchema:
         schema = self.schema.get(name)

diff --git a/tests/test_variables.py b/tests/test_variables.py
@@ -33,6 +33,7 @@ def test_parse_timezone() -> None:
 
 def test_variable_mapping() -> None:
     test_vars = TestVars()
+    assert test_vars
 
     assert test_vars.get_variable("foo") == "bar"
     assert test_vars["foo"] == "bar"