/*
 * Copyright 2004-2014 Brian McCallister
 * Copyright 2010-2014 Ning, Inc.
 * Copyright 2014-2020 Groupon, Inc
 * Copyright 2020-2020 Equinix, Inc
 * Copyright 2014-2020 The Billing Project, LLC
 *
 * The Billing Project 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.
 */

package org.skife.jdbi.v2.st4;

import java.sql.ResultSet;
import java.sql.SQLException;

import org.junit.Rule;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.skife.jdbi.v2.DBI;
import org.skife.jdbi.v2.Handle;
import org.skife.jdbi.v2.JDBITests;
import org.skife.jdbi.v2.StatementContext;
import org.skife.jdbi.v2.sqlobject.stringtemplate.ST4StatementLocator;
import org.skife.jdbi.v2.tweak.HandleCallback;
import org.skife.jdbi.v2.tweak.ResultSetMapper;
import org.stringtemplate.v4.STGroupFile;

import static org.assertj.core.api.Assertions.assertThat;

public class FluentTest {

    @Rule
    public H2Rule h2 = new H2Rule();

    @Test
    @Category(JDBITests.class)
    public void testFoo() throws Exception {
        final DBI dbi = new DBI(this.h2.getDataSource());
        dbi.setStatementLocator(new ST4StatementLocator(new STGroupFile("org/skife/jdbi/v2/st4/DaoTest.InnerDao.sql.stg")));
        dbi.withHandle(new HandleCallback<Object>() {
            @Override
            public Object withHandle(final Handle h) throws Exception {
                h.execute("createSomething");
                h.createStatement("insert")
                 .define("table", "something")
                 .bind("id", 1)
                 .bind("name", "Ven")
                 .execute();

                final Something s = h.createQuery("findById")
                                     .bind("id", 1)
                                     .map(new ResultSetMapper<Something>() {
                                         @Override
                                         public Something map(final int index, final ResultSet r, final StatementContext ctx) throws SQLException {
                                             return new Something(r.getInt("id"), r.getString("name"));
                                         }
                                     })
                                     .first();
                assertThat(s).isEqualTo(new Something(1, "Ven"));
                return null;
            }
        });
    }
}
