AliasedFieldsTest.java

/*
 * Copyright 2011 JBoss Inc
 *
 * 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.drools.pmml.pmml_4_1.transformations;


import org.drools.definition.type.FactType;
import org.drools.pmml.pmml_4_1.DroolsAbstractPMMLTest;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

public class AliasedFieldsTest extends DroolsAbstractPMMLTest {

    private static final boolean VERBOSE = true;
    private static final String source = "org/drools/pmml/pmml_4_1/test_derived_fields_alias.xml";
    private static final String packageName = "org.drools.pmml.pmml_4_1.test";



    @Before
    public void setUp() throws Exception {
        setKSession(getModelSession(source, VERBOSE));
        setKbase(getKSession().getKnowledgeBase());
    }

    @After
    public void tearDown() {
        getKSession().dispose();
    }

    @Test
    public void testKonst() throws Exception {
        FactType konst = getKbase().getFactType(packageName, "Konst");
        assertNotNull(konst);

        getKSession().fireAllRules();

        System.out.println(reportWMObjects(getKSession()));

        assertEquals(1, getKSession().getObjects().size());

        checkFirstDataFieldOfTypeStatus(konst,true,false, null,8);


    }


    @Test
    public void testAlias() throws Exception {
        FactType alias = getKbase().getFactType( packageName, "AliasAge" );
        FactType aliasmm = getKbase().getFactType( packageName, "AliasAgeMM" );
        assertNotNull( alias );
        assertNotNull( aliasmm );

        getKSession().getWorkingMemoryEntryPoint( "in_Age" ).insert( 33 );
        getKSession().fireAllRules();

        checkFirstDataFieldOfTypeStatus( alias, true, false, null, 33 );

        refreshKSession();

        getKSession().getWorkingMemoryEntryPoint( "in_Age" ).insert( -1 );
        getKSession().fireAllRules();

        checkFirstDataFieldOfTypeStatus( alias, true, true, null, -1 );
        checkFirstDataFieldOfTypeStatus( aliasmm, true, false, null, 99 );

    }





}