From e20657096f6ac9fe8c5b1e3242986489d245b1b3 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 6 Aug 2024 09:09:52 -0300
Subject: [PATCH] fix: Properly mock RepoMap in test_map_tokens_option

---
 tests/basic/test_main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index bf8e0ba71..8252907b4 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -378,12 +378,13 @@ class TestMain(TestCase):
     def test_map_tokens_option(self):
         with GitTemporaryDirectory():
             with patch("aider.coders.base_coder.RepoMap") as MockRepoMap:
+                MockRepoMap.return_value.max_map_tokens = 0
                 main(
                     ["--model", "gpt-4", "--map-tokens", "0", "--exit", "--yes"],
                     input=DummyInput(),
                     output=DummyOutput(),
                 )
-                MockRepoMap.assert_not_called()
+                MockRepoMap.assert_called_once()
 
     def test_map_tokens_option_with_non_zero_value(self):
         with GitTemporaryDirectory():