-- Licensed to the Apache Software Foundation (ASF) under one
-- or more contributor license agreements.  See the NOTICE file
-- distributed with this work for additional information
-- regarding copyright ownership.  The ASF licenses this file
-- to you under the Apache License, Version 2.0 (the
-- "License"); you may not use this file except in compliance
-- with the License.  You may obtain a copy of the License at
--
--   http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing,
-- software distributed under the License is distributed on an
-- "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-- KIND, either express or implied.  See the License for the
-- specific language governing permissions and limitations
-- under the License.

-- Pin the session timezone so the test exercises the non-UTC path regardless of the JVM
-- default. Enable the codegen dispatcher so non-UTC and non-whitelisted formats stay inside
-- Comet via Spark's own DateFormatClass.doGenCode instead of falling back to Spark.
-- Config: spark.sql.session.timeZone=America/Los_Angeles
-- Config: spark.comet.exec.scalaUDF.codegen.enabled=true

statement
CREATE TABLE test_date_format(ts timestamp) USING parquet

statement
INSERT INTO test_date_format VALUES (timestamp('2024-06-15 10:30:45')), (timestamp('1970-01-01 00:00:00')), (NULL)

query
SELECT date_format(ts, 'yyyy-MM-dd') FROM test_date_format

query
SELECT date_format(ts, 'HH:mm:ss') FROM test_date_format

query
SELECT date_format(ts, 'yyyy-MM-dd HH:mm:ss') FROM test_date_format

-- literal arguments
query
SELECT date_format(timestamp('2024-06-15 10:30:45'), 'yyyy-MM-dd')
