more tests

Author: m4tx

cot-cli/src/handlers.rs

@@ -145,6 +145,19 @@ mod tests {
         assert!(result.is_err());
     }
 
+    #[test]
+    fn migration_new_wrong_directory() {
+        let args = MigrationNewArgs {
+            name: "test_migration".to_string(),
+            path: Some(PathBuf::from("nonexistent")),
+            app_name: None,
+        };
+
+        let result = handle_migration_new(args);
+
+        assert!(result.is_err());
+    }
+
     #[test]
     fn generate_manpages() {
         let temp_dir = tempfile::tempdir().unwrap();

cot-cli/tests/snapshot_testing/migration/mod.rs

@@ -114,3 +114,23 @@ fn migration_make_existing_model() {
         );
     }
 }
+
+#[test]
+#[expect(clippy::cast_possible_truncation)]
+fn migration_new() {
+    let cmd = cot_cli!("migration", "new", "custom");
+    for (idx, mut cli) in cot_clis_with_verbosity(&cmd).into_iter().enumerate() {
+        let filter = Verbosity::<OffLevel>::new(idx as u8, 0).filter();
+
+        let temp_dir = tempfile::TempDir::with_prefix("cot-test-").unwrap();
+        test_utils::make_package(temp_dir.path()).unwrap();
+
+        insta::with_settings!(
+            {
+                description => format!("Verbosity level: {filter}"),
+                filters => [GENERIC_FILTERS, TEMP_PATH_FILTERS, TEMP_PROJECT_FILTERS].concat()
+            },
+            { assert_cmd_snapshot!(cli.current_dir(temp_dir.path())) }
+        );
+    }
+}

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new-2.snap

@@ -0,0 +1,20 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: error"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-v"
+---
+success: true
+exit_code: 0
+----- stdout -----
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new-3.snap

@@ -0,0 +1,20 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: warn"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-vv"
+---
+success: true
+exit_code: 0
+----- stdout -----
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new-4.snap

@@ -0,0 +1,21 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: info"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-vvv"
+---
+success: true
+exit_code: 0
+----- stdout -----
+TIMESTAMP DEBUG cot_cli::migration_generator: Parsing file: "/tmp/TEMP_PATH/src/main.rs"
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new-5.snap

@@ -0,0 +1,22 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: debug"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-vvvv"
+---
+success: true
+exit_code: 0
+----- stdout -----
+TIMESTAMP DEBUG cot_cli::migration_generator: Parsing file: "/tmp/TEMP_PATH/src/main.rs"
+TIMESTAMP TRACE cot_cli::migration_generator: Processing file: "main.rs"
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new-6.snap

@@ -0,0 +1,22 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: trace"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-vvvvv"
+---
+success: true
+exit_code: 0
+----- stdout -----
+TIMESTAMP DEBUG cot_cli::migration_generator: Parsing file: "/tmp/TEMP_PATH/src/main.rs"
+TIMESTAMP TRACE cot_cli::migration_generator: Processing file: "main.rs"
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot-cli/tests/snapshot_testing/migration/snapshots/cli__snapshot_testing__migration__migration_new.snap

@@ -0,0 +1,20 @@
+---
+source: cot-cli/tests/snapshot_testing/migration/mod.rs
+description: "Verbosity level: off"
+info:
+  program: cot
+  args:
+    - migration
+    - new
+    - custom
+    - "-q"
+---
+success: true
+exit_code: 0
+----- stdout -----
+
+----- stderr -----
+    Creating Migration 'm_0001_custom'
+    Creating Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration file '/tmp/TEMP_PATH/src/migrations/m_0001_custom.rs'
+     Created Migration 'm_0001_custom'

cot/src/db/migrations.rs

@@ -2205,6 +2205,26 @@ mod tests {
         assert!(result.is_err());
     }
 
+    #[cot::test]
+    #[cfg_attr(
+        miri,
+        ignore = "unsupported operation: can't call foreign function `sqlite3_open_v2`"
+    )]
+    async fn operation_custom_backwards_not_implemented() {
+        // test only on SQLite because we are using raw SQL
+        let test_db = TestDatabase::new_sqlite().await.unwrap();
+
+        #[migration_op]
+        async fn forwards(_ctx: MigrationContext<'_>) -> Result<()> {
+            Ok(())
+        }
+
+        let operation = Operation::custom(forwards).build();
+        let result = operation.backwards(&test_db.database()).await;
+
+        assert!(result.is_err());
+    }
+
     #[test]
     fn field_new() {
         let field = Field::new(Identifier::new("id"), ColumnType::Integer)