| 1 | #include "catch.hpp" |
| 2 | #include "test_helpers.hpp" |
| 3 | |
| 4 | using namespace duckdb; |
| 5 | using namespace std; |
| 6 | |
| 7 | TEST_CASE("NOT NULL constraint" , "[constraints]" ) { |
| 8 | unique_ptr<QueryResult> result; |
| 9 | DuckDB db(nullptr); |
| 10 | Connection con(db); |
| 11 | |
| 12 | REQUIRE_NO_FAIL(con.Query("CREATE TABLE integers(i INTEGER NOT NULL)" )); |
| 13 | // insert normal value |
| 14 | REQUIRE_NO_FAIL(con.Query("INSERT INTO integers VALUES (3)" )); |
| 15 | // insert NULL |
| 16 | REQUIRE_FAIL(con.Query("INSERT INTO integers VALUES (NULL)" )); |
| 17 | |
| 18 | // update normal value |
| 19 | REQUIRE_NO_FAIL(con.Query("UPDATE integers SET i=4" )); |
| 20 | // update NULL |
| 21 | REQUIRE_FAIL(con.Query("UPDATE integers SET i=NULL" )); |
| 22 | |
| 23 | REQUIRE_NO_FAIL(con.Query("CREATE TABLE integers_with_null(i INTEGER)" )); |
| 24 | REQUIRE_NO_FAIL(con.Query("INSERT INTO integers_with_null VALUES (3), (4), (5), (NULL);" )); |
| 25 | |
| 26 | //! INSERT from SELECT with NULL |
| 27 | REQUIRE_FAIL(con.Query("INSERT INTO integers (i) SELECT * FROM integers_with_null" )); |
| 28 | //! INSERT from SELECT without NULL |
| 29 | REQUIRE_NO_FAIL(con.Query("INSERT INTO integers (i) SELECT * FROM " |
| 30 | "integers_with_null WHERE i IS NOT NULL" )); |
| 31 | |
| 32 | result = con.Query("SELECT * FROM integers ORDER BY i" ); |
| 33 | REQUIRE(CHECK_COLUMN(result, 0, {3, 4, 4, 5})); |
| 34 | |
| 35 | REQUIRE_NO_FAIL(con.Query("UPDATE integers SET i=4 WHERE i>4" )); |
| 36 | |
| 37 | result = con.Query("SELECT * FROM integers ORDER BY i" ); |
| 38 | REQUIRE(CHECK_COLUMN(result, 0, {3, 4, 4, 4})); |
| 39 | } |
| 40 | |