From 03012a81fdc0ffef457a1a19ba32bf7d5769011d Mon Sep 17 00:00:00 2001
From: Arnaud Gelas <arnaudgelas@gmail.com>
Date: Tue, 16 Jan 2024 21:44:45 +0100
Subject: [PATCH] In some cases when trying to create tests, metagpt crashes.

Adding some more safeguard to handle the case where code_doc is None.
---
 metagpt/roles/qa_engineer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/metagpt/roles/qa_engineer.py b/metagpt/roles/qa_engineer.py
index 81082ef59..0e323893e 100644
--- a/metagpt/roles/qa_engineer.py
+++ b/metagpt/roles/qa_engineer.py
@@ -63,6 +63,8 @@ class QaEngineer(Role):
             if not filename or "test" in filename:
                 continue
             code_doc = await src_file_repo.get(filename)
+            if not code_doc:
+                continue
             test_doc = await tests_file_repo.get("test_" + code_doc.filename)
             if not test_doc:
                 test_doc = Document(