add ContextType field to context table/model

codecov · Jan 12, 2024 · 0bb4105 · 0bb4105
1 parent d2a9ba1
commit 0bb4105
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/migrations/2024-01-09-033635_init/up.sql b/migrations/2024-01-09-033635_init/up.sql
@@ -27,5 +27,6 @@ CREATE TABLE context_assoc (
 
 CREATE TABLE context (
     id INTEGER PRIMARY KEY,
+    context_type VARCHAR NOT NULL,
     name VARCHAR NOT NULL
 );
diff --git a/src/report.rs b/src/report.rs
@@ -62,6 +62,7 @@ mod tests {
 
         let mock_context = models::Context {
             id: 0,
+            context_type: schema::ContextType::TestCase,
             name: "mock_context".to_string(),
         };
         diesel::insert_into(schema::context::table)

diff --git a/src/report/models.rs b/src/report/models.rs
@@ -46,5 +46,6 @@ pub struct ContextAssoc {
 #[diesel(table_name = context)]
 pub struct Context {
     pub id: i32,
+    pub context_type: ContextType,
     pub name: String,
 }
diff --git a/src/report/schema.rs b/src/report/schema.rs
@@ -7,6 +7,12 @@ pub enum CoverageStatus {
     Partial,
 }
 
+#[derive(diesel_derive_enum::DbEnum, Debug, PartialEq)]
+pub enum ContextType {
+    TestCase,
+    Upload,
+}
+
 diesel::table! {
     use diesel::sql_types::{VarChar, Integer};
     source_file (id) {
@@ -49,8 +55,10 @@ diesel::table! {
 
 diesel::table! {
     use diesel::sql_types::{VarChar, Integer};
+    use super::ContextTypeMapping;
     context (id) {
         id -> Integer,
+        context_type -> ContextTypeMapping,
         name -> VarChar,
     }
 }