Fix verbatim partition pruner

src/Storages/MergeTree/KeyCondition.cpp

@@ -714,6 +714,13 @@ bool KeyCondition::canConstantBeWrappedByFunctions(
                 break;
             }
         }
+
+        // Check if we have enough columns to fulfill the action.
+        for (const auto & name : action.getNeededColumns())
+        {
+            if (!transform.has(name))
+                return false;
+        }
         action.execute(transform, true);
     }
 

tests/queries/0_stateless/01540_verbatim_partition_pruning.reference

@@ -1,3 +1,4 @@
 2	3
 9	5
 8	4
+1	2	3

tests/queries/0_stateless/01540_verbatim_partition_pruning.sql

@@ -21,3 +21,10 @@ select * from xy where intHash64(x) % 2 = intHash64(2) % 2;
 select * from xy where x = 8;
 
 drop table if exists xy;
+
+-- Test if we provide enough columns to generate a partition value
+drop table if exists xyz;
+create table xyz(x int, y int, z int) engine MergeTree partition by if(toUInt8(x), y, z) order by x settings index_granularity = 1;
+insert into xyz values (1, 2, 3);
+select * from xyz where y = 2;
+drop table if exists xyz;