SIMSBIOHUB-665: Habitat Features Database Schema

api/src/database-models/alert.ts

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+import { AlertSeverityType } from '../database-units/alert_severity';
 
 /**
  * Alert Model.
@@ -10,7 +11,7 @@ export const AlertModel = z.object({
   alert_type_id: z.number(),
   name: z.string(),
   message: z.string(),
-  severity: z.enum(['info', 'success', 'error', 'warning']),
+  severity: AlertSeverityType,
   data: z.object({}).nullable(),
   record_end_date: z.string().nullable(),
   create_date: z.string(),

api/src/database-units/alert_severity.ts

@@ -0,0 +1,20 @@
+import { z } from 'zod';
+
+/**
+ * Alert Severity Data Type.
+ *
+ * @description Enum for `alert_severity` database type.
+ */
+export enum AlertSeverity {
+  INFO = 'info',
+  WARNING = 'warning',
+  ERROR = 'error',
+  SUCCESS = 'success'
+}
+
+/**
+ * Alert Severity Data Type.
+ *
+ * @description Type for `alert_severity` database type.
+ */
+export const AlertSeverityType = z.nativeEnum(AlertSeverity);

api/src/database-units/quantitative_unit.ts

@@ -0,0 +1,36 @@
+import { z } from 'zod';
+
+/**
+ * Quantitative Unit Data Type.
+ *
+ * @description Enum for `quantitative_unit` database type.
+ */
+export enum QuantitativeUnit {
+  MILLIMETER = 'millimeter',
+  CENTIMETER = 'centimeter',
+  METER = 'meter',
+  MILLIGRAM = 'milligram',
+  GRAM = 'gram',
+  KILOGRAM = 'kilogram',
+  PERCENT = 'percent',
+  CELSIUS = 'celsius',
+  PPT = 'ppt',
+  SCF = 'SCF',
+  DEGREES = 'degrees',
+  PH = 'pH',
+  SECONDS = 'seconds',
+  METERS_SQUARED = 'meters squared',
+  COUNT = 'count',
+  GHZ = 'GHz',
+  HZ = 'Hz',
+  AMPS = 'amps',
+  VOLTS = 'volts',
+  MEGAPIXELS = 'megapixels'
+}
+
+/**
+ * Quantitative Unit Data Type.
+ *
+ * @description Type for `quantitative_unit` database type.
+ */
+export const QuantitativeUnitType = z.nativeEnum(QuantitativeUnit);

api/src/models/alert-view.ts

@@ -1,9 +1,12 @@
 import { z } from 'zod';
 import { AlertRecord } from '../database-models/alert';
 
+export const AlertStatus = z.enum(['active', 'expired']);
+export type AlertStatus = z.infer<typeof AlertStatus>;
+
 export const AlertRecordWithStatus = AlertRecord.extend({
   create_date: z.string(),
-  status: z.enum(['active', 'expired'])
+  status: AlertStatus
 });
 export type AlertRecordWithStatus = z.infer<typeof AlertRecordWithStatus>;
 
@@ -17,7 +20,3 @@ export interface IAlertFilterObject {
   expiresAfter?: string;
   types?: string[];
 }
-
-// Define severity and status types
-export type IAlertSeverity = 'info' | 'success' | 'error' | 'warning';
-export type IAlertStatus = 'active' | 'expired';

api/src/openapi/schemas/alert.ts

@@ -1,4 +1,5 @@
 import { OpenAPIV3 } from 'openapi-types';
+import { AlertSeverity } from '../../database-units/alert_severity';
 
 /**
  * Base schema for system alerts
@@ -23,7 +24,7 @@ const baseSystemAlertSchema: OpenAPIV3.SchemaObject = {
     severity: {
       description: 'Severity level of the alert',
       type: 'string',
-      enum: ['info', 'success', 'warning', 'error']
+      enum: Object.values(AlertSeverity)
     },
     data: {
       description: 'Data associated with the alert',

api/src/paths/alert/index.test.ts

@@ -4,9 +4,10 @@ import sinon from 'sinon';
 import sinonChai from 'sinon-chai';
 import { SYSTEM_IDENTITY_SOURCE } from '../../constants/database';
 import { SYSTEM_ROLE } from '../../constants/roles';
+import { AlertSeverity } from '../../database-units/alert_severity';
 import * as db from '../../database/db';
 import { HTTPError } from '../../errors/http-error';
-import { IAlertSeverity, IAlertStatus } from '../../models/alert-view';
+import { AlertRecordWithStatus } from '../../models/alert-view';
 import { AlertService } from '../../services/alert-service';
 import { getMockDBConnection, getRequestHandlerMocks } from '../../__mocks__/db';
 import { createAlert, getAlerts } from '../alert';
@@ -21,14 +22,14 @@ describe('getAlerts', () => {
   describe('as a system user', () => {
     it('returns a list of system alerts', async () => {
       const mockTotal = 10;
-      const mockAlerts = [
+      const mockAlerts: AlertRecordWithStatus[] = [
         {
           alert_id: 1,
           name: 'Alert 1',
           message: 'Message 1',
           alert_type_id: 1,
-          severity: 'error' as IAlertSeverity,
-          status: 'active' as IAlertStatus,
+          severity: AlertSeverity.ERROR,
+          status: 'active',
           data: null,
           record_end_date: null,
           create_date: '2020-01-01T10:10:10'
@@ -38,8 +39,8 @@ describe('getAlerts', () => {
           name: 'Alert 2',
           message: 'Message 2',
           alert_type_id: 2,
-          severity: 'error' as IAlertSeverity,
-          status: 'active' as IAlertStatus,
+          severity: AlertSeverity.ERROR,
+          status: 'active',
           data: null,
           record_end_date: null,
           create_date: '2020-01-01T10:10:10'
@@ -126,7 +127,7 @@ describe('createAlert', () => {
       name: 'New Alert',
       message: 'New alert message',
       alert_type_id: 1,
-      severity: 'medium'
+      severity: AlertSeverity.INFO
     };
 
     const mockDBConnection = getMockDBConnection({ open: sinon.stub(), commit: sinon.stub() });

api/src/paths/alert/{alertId}/index.test.ts

@@ -5,9 +5,10 @@ import sinonChai from 'sinon-chai';
 import { deleteAlert, getAlertById, updateAlert } from '.';
 import { SYSTEM_IDENTITY_SOURCE } from '../../../constants/database';
 import { SYSTEM_ROLE } from '../../../constants/roles';
+import { AlertSeverity } from '../../../database-units/alert_severity';
 import * as db from '../../../database/db';
 import { HTTPError } from '../../../errors/http-error';
-import { IAlertSeverity, IAlertStatus } from '../../../models/alert-view';
+import { AlertRecordWithStatus } from '../../../models/alert-view';
 import { AlertService } from '../../../services/alert-service';
 import { getMockDBConnection, getRequestHandlerMocks } from '../../../__mocks__/db';
 
@@ -20,13 +21,13 @@ describe('getAlerts', () => {
 
   describe('as a system user', () => {
     it('returns a single system alert', async () => {
-      const mockAlert = {
+      const mockAlert: AlertRecordWithStatus = {
         alert_id: 1,
         name: 'Alert 1',
         message: 'Message 1',
         alert_type_id: 1,
-        severity: 'error' as IAlertSeverity,
-        status: 'active' as IAlertStatus,
+        severity: AlertSeverity.ERROR,
+        status: 'active',
         data: null,
         record_end_date: null,
         create_date: '2020-01-01T10:10:10'

api/src/repositories/alert-repository.test.ts

@@ -2,7 +2,8 @@ import chai, { expect } from 'chai';
 import { QueryResult } from 'pg';
 import sinon from 'sinon';
 import sinonChai from 'sinon-chai';
-import { IAlertSeverity } from '../models/alert-view';
+import { AlertSeverity } from '../database-units/alert_severity';
+import { AlertRecordWithStatus, IAlertCreateObject, IAlertUpdateObject } from '../models/alert-view';
 import { getMockDBConnection } from '../__mocks__/db';
 import { AlertRepository } from './alert-repository';
 
@@ -18,16 +19,17 @@ describe('AlertRepository', () => {
 
   describe('getAlerts', () => {
     it('should return an array of alerts with empty filters', async () => {
-      const mockRows = [
+      const mockRows: AlertRecordWithStatus[] = [
         {
           alert_id: 1,
           name: 'Alert 1',
           message: 'This is an alert.',
           alert_type_id: 1,
           data: {},
-          severity: 'error' as IAlertSeverity,
+          severity: AlertSeverity.ERROR,
+          status: 'active',
           record_end_date: null,
-          status: 'active'
+          create_date: '2020-01-01'
         }
       ];
       const mockQueryResponse = { rows: mockRows, rowCount: 1 } as unknown as QueryResult<any>;
@@ -52,7 +54,7 @@ describe('AlertRepository', () => {
           message: 'This is an alert.',
           alert_type_id: 1,
           data: {},
-          severity: 'error',
+          severity: AlertSeverity.ERROR,
           record_end_date: null,
           status: 'active'
         }
@@ -80,7 +82,7 @@ describe('AlertRepository', () => {
           message: 'This is an alert.',
           alert_type_id: 1,
           data: {},
-          severity: 'error',
+          severity: AlertSeverity.ERROR,
           record_end_date: null,
           status: 'active'
         }
@@ -109,13 +111,13 @@ describe('AlertRepository', () => {
       });
 
       const alertRepository = new AlertRepository(mockDBConnection);
-      const alert = {
+      const alert: IAlertUpdateObject = {
         alert_id: 1,
         name: 'Updated Alert',
         message: 'Updated message',
         alert_type_id: 1,
         data: {},
-        severity: 'error' as IAlertSeverity,
+        severity: AlertSeverity.ERROR,
         record_end_date: null
       };
 
@@ -135,12 +137,12 @@ describe('AlertRepository', () => {
       });
 
       const alertRepository = new AlertRepository(mockDBConnection);
-      const alert = {
+      const alert: IAlertCreateObject = {
         name: 'New Alert',
         message: 'New alert message',
         alert_type_id: 1,
         data: {},
-        severity: 'error' as IAlertSeverity,
+        severity: AlertSeverity.ERROR,
         record_end_date: null
       };
 

api/src/repositories/observation-subcount-environment-repository.ts

@@ -1,26 +1,9 @@
 import SQL from 'sql-template-strings';
 import { z } from 'zod';
+import { QuantitativeUnitType } from '../database-units/quantitative_unit';
 import { getKnex } from '../database/db';
 import { BaseRepository } from './base-repository';
 
-// Environment unit type definition.
-export const EnvironmentUnit = z.enum([
-  // Should be kept in sync with the database table `environment_unit`
-  'millimeter',
-  'centimeter',
-  'meter',
-  'milligram',
-  'gram',
-  'kilogram',
-  'percent',
-  'celsius',
-  'ppt',
-  'SCF',
-  'degrees',
-  'pH'
-]);
-export type EnvironmentUnit = z.infer<typeof EnvironmentUnit>;
-
 // Qualitative environment option type definition.
 const QualitativeEnvironmentOption = z.object({
   environment_qualitative_option_id: z.string().uuid(),
@@ -46,7 +29,7 @@ const QuantitativeEnvironmentTypeDefinition = z.object({
   description: z.string().nullable(),
   min: z.number().nullable(),
   max: z.number().nullable(),
-  unit: EnvironmentUnit.nullable()
+  unit: QuantitativeUnitType.nullable()
 });
 export type QuantitativeEnvironmentTypeDefinition = z.infer<typeof QuantitativeEnvironmentTypeDefinition>;
 

api/src/services/alert-service.test.ts

@@ -2,7 +2,8 @@ import chai, { expect } from 'chai';
 import { afterEach, describe, it } from 'mocha';
 import sinon from 'sinon';
 import sinonChai from 'sinon-chai';
-import { AlertRecordWithStatus, IAlertCreateObject, IAlertFilterObject, IAlertSeverity } from '../models/alert-view';
+import { AlertSeverity } from '../database-units/alert_severity';
+import { AlertRecordWithStatus, IAlertCreateObject, IAlertFilterObject } from '../models/alert-view';
 import { AlertRepository } from '../repositories/alert-repository';
 import { getMockDBConnection } from '../__mocks__/db';
 import { AlertService } from './alert-service';
@@ -26,7 +27,7 @@ describe('AlertService', () => {
           message: 'Message 1',
           alert_type_id: 1,
           data: {},
-          severity: 'error' as IAlertSeverity,
+          severity: AlertSeverity.ERROR,
           status: 'active',
           record_end_date: null,
           create_date: '2020-01-01T10:10:10'
@@ -73,7 +74,7 @@ describe('AlertService', () => {
         message: 'Message 1',
         alert_type_id: 1,
         data: {},
-        severity: 'error' as IAlertSeverity,
+        severity: AlertSeverity.ERROR,
         status: 'active',
         record_end_date: null,
         create_date: '2020-01-01T10:10:10'
@@ -99,7 +100,7 @@ describe('AlertService', () => {
         message: 'New alert message',
         alert_type_id: 1,
         data: {},
-        severity: 'error' as IAlertSeverity,
+        severity: AlertSeverity.ERROR,
         record_end_date: null
       };
 
@@ -124,7 +125,7 @@ describe('AlertService', () => {
         message: 'Updated message',
         alert_type_id: 1,
         data: {},
-        severity: 'error' as IAlertSeverity,
+        severity: AlertSeverity.ERROR,
         status: 'active',
         record_end_date: null,
         create_date: '2020-01-01T10:10:10'

app/src/interfaces/useReferenceApi.interface.ts

@@ -1,7 +1,29 @@
 /**
- * A qualitative environment unit.
+ * A quantitative unit.
+ *
+ * Note: should be kept in sync with the `quantitative_unit` enum in the database.
  */
-export type EnvironmentUnit = 'millimeter' | 'centimeter' | 'meter' | 'milligram' | 'gram' | 'kilogram';
+export type QuantitativeUnit =
+  | 'millimeter'
+  | 'centimeter'
+  | 'meter'
+  | 'milligram'
+  | 'gram'
+  | 'kilogram'
+  | 'percent'
+  | 'celsius'
+  | 'ppt'
+  | 'SCF'
+  | 'degrees'
+  | 'pH'
+  | 'seconds'
+  | 'meters squared'
+  | 'count'
+  | 'GHz'
+  | 'Hz'
+  | 'amps'
+  | 'volts'
+  | 'megapixels';
 
 /**
  * A quantitative environment type definition.
@@ -12,7 +34,7 @@ export type EnvironmentQuantitativeTypeDefinition = {
   description: string | null;
   min: number | null;
   max: number | null;
-  unit: EnvironmentUnit | null;
+  unit: QuantitativeUnit | null;
 };
 
 /**