/*
 * Licensed 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.jdbi.v3.sqlobject;

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

import org.jdbi.v3.core.rule.H2DatabaseRule;
import org.jdbi.v3.sqlobject.config.RegisterBeanMapper;
import org.jdbi.v3.sqlobject.statement.MapTo;
import org.jdbi.v3.sqlobject.statement.SqlQuery;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

public class TestPolymorphicReturn {
    @Rule
    public H2DatabaseRule dbRule = new H2DatabaseRule().withPlugin(new SqlObjectPlugin());

    private SheepDao dao;

    @Before
    public void makeSheep() {
        dbRule.getSharedHandle().execute("insert into something(name, intValue) values('Fluffy', 5)");
        dao = dbRule.getSharedHandle().attach(SheepDao.class);
    }

    @Test
    public void testPolymorphicReturnSuperclass() throws Exception {
        Sheep normalSheep = dao.get(Sheep.class, "Fluffy");
        assertThat(normalSheep.getName()).isEqualTo("Fluffy");
    }

    @Test
    public void testPolymorphicReturnSubclass() throws Exception {
        FlyingSheep flyingSheep = dao.get(FlyingSheep.class, "Fluffy");
        assertThat(flyingSheep.getName()).isEqualTo("Fluffy");
        assertThat(flyingSheep.getNumWings()).isEqualTo(5);
    }

    @Test(expected=UnsupportedOperationException.class)
    public void testBadArg() throws Exception {
        dao.getBad("Fluffy is sad :(");
    }


    @RegisterBeanMapper(Sheep.class)
    public interface SheepDao {
        @RegisterBeanMapper(FlyingSheep.class)
        @SqlQuery("select name, intValue as numWings from something where name=:name")
        <T extends Sheep> T get(@MapTo Class<T> klass, String name);

        @SqlQuery("baaaaaa")
        Sheep getBad(@MapTo String baaaaa);
    }

    public static class Sheep {
        private String name;

        public String getName() {
            return name;
        }
        public void setName(String name) {
            this.name = name;
        }
    }

    public static class FlyingSheep extends Sheep {
        private int numWings;

        public int getNumWings() {
            return numWings;
        }
        public void setNumWings(int numWings) {
            this.numWings = numWings;
        }
    }
}