1   package org.apache.velocity.test.issues;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *   http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.    
20   */
21  
22  import org.apache.velocity.app.VelocityEngine;
23  import org.apache.velocity.test.BaseTestCase;
24  import org.apache.velocity.exception.ResourceNotFoundException;
25  
26  /**
27   * This class tests VELOCITY-742.
28   */
29  public class Velocity742TestCase extends BaseTestCase
30  {
31      public Velocity742TestCase(String name)
32      {
33          super(name);
34      }
35      
36      protected void setUpEngine(VelocityEngine engine)
37      {
38          // we need to call init here because otherwise it is not called until assertEvalEquals
39          // and therefore the removeDirective call is ignored.
40          engine.init();
41      }
42  
43      public void testDisableAndRestoreDirective()
44      {
45          String s = "#include('doesnotexist.vm') directive is disabled";
46          
47          // first remove  the #include directive and see that is treated as normal text
48          engine.removeDirective("include");
49          assertEvalEquals(s, s);
50          
51          // now reload the directive and see that the include directive works again and
52          // Velocity throws ResourceNotFoundException because it can't find the template
53          engine.loadDirective("org.apache.velocity.runtime.directive.Include");
54          assertEvalException(s, ResourceNotFoundException.class);
55      }
56  }