Refactor getter 'getAnnotationComponent' (#345)

neurobagel · Feb 6, 2023 · a5e35f0 · a5e35f0
1 parent f0bb666
commit a5e35f0
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 3 deletions.
diff --git a/cypress/unit/store-getter-getAnnotationComponent.cy.js b/cypress/unit/store-getter-getAnnotationComponent.cy.js
@@ -0,0 +1,59 @@
+import { getters } from "~/store";
+
+describe("The getAnnotationComponent getter", () => {
+
+    let store = {};
+
+    beforeEach(() => {
+
+        store = {
+
+            state: {
+
+                categories: {
+
+                    "Subject ID": {},
+                    "Age": {
+
+                        componentName: "annot-continuous-values"
+                    },
+                    "Sex": {
+
+                        componentName: "annot-categorical"
+                    },
+                    "Diagnosis": {
+
+                        componentName: "annot-categorical"
+                    }
+                }
+            }
+        };
+    });
+
+    it("Get the component name for the 'Age' category", () => {
+
+        // Act
+        const componentName = getters.getAnnotationComponent(store.state)("Age");
+
+        // Assert
+        expect(componentName).to.equal("annot-continuous-values");
+    });
+
+    it("Get the component name for the 'Sex' category", () => {
+
+        // Act
+        const componentName = getters.getAnnotationComponent(store.state)("Sex");
+
+        // Assert
+        expect(componentName).to.equal("annot-categorical");
+    });
+
+    it("Get the component name for the 'Diagnosis' category", () => {
+
+        // Act
+        const componentName = getters.getAnnotationComponent(store.state)("Diagnosis");
+
+        // Assert
+        expect(componentName).to.equal("annot-categorical");
+    });
+});
diff --git a/store/index.js b/store/index.js
@@ -6,9 +6,18 @@ export const state = () => ({
     categories: {
 
         "Subject ID": {},
-        "Age": {},
-        "Sex": {},
-        "Diagnosis": {}
+        "Age": {
+
+            componentName: "annot-continuous-values"
+        },
+        "Sex": {
+
+            componentName: "annot-categorical"
+        },
+        "Diagnosis": {
+
+            componentName: "annot-categorical"
+        }
     },
 
     colorInfo: {
@@ -91,6 +100,11 @@ export const state = () => ({
 
 export const getters = {
 
+    getAnnotationComponent: (p_state) => (p_category) => {
+
+        return p_state.categories[p_category].componentName;
+    },
+
     getCategoryNames (p_state) {