fix(tmdb): Allow a broader range of totalSeasons values (#524)

* fix(tmdb): issue #523 allows a broader range of totalSeasons values * refactor(test): fix test file * chore(package): remove unused test scripts
consumet · May 20, 2024 · 69dda77 · 69dda77
1 parent 440cde1
commit 69dda77
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 13 deletions.
diff --git a/dist/providers/anime/gogoanime.js.map b/dist/providers/anime/gogoanime.js.map
diff --git a/dist/providers/meta/tmdb.js b/dist/providers/meta/tmdb.js
diff --git a/dist/providers/meta/tmdb.js.map b/dist/providers/meta/tmdb.js.map
diff --git a/src/providers/meta/tmdb.ts b/src/providers/meta/tmdb.ts
@@ -407,16 +407,13 @@ class TMDB extends MovieParser {
 
     // console.log({ test1: findMedia.results });
 
-    // check if the result contains the total number of seasons and compare it to the extraData by 1 up or down and make sure that its a number
+    // Check if the result contains the total number of seasons and compare it to the extraData.
+    // Allow for a range of ±2 seasons and ensure that the seasons value is a number.
     if (extraData && extraData.totalSeasons && extraData.type === TvType.TVSERIES) {
       findMedia.results = findMedia.results.filter(result => {
         const totalSeasons = (result.seasons as number) || 0;
         const extraDataSeasons = (extraData.totalSeasons as number) || 0;
-        return (
-          totalSeasons === extraDataSeasons ||
-          totalSeasons === extraDataSeasons + 1 ||
-          totalSeasons === extraDataSeasons - 1
-        );
+        return totalSeasons >= extraDataSeasons - 2 && totalSeasons <= extraDataSeasons + 2;
       });
     }
 

diff --git a/test/meta/tmdb.test.ts b/test/meta/tmdb.test.ts
@@ -33,7 +33,7 @@ test('returns a filled array of all trending list', async () => {
 });
 
 test('returns a filled object of anime data', async () => {
-  const data = await tmdb.fetchMediaInfo('60735', 'tv');
+  const data = await tmdb.fetchMediaInfo('85937', 'tv');
   expect(data).not.toBeNull();
   expect(data.episodes).not.toEqual([]);
   expect(data.description).not.toBeNull();