| package org.codehaus.plexus.taskqueue; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF 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. |
| */ |
| |
| import junit.framework.TestCase; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.springframework.scheduling.TaskScheduler; |
| import org.springframework.scheduling.backportconcurrent.ThreadPoolTaskExecutor; |
| import org.springframework.test.context.ContextConfiguration; |
| import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
| |
| import javax.inject.Inject; |
| import javax.inject.Named; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.concurrent.BlockingQueue; |
| import java.util.concurrent.LinkedBlockingQueue; |
| |
| /** |
| * @author <a href="mailto:trygvis@inamo.no">Trygve Laugstøl</a> |
| * @version $Id$ |
| */ |
| @RunWith( SpringJUnit4ClassRunner.class ) |
| @ContextConfiguration( locations = { "classpath*:/META-INF/spring-context.xml", "classpath*:/spring-context.xml" } ) |
| public class TaskQueueTest |
| extends TestCase |
| { |
| @Inject |
| @Named( value = "taskQueue#taskQueueTest" ) |
| private TaskQueue taskQueue; |
| |
| |
| // NOTE: If we were using a blocking queue, the sleep/continue in the ThreadedTaskQueueExecutor wouldn't |
| // be necessary; the queue would block until an element was available. |
| @Test |
| public void testEmptyQueue() |
| throws Exception |
| { |
| assertNull( taskQueue.take() ); |
| } |
| |
| @Test |
| public void testTaskEntryAndExitEvaluators() |
| throws Exception |
| { |
| assertTaskIsAccepted( new BuildProjectTask( true, true, true, true ) ); |
| |
| assertTaskIsRejected( new BuildProjectTask( false, true, true, true ) ); |
| |
| assertTaskIsRejected( new BuildProjectTask( true, false, true, true ) ); |
| |
| assertTaskIsRejected( new BuildProjectTask( true, true, false, true ) ); |
| |
| assertTaskIsRejected( new BuildProjectTask( true, true, true, false ) ); |
| } |
| |
| @Test |
| public void testTaskViabilityEvaluators() |
| throws Exception |
| { |
| // The first and last task should be accepted |
| |
| Task task1 = new BuildProjectTask( 0 ); |
| |
| Task task2 = new BuildProjectTask( 10 ); |
| |
| Task task3 = new BuildProjectTask( 20 ); |
| |
| Task task4 = new BuildProjectTask( 30 ); |
| |
| Task task5 = new BuildProjectTask( 40 ); |
| |
| Task task6 = new BuildProjectTask( 100 ); |
| |
| assertTrue( taskQueue.put( task1 ) ); |
| |
| assertTrue( taskQueue.put( task2 ) ); |
| |
| assertTrue( taskQueue.put( task3 ) ); |
| |
| assertTrue( taskQueue.put( task4 ) ); |
| |
| assertTrue( taskQueue.put( task5 ) ); |
| |
| assertTrue( taskQueue.put( task6 ) ); |
| |
| Task actualTask1 = taskQueue.take(); |
| |
| assertNotNull( actualTask1 ); |
| |
| assertEquals( task1, actualTask1 ); |
| |
| Task actualTask6 = taskQueue.take(); |
| |
| assertNotNull( actualTask6 ); |
| |
| assertEquals( task6, actualTask6 ); |
| |
| assertNull( taskQueue.take() ); |
| } |
| |
| @Test |
| public void testRemoveTask() |
| throws Exception |
| { |
| Task task = new BuildProjectTask( 0 ); |
| |
| taskQueue.put( task ); |
| |
| taskQueue.remove( task ); |
| |
| assertNull( taskQueue.take() ); |
| } |
| |
| @Test |
| public void testRemoveAll() |
| throws Exception |
| { |
| |
| BlockingQueue<String> foo = new LinkedBlockingQueue<String>(); |
| foo.offer("1"); |
| foo.offer("2"); |
| |
| Task firstTask = new BuildProjectTask( 110 ); |
| |
| taskQueue.put( firstTask ); |
| |
| Task secondTask = new BuildProjectTask( 11120 ); |
| |
| taskQueue.put( secondTask ); |
| |
| assertEquals( 2, taskQueue.getQueueSnapshot().size() ); |
| |
| List tasks = new ArrayList(); |
| |
| tasks.add( firstTask ); |
| |
| tasks.add( secondTask ); |
| |
| taskQueue.removeAll( tasks ); |
| |
| assertTrue( taskQueue.getQueueSnapshot().isEmpty() ); |
| } |
| |
| // ---------------------------------------------------------------------- |
| // |
| // ---------------------------------------------------------------------- |
| |
| private void assertTaskIsAccepted( Task expectedTask ) |
| throws Exception |
| { |
| taskQueue.put( expectedTask ); |
| |
| Task actualTask = taskQueue.take(); |
| |
| assertEquals( expectedTask, actualTask ); |
| } |
| |
| private void assertTaskIsRejected( Task expectedTask ) |
| throws Exception |
| { |
| taskQueue.put( expectedTask ); |
| |
| Task actualTask = taskQueue.take(); |
| |
| assertNull( actualTask ); |
| } |
| } |