1   package org.displaytag.jsptests;
2   
3   import org.displaytag.properties.MediaTypeEnum;
4   import org.displaytag.tags.TableTagParameters;
5   import org.displaytag.test.DisplaytagCase;
6   import org.displaytag.util.ParamEncoder;
7   
8   import com.meterware.httpunit.GetMethodWebRequest;
9   import com.meterware.httpunit.WebRequest;
10  import com.meterware.httpunit.WebResponse;
11  import com.meterware.httpunit.WebTable;
12  
13  
14  /***
15   * Tests for optimized iterations (don't evaluate unneeded body of columns).
16   * @author Fabrizio Giustina
17   * @version $Revision: 1.2 $ ($Author: fgiust $)
18   */
19  public class OptimizedIteration2Test extends DisplaytagCase
20  {
21  
22      /***
23       * @see org.displaytag.test.DisplaytagCase#getJspName()
24       */
25      public String getJspName()
26      {
27          return "optimizediteration2.jsp";
28      }
29  
30      /***
31       * Verifies that the generated page contains the pagination links with the inupt parameter. Tests #917200 ("{}" in
32       * parameters).
33       * @param jspName jsp name, with full path
34       * @throws Exception any axception thrown during test.
35       */
36      public void doTest(String jspName) throws Exception
37      {
38  
39          WebRequest request = new GetMethodWebRequest(jspName);
40          ParamEncoder encoder = new ParamEncoder("table");
41  
42          // page 1, not sorted
43          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "1");
44          checkNumberOfIterations(runner.getResponse(request), 1);
45  
46          // page 2, not sorted
47          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "2");
48          checkNumberOfIterations(runner.getResponse(request), 1);
49  
50          // page 1, sorted (only current page)
51          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_SORT), "1");
52          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "1");
53          checkNumberOfIterations(runner.getResponse(request), 1);
54  
55          // page 2, sorted (only current page)
56          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_SORT), "1");
57          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "2");
58          checkNumberOfIterations(runner.getResponse(request), 1);
59  
60          // page 1, export single page
61          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "2");
62          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_EXPORTTYPE), Integer
63              .toString(MediaTypeEnum.CSV.getCode()));
64          request.setParameter(encoder.encodeParameterName(TableTagParameters.PARAMETER_PAGE), "1");
65  
66          WebResponse response = runner.getResponse(request);
67          String csvExport = response.getText();
68          if (log.isDebugEnabled())
69          {
70              log.debug(response.getText());
71          }
72  
73          assertEquals("Wrong csv export", "ant,1\n", csvExport);
74  
75      }
76  
77      /***
78       * @param response WebResponse
79       * @param iterations expected number of iterations
80       * @throws Exception any axception thrown during test.
81       */
82      private void checkNumberOfIterations(WebResponse response, int iterations) throws Exception
83      {
84          if (log.isDebugEnabled())
85          {
86              log.debug(response.getText());
87          }
88  
89          WebTable[] tables = response.getTables();
90          assertEquals("Expected 1 table in result.", 1, tables.length);
91          assertEquals("Expected 2 rows in table.", 2, tables[0].getRowCount());
92  
93          assertEquals("Wrong number of iterations. Evaluated column bodies number is different from expected", Integer
94              .toString(iterations), response.getElementWithID("iterations").getText());
95      }
96  }