Merge pull request #52 from NikkeTryHard/feat/0.3.0-db-integration

feat: 0.3.0 Database Integration (#22, #40, #37)

Author: NikkeTryHard

src/core/config.rs

@@ -233,6 +233,24 @@ pub struct Cli {
     #[arg(long)]
     pub force_toxic: bool,
 
+    // =========================================================================
+    // Django Database Options (pytest-django compatible)
+    // =========================================================================
+    /// Reuse existing test database between runs.
+    ///
+    /// Skips database creation and migrations if the test database exists.
+    /// Speeds up repeated test runs. Use --create-db to force recreation
+    /// after schema changes.
+    #[arg(long)]
+    pub reuse_db: bool,
+
+    /// Force recreation of test database.
+    ///
+    /// Drops and recreates the test database even if --reuse-db is set.
+    /// Use this after schema changes or when the database is corrupted.
+    #[arg(long)]
+    pub create_db: bool,
+
     /// Ignore .gitignore and .ignore files during test discovery.
     ///
     /// Bypasses standard ignore files (like .gitignore, .ignore) when
@@ -647,6 +665,10 @@ pub struct MergedConfig {
     pub disabled_plugins: Vec<String>,
     /// Network isolation configuration
     pub network: Option<NetworkConfig>,
+    /// Reuse existing test database (Django)
+    pub reuse_db: bool,
+    /// Force recreation of test database (Django)
+    pub create_db: bool,
 }
 
 impl MergedConfig {
@@ -686,6 +708,8 @@ impl MergedConfig {
             coverage_format: cov_config.format.unwrap_or_else(|| "lcov".to_string()),
             disabled_plugins,
             network: file_config.network.clone(),
+            reuse_db: cli.reuse_db,
+            create_db: cli.create_db,
         }
     }
 }
@@ -1263,4 +1287,42 @@ LITERAL_BRACES = "{{NOT_EXPANDED}}"
             std::env::remove_var("LITERAL_BRACES");
         }
     }
+
+    // =========================================================================
+    //  Django Database CLI Flag Tests
+    // =========================================================================
+
+    #[test]
+    fn test_reuse_db_flag_parsing() {
+        use clap::Parser;
+        let cli = Cli::parse_from(["tach", "--reuse-db", "."]);
+        assert!(cli.reuse_db);
+        assert!(!cli.create_db);
+    }
+
+    #[test]
+    fn test_create_db_flag_parsing() {
+        use clap::Parser;
+        let cli = Cli::parse_from(["tach", "--create-db", "."]);
+        assert!(cli.create_db);
+        assert!(!cli.reuse_db);
+    }
+
+    #[test]
+    fn test_create_db_overrides_reuse_db() {
+        use clap::Parser;
+        let cli = Cli::parse_from(["tach", "--reuse-db", "--create-db", "."]);
+        assert!(cli.create_db);
+        assert!(cli.reuse_db);
+    }
+
+    #[test]
+    fn test_merged_config_includes_db_flags() {
+        use clap::Parser;
+        let cli = Cli::parse_from(["tach", "--reuse-db", "."]);
+        let file_config = TachConfig::default();
+        let merged = MergedConfig::from_cli_and_file(&cli, &file_config);
+        assert!(merged.reuse_db);
+        assert!(!merged.create_db);
+    }
 }

src/core/protocol.rs

@@ -59,6 +59,14 @@ pub struct TestPayload {
     /// Contains parsed marker arguments like @pytest.mark.django_db(transaction=True)
     #[serde(default)]
     pub marker_info: Vec<MarkerInfo>,
+
+    /// Whether to reuse existing test database (--reuse-db)
+    #[serde(default)]
+    pub reuse_db: bool,
+
+    /// Whether to force database recreation (--create-db)
+    #[serde(default)]
+    pub create_db: bool,
 }
 
 /// Fixture info for payload
@@ -507,6 +515,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&payload).unwrap();
@@ -631,6 +641,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&payload).unwrap();
@@ -660,6 +672,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
         let encoded = encode_with_length(&payload).unwrap();
 
@@ -742,6 +756,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&payload).unwrap();
@@ -793,6 +809,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec!["django_db".to_string(), "slow".to_string()],
             marker_info: marker_info.clone(),
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&payload).unwrap();
@@ -931,4 +949,60 @@ mod tests {
             "Should reject invalid magic before checking length"
         );
     }
+
+    // =========================================================================
+    // Database Lifecycle Tests
+    // =========================================================================
+
+    #[test]
+    fn test_payload_with_db_lifecycle() {
+        let payload = TestPayload {
+            test_id: 1,
+            file_path: "test.py".to_string(),
+            test_name: "test_db".to_string(),
+            is_async: false,
+            fixtures: vec![],
+            log_fd: -1,
+            debug_socket_path: String::new(),
+            is_toxic: false,
+            timeout_secs: None,
+            hooks: vec![],
+            cached_effects: vec![],
+            markers: vec![],
+            marker_info: vec![],
+            reuse_db: true,
+            create_db: false,
+        };
+
+        let encoded = encode_with_length(&payload).unwrap();
+        let decoded: TestPayload = decode_with_limit(&encoded, MAX_PAYLOAD_SIZE).unwrap();
+        assert!(decoded.reuse_db);
+        assert!(!decoded.create_db);
+    }
+
+    #[test]
+    fn test_payload_db_flags_default_false() {
+        let payload = TestPayload {
+            test_id: 1,
+            file_path: "test.py".to_string(),
+            test_name: "test".to_string(),
+            is_async: false,
+            fixtures: vec![],
+            log_fd: -1,
+            debug_socket_path: String::new(),
+            is_toxic: false,
+            timeout_secs: None,
+            hooks: vec![],
+            cached_effects: vec![],
+            markers: vec![],
+            marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
+        };
+
+        let encoded = encode_with_length(&payload).unwrap();
+        let decoded: TestPayload = decode_with_limit(&encoded, MAX_PAYLOAD_SIZE).unwrap();
+        assert!(!decoded.reuse_db);
+        assert!(!decoded.create_db);
+    }
 }

src/execution/scheduler.rs

@@ -445,6 +445,8 @@ impl Scheduler {
             cached_effects,
             markers: test.markers.clone(),
             marker_info: test.marker_info.clone(),
+            reuse_db: false,
+            create_db: false,
         };
 
         // Use encode_with_length which includes protocol header

src/execution/zygote.rs

@@ -2286,6 +2286,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&original).expect("Serialization should succeed");
@@ -2333,6 +2335,8 @@ mod tests {
             cached_effects: vec![],
             markers: vec![],
             marker_info: vec![],
+            reuse_db: false,
+            create_db: false,
         };
 
         let encoded = encode_with_length(&payload).unwrap();