/*
* Copyright 2013 Red Hat, Inc. and/or its affiliates.
*
* 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.kie.spring.tests;
import org.drools.compiler.kie.builder.impl.InternalKieModule;
import org.junit.Test;
import org.kie.api.KieBase;
import org.kie.api.KieServices;
import org.kie.api.builder.KieScanner;
import org.kie.api.builder.ReleaseId;
import org.kie.api.runtime.KieSession;
import org.kie.scanner.MavenRepository;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
public class KieSpringScannerTest extends AbstractKieSpringDynamicModuleTest {
private static ApplicationContext context = null;
private final int FIRST_VALUE = 5;
private final int SECOND_VALUE = 10;
@Test
public void testSpringKieScanner() throws Exception {
KieServices ks = KieServices.Factory.get();
//step 1: deploy the test module to MAVEN Repo
MavenRepository repository = createAndInstallModule( ks, FIRST_VALUE );
//step 2: load the spring context
createSpringContext();
//step 3: check the basic spring objects
lookupNamedKieBase();
lookupReleaseId();
lookupReleaseIdScanner();
//step 4: test the value
checkForValue(FIRST_VALUE);
//step 5: reploy the module
reinstallModule( repository, ks );
//step 6: force the kie-scanner to scan
KieScanner releaseIdScanner = context.getBean("spring-scanner-releaseId#scanner", KieScanner.class);
releaseIdScanner.scanNow();
//step 7: retest the value to ensure the scanner has picked up the new value
checkForValue(SECOND_VALUE);
//step 8: cleanup. Remove the module
ks.getRepository().removeKieModule(releaseId);
}
protected void createSpringContext() throws Exception {
context = new ClassPathXmlApplicationContext("org/kie/spring/kie-scanner.xml");
assertNotNull(context);
}
protected void lookupNamedKieBase() throws Exception {
KieBase kieBase = context.getBean("KBase1", KieBase.class);
assertNotNull(kieBase);
}
protected void lookupReleaseId() throws Exception {
ReleaseId releaseId = context.getBean("spring-scanner-releaseId", ReleaseId.class);
assertNotNull(releaseId);
}
protected void lookupReleaseIdScanner() throws Exception {
KieScanner releaseIdScanner = context.getBean("spring-scanner-releaseId#scanner", KieScanner.class);
assertNotNull(releaseIdScanner);
}
protected void reinstallModule( MavenRepository repository, KieServices ks ) throws IOException {
InternalKieModule kJar2 = createKieJarWithClass(ks, releaseId, SECOND_VALUE);
File kPom = createKPom( releaseId );
repository.installArtifact(releaseId, kJar2, kPom);
}
protected void checkForValue(int value) {
List<Integer> list = new ArrayList<Integer>();
KieBase kieBase = context.getBean("KBase1", KieBase.class);
KieSession ksession = kieBase.newKieSession();
ksession.setGlobal( "list", list );
ksession.fireAllRules();
ksession.dispose();
assertTrue("Expected:<" + value + "> but was:<" + list.get(0) + ">", list.get(0) == value);
}
}