From 6a899e25430211cd505f5b2fa8a754a091fdd4c4 Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@sdg.moe>
Date: Thu, 11 Nov 2021 22:15:53 +0100
Subject: [PATCH] SearchAPI: Fixing tests of the episode search api

---
 src/Kyoo.Core/Controllers/Repositories/EpisodeRepository.cs | 5 ++++-
 tests/Kyoo.Tests/Database/SpecificTests/EpisodeTests.cs     | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/Kyoo.Core/Controllers/Repositories/EpisodeRepository.cs b/src/Kyoo.Core/Controllers/Repositories/EpisodeRepository.cs
index f95fbb1c..d54d251f 100644
--- a/src/Kyoo.Core/Controllers/Repositories/EpisodeRepository.cs
+++ b/src/Kyoo.Core/Controllers/Repositories/EpisodeRepository.cs
@@ -120,13 +120,16 @@ namespace Kyoo.Core.Controllers
 		/// <inheritdoc />
 		public override async Task<ICollection<Episode>> Search(string query)
 		{
-			return await _database.Episodes
+			List<Episode> ret = await _database.Episodes
 				.Include(x => x.Show)
 				.Where(x => x.EpisodeNumber != null || x.AbsoluteNumber != null)
 				.Where(_database.Like<Episode>(x => x.Title, $"%{query}%"))
 				.OrderBy(DefaultSort)
 				.Take(20)
 				.ToListAsync();
+			foreach (Episode ep in ret)
+				ep.Show.Episodes = null;
+			return ret;
 		}
 
 		/// <inheritdoc />
diff --git a/tests/Kyoo.Tests/Database/SpecificTests/EpisodeTests.cs b/tests/Kyoo.Tests/Database/SpecificTests/EpisodeTests.cs
index 58befa98..88d6cb84 100644
--- a/tests/Kyoo.Tests/Database/SpecificTests/EpisodeTests.cs
+++ b/tests/Kyoo.Tests/Database/SpecificTests/EpisodeTests.cs
@@ -338,6 +338,7 @@ namespace Kyoo.Tests.Database
 			};
 			await _repository.Create(value);
 			ICollection<Episode> ret = await _repository.Search(query);
+			value.Show = TestSample.Get<Show>();
 			KAssert.DeepEqual(value, ret.First());
 		}
 	}