querydsl/querydsl-sql/src/test/java/com/mysema/query/sql/DefaultNamingStrategyTest.java

/*
 * Copyright (c) 2010 Mysema Ltd.
 * All rights reserved.
 *
 */
package com.mysema.query.sql;

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import com.mysema.codegen.model.Types;
import com.mysema.query.codegen.EntityType;

public class DefaultNamingStrategyTest {

    private NamingStrategy namingStrategy = new DefaultNamingStrategy();

    @Test
    public void testGetClassName() {
        assertEquals("QUserData", namingStrategy.getClassName("Q", "user_data"));
        assertEquals("QU", namingStrategy.getClassName("Q", "u"));
        assertEquals("QUs",namingStrategy.getClassName("Q", "us"));
        assertEquals("QU", namingStrategy.getClassName("Q", "u_"));
        assertEquals("QUs",namingStrategy.getClassName("Q", "us_"));
    }

    @Test
    public void testGetPropertyName() {
        EntityType entityModel = new EntityType("Q", Types.OBJECT);
        assertEquals("whileCol", namingStrategy.getPropertyName("while", "Q", entityModel));
        assertEquals("name", namingStrategy.getPropertyName("name", "Q", entityModel));
        assertEquals("userId", namingStrategy.getPropertyName("user_id", "Q", entityModel));
    }

}