From 7a34d49264abaa885069900dedc77761997a3aef Mon Sep 17 00:00:00 2001
From: Robin Richtsfeld <robin.richtsfeld@gmail.com>
Date: Sat, 13 May 2023 16:19:18 +0200
Subject: [PATCH] Fix `milvus_memory_test.py` mock `Config` (#3424)

Co-authored-by: k-boikov <64261260+k-boikov@users.noreply.github.com>
---
 tests/milvus_memory_test.py | 23 +++++++++++------------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/tests/milvus_memory_test.py b/tests/milvus_memory_test.py
index 46010f7e..9672326b 100644
--- a/tests/milvus_memory_test.py
+++ b/tests/milvus_memory_test.py
@@ -5,22 +5,21 @@ import sys
 import unittest
 
 try:
+    from autogpt.config import Config
     from autogpt.memory.milvus import MilvusMemory
 
-    def mock_config() -> dict:
+    def mock_config() -> Config:
         """Mock the config object for testing purposes."""
+
         # Return a mock config object with the required attributes
-        return type(
-            "MockConfig",
-            (object,),
-            {
-                "debug_mode": False,
-                "continuous_mode": False,
-                "speak_mode": False,
-                "milvus_collection": "autogpt",
-                "milvus_addr": "localhost:19530",
-            },
-        )
+        class MockConfig(Config):
+            debug_mode = False
+            continuous_mode = False
+            speak_mode = False
+            milvus_collection = "autogpt"
+            milvus_addr = "localhost:19530"
+
+        return MockConfig()
 
     class TestMilvusMemory(unittest.TestCase):
         """Tests for the MilvusMemory class."""