Add proper tests for querying enums

Signed-off-by: Jim Crossley <[email protected]>

common/src/db/query.rs

@@ -244,7 +244,7 @@ pub(crate) mod tests {
         impl ActiveModelBehavior for ActiveModel {}
 
         #[derive(Debug, Clone, PartialEq, Eq, EnumIter, DeriveActiveEnum)]
-        #[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "severity")]
+        #[sea_orm(rs_type = "String", db_type = "Enum")]
         pub enum Severity {
             #[sea_orm(string_value = "low")]
             Low,

common/src/db/query/columns.rs

@@ -27,13 +27,11 @@ impl Display for Columns {
                 ColumnType::Text | ColumnType::String(_) | ColumnType::Char(_) => {
                     write!(f, "String")?
                 }
-                ColumnType::Enum {
-                    name: _,
-                    variants: v,
-                } => write!(
+                ColumnType::Enum { name, variants } => write!(
                     f,
-                    "Enum {:?}",
-                    v.iter().map(|v| v.to_string()).collect::<Vec<_>>()
+                    "Enum({}) {:?}",
+                    name.to_string(),
+                    variants.iter().map(|v| v.to_string()).collect::<Vec<_>>()
                 )?,
                 t => write!(f, "  {t:?}")?,
             }

common/src/db/query/filter.rs

@@ -412,6 +412,14 @@ pub(crate) mod tests {
             where_clause("published!=NULL")?,
             r#""advisory"."published" IS NOT NULL"#
         );
+        assert_eq!(
+            where_clause("severity=high")?,
+            r#""advisory"."severity" = (CAST('high' AS "Severity"))"#
+        );
+        assert_eq!(
+            where_clause("severity>low")?,
+            r#""advisory"."severity" > (CAST('low' AS "Severity"))"#
+        );
 
         Ok(())
     }