From ec836b99cc6b08177d7542e23f45bf866461d60b Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Thu, 15 Aug 2024 14:45:13 -0700
Subject: [PATCH] feat: add test for --model-metadata-file option

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

diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 1f0d467b5..9a2034be2 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -1,6 +1,7 @@
 import os
 import subprocess
 import tempfile
+import json
 from io import StringIO
 from pathlib import Path
 from unittest import TestCase
@@ -489,3 +490,22 @@ class TestMain(TestCase):
                 self.assertIn(real_external_file_path, coder.abs_read_only_fnames)
         finally:
             os.unlink(external_file_path)
+
+    def test_model_metadata_file(self):
+        with GitTemporaryDirectory():
+            metadata_file = Path(".aider.model.metadata.json")
+            metadata_content = {
+                "deepseek-chat": {
+                    "max_input_tokens": 1234
+                }
+            }
+            metadata_file.write_text(json.dumps(metadata_content))
+
+            coder = main(
+                ["--model", "deepseek-chat", "--model-metadata-file", str(metadata_file), "--exit", "--yes"],
+                input=DummyInput(),
+                output=DummyOutput(),
+                return_coder=True,
+            )
+
+            self.assertEqual(coder.main_model.info["max_input_tokens"], 1234)