Add transcript filter functionality to entity viewer (#465)

* ENSWBSITES-888: Filter transcript functionality and writing test for the functionality of filtering by so_term

src/ensembl/src/content/app/entity-viewer/gene-view/components/default-transcripts-list/DefaultTranscriptsList.tsx

@@ -25,6 +25,8 @@ import {
   defaultSort
 } from 'src/content/app/entity-viewer/shared/helpers/transcripts-sorter';
 
+import { filterTranscriptsBySOTerm } from 'src/content/app/entity-viewer/shared/helpers/transcripts-filter';
+
 import {
   getExpandedTranscriptIds,
   getExpandedTranscriptDownloadIds,
@@ -93,6 +95,9 @@ const DefaultTranscriptslist = (props: Props) => {
 
   const sortingFunction = transcriptSortingFunctions[sortingRule];
   const sortedTranscripts = sortingFunction(gene.transcripts) as Transcript[];
+  const filteredTranscripts = Object.values(filters).some(Boolean)
+    ? (filterTranscriptsBySOTerm(sortedTranscripts, filters) as Transcript[])
+    : sortedTranscripts;
 
   const [isFilterOpen, setFilterOpen] = useState(false);
 
@@ -101,7 +106,7 @@ const DefaultTranscriptslist = (props: Props) => {
 
     // Expand the first transcript by default
     if (!hasExpandedTranscripts) {
-      dispatch(toggleTranscriptInfo(sortedTranscripts[0].stable_id));
+      dispatch(toggleTranscriptInfo(filteredTranscripts[0].stable_id));
     }
   }, []);
 
@@ -129,7 +134,7 @@ const DefaultTranscriptslist = (props: Props) => {
           <TranscriptsFilter
             label={filterLabel}
             toggleFilter={toggleFilter}
-            transcripts={sortedTranscripts}
+            transcripts={filteredTranscripts}
           />
         )}
         <div className={styles.row}>
@@ -144,7 +149,7 @@ const DefaultTranscriptslist = (props: Props) => {
       </div>
       <div className={styles.content}>
         <StripedBackground {...props} />
-        {sortedTranscripts.map((transcript, index) => {
+        {filteredTranscripts.map((transcript, index) => {
           const expandTranscript = expandedTranscriptIds.includes(
             transcript.stable_id
           );

src/ensembl/src/content/app/entity-viewer/shared/helpers/tests/transcripts-filter.test.ts

+import { filterTranscriptsBySOTerm } from '../transcripts-filter';
+
+import { createTranscript } from 'tests/fixtures/entity-viewer/transcript';
+
+/* Creating filters with different filter set to true/false */
+const proteinCodingFilters = {
+  protein_coding: true,
+  retained_intron: false,
+  processed_transcript: false,
+  nonsense_mediated_decay: false
+};
+
+/* Creating dummy transcripts with different different so_term  to test so_term filtering */
+/* note that the so_term can be any string matching the filters object above */
+const createProteinCodingTranscript = () => {
+  const transcript = createTranscript();
+  transcript.so_term = 'protein_coding';
+  return transcript;
+};
+
+const createProcessedTranscript = () => {
+  const transcript = createTranscript();
+  transcript.so_term = 'processed_transcript';
+  return transcript;
+};
+
+const createRetainedIntronTranscript = () => {
+  const transcript = createTranscript();
+  transcript.so_term = 'retained_intron';
+  return transcript;
+};
+
+const createNonsenseMediatedDecayTranscript = () => {
+  const transcript = createTranscript();
+  transcript.so_term = 'nonsense_mediated_decay';
+  return transcript;
+};
+
+const ProteinCodingTranscript = createProteinCodingTranscript();
+const ProcessedTranscript = createProcessedTranscript();
+const RetainedIntronTranscript = createRetainedIntronTranscript();
+const NonsenseMediatedDecayTranscript = createNonsenseMediatedDecayTranscript();
+
+describe('filterTranscriptsBySOTerm', () => {
+  it('filters transcripts by so_term correctly', () => {
+    const Transcripts = [
+      ProteinCodingTranscript,
+      ProcessedTranscript,
+      RetainedIntronTranscript,
+      NonsenseMediatedDecayTranscript
+    ];
+
+    const expectedTranscripts = [ProteinCodingTranscript];
+
+    const filteredTranscripts = filterTranscriptsBySOTerm(
+      Transcripts,
+      proteinCodingFilters
+    );
+
+    expect(filteredTranscripts).toEqual(expectedTranscripts);
+  });
+});

src/ensembl/src/content/app/entity-viewer/shared/helpers/transcripts-filter.ts

+import { Filters } from 'src/content/app/entity-viewer/state/gene-view/transcripts/geneViewTranscriptsSlice';
+
+import { FullTranscript } from 'src/shared/types/thoas/transcript';
+
+type TranscriptSOTerm = Pick<FullTranscript, 'so_term'>;
+
+export function filterTranscriptsBySOTerm(
+  transcripts: TranscriptSOTerm[],
+  filters: Filters
+) {
+  const soTerms = Object.keys(filters).filter((key) => filters[key]);
+  const filteredTranscripts = transcripts.filter((transcript) => {
+    return soTerms.includes(transcript.so_term);
+  });
+
+  return filteredTranscripts;
+}