1   /*
2    *
3    * The DbUnit Database Testing Framework
4    * Copyright (C)2005, DbUnit.org
5    *
6    * This library is free software; you can redistribute it and/or
7    * modify it under the terms of the GNU Lesser General Public
8    * License as published by the Free Software Foundation; either
9    * version 2.1 of the License, or (at your option) any later version.
10   *
11   * This library is distributed in the hope that it will be useful,
12   * but WITHOUT ANY WARRANTY; without even the implied warranty of
13   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14   * Lesser General Public License for more details.
15   *
16   * You should have received a copy of the GNU Lesser General Public
17   * License along with this library; if not, write to the Free Software
18   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19   *
20   */
21  
22  package org.dbunit.util.search;
23  
24  import java.util.SortedSet;
25  
26  /**
27   * @author Felipe Leme (dbunit@felipeal.net)
28   * @version $Revision: 769 $
29   * @since Aug 25, 2005
30   */
31  
32  public class IncludeNodesSearchCallbackTest extends AbstractSearchTestCase {
33  
34    private ISearchCallback callback;
35  
36    protected ISearchCallback getCallback() {
37      return this.callback;    
38    }
39    
40    protected void setAllowed( Object[] allowedNodes ) {
41      this.callback =  new AbstractIncludeNodesSearchCallback( allowedNodes ) {
42        
43        public SortedSet getEdges(Object fromNode) throws SearchException {
44          return getEdgesFromNode(fromNode);
45        }
46      };
47      
48    }
49      
50    public void testSingleNode() throws Exception {
51      setInput(new String[] { A });
52      setAllowed( new String[] { A } );
53      setOutput(new String[] { A });
54      doIt();
55    }
56  
57    public void testSingleEdgeAllowedA() throws Exception {
58      setInput(new String[] { A });
59      addEdges(A, new String[] { B });
60      setAllowed( new String[] { A } );
61      setOutput(new String[] { A });
62      doIt();
63    }
64  
65    public void testSingleEdgeAllowedB() throws Exception {
66      setInput(new String[] { A });
67      addEdges(A, new String[] { B });
68      setAllowed( new String[] { B } );
69      doIt();
70    }
71    public void testSingleEdgeMultipleInputAllowedB() throws Exception {
72      setInput(new String[] { A, B });
73      addEdges(A, new String[] { B });
74      setAllowed( new String[] { B } );
75      setOutput(new String[] { B });
76      doIt();
77    }
78  
79    public void testDisconnected() throws Exception {
80      setInput(new String[] { A, C });
81      addEdges(A, new String[] { B });
82      setAllowed( new String[] { B, A } );
83      setOutput(new String[] { B, A });
84      doIt();
85    }
86  
87    public void testDisconnectedAllowedC() throws Exception {
88      setInput(new String[] { A, C });
89      addEdges(A, new String[] { B });
90      setAllowed( new String[] { C } );
91      setOutput(new String[] { C });
92      doIt();
93    }
94    
95  
96  }