How to use UtilTest class of org.easymock.tests package

1// Copyright 2010 Google Inc. All Rights Reserved.
2
3package com.google.appinventor.blockseditor.jsonp;
4
5import com.google.appinventor.common.testutils.TestUtils;
6
7import junit.framework.TestCase;
8
9import org.easymock.EasyMock;
10
11import java.io.File;
12import java.io.FileInputStream;
13import java.io.FileNotFoundException;
14import java.io.IOException;
15import java.io.InputStream;
16import java.net.URLConnection;
17import java.util.Map;
18
19
20/**
21 * Unit tests for {@link Util}.
22 *
23 * @author sharon@google.com (Sharon Perl)
24 */
25public class UtilTest extends TestCase {
26
27  private URLConnection mockConn = EasyMock.createMock(URLConnection.class);
28  private static final String TESTING_SOURCE_PATH = TestUtils.APP_INVENTOR_ROOT_DIR +
29      "/blockseditor/tests/com/google/appinventor/blockseditor/jsonp/testing_source_files/";
30
31
32  public void testDownloadZipFile() throws FileNotFoundException, IOException {
33    File zipFile = new File(TESTING_SOURCE_PATH + "zipfile.zip");
34    InputStream zipInput = new FileInputStream(zipFile);
35    assertNotNull(zipInput);
36    
37    EasyMock.expect(mockConn.getInputStream()).andReturn(zipInput);
38    EasyMock.replay(mockConn);
39    
40    Map<String,String> zipFiles = Util.downloadZipFile(mockConn);
41    // check that we got back the files we expect
42    assertTrue(zipFiles.containsKey("assets/cat.wav"));
43    assertTrue(zipFiles.containsKey("assets/kitty.jpg"));
44    assertEquals(2, zipFiles.size());
45    // check that we were able to give the unzipped files their proper names
46    assertTrue(zipFiles.get("assets/cat.wav").contains("assets/cat.wav"));
47    assertTrue(zipFiles.get("assets/kitty.jpg").contains("assets/kitty.jpg"));
48    // check that we extracted the file contents that we expect
49    assertTrue(filesEqual(TESTING_SOURCE_PATH + "assets/cat.wav", zipFiles.get("assets/cat.wav")));
50    assertTrue(filesEqual(TESTING_SOURCE_PATH + "assets/kitty.jpg", 
51        zipFiles.get("assets/kitty.jpg")));
52    EasyMock.verify(mockConn);
53  }
54  
55  private String basename(String path) {
56    return path.substring(path.lastIndexOf('/') + 1);
57  }
58  
59  private boolean filesEqual(String filePath1, String filePath2) throws IOException {
60    File file1 = new File(filePath1);
61    File file2 = new File(filePath2);
62    InputStream inp1 = new FileInputStream(file1);
63    InputStream inp2 = new FileInputStream(file2);
64    int b1;
65    int b2;
66    while ((b1 = inp1.read()) == (b2 = inp2.read()) && b1 != -1);
67    if (b1 != -1 || b2 != -1) {
68      return false;
69    }
70    return true;
71  }
72}
73

Source: UtilTest.java

// Copyright 2010 Google Inc. All Rights Reserved.

package com.google.appinventor.blockseditor.jsonp;

import com.google.appinventor.common.testutils.TestUtils;

import junit.framework.TestCase;

import org.easymock.EasyMock;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLConnection;
import java.util.Map;


/**
 * Unit tests for {@link Util}.
 *
 * @author sharon@google.com (Sharon Perl)
 */
public class UtilTest extends TestCase {

  private URLConnection mockConn = EasyMock.createMock(URLConnection.class);
  private static final String TESTING_SOURCE_PATH = TestUtils.APP_INVENTOR_ROOT_DIR +
      "/blockseditor/tests/com/google/appinventor/blockseditor/jsonp/testing_source_files/";


  public void testDownloadZipFile() throws FileNotFoundException, IOException {
    File zipFile = new File(TESTING_SOURCE_PATH + "zipfile.zip");
    InputStream zipInput = new FileInputStream(zipFile);
    assertNotNull(zipInput);
    
    EasyMock.expect(mockConn.getInputStream()).andReturn(zipInput);
    EasyMock.replay(mockConn);
    
    Map<String,String> zipFiles = Util.downloadZipFile(mockConn);
    // check that we got back the files we expect
    assertTrue(zipFiles.containsKey("assets/cat.wav"));
    assertTrue(zipFiles.containsKey("assets/kitty.jpg"));
    assertEquals(2, zipFiles.size());
    // check that we were able to give the unzipped files their proper names
    assertTrue(zipFiles.get("assets/cat.wav").contains("assets/cat.wav"));
    assertTrue(zipFiles.get("assets/kitty.jpg").contains("assets/kitty.jpg"));
    // check that we extracted the file contents that we expect
    assertTrue(filesEqual(TESTING_SOURCE_PATH + "assets/cat.wav", zipFiles.get("assets/cat.wav")));
    assertTrue(filesEqual(TESTING_SOURCE_PATH + "assets/kitty.jpg", 
        zipFiles.get("assets/kitty.jpg")));
    EasyMock.verify(mockConn);
  }
  
  private String basename(String path) {
    return path.substring(path.lastIndexOf('/') + 1);
  }
  
  private boolean filesEqual(String filePath1, String filePath2) throws IOException {
    File file1 = new File(filePath1);
    File file2 = new File(filePath2);
    InputStream inp1 = new FileInputStream(file1);
    InputStream inp2 = new FileInputStream(file2);
    int b1;
    int b2;
    while ((b1 = inp1.read()) == (b2 = inp2.read()) && b1 != -1);
    if (b1 != -1 || b2 != -1) {
      return false;
    }
    return true;
  }
}