Java Code Examples for org.powermock.api.easymock.PowerMock#mockStatic()

@Before
public void setupMock(){

    List<InstanceInfo> servers = LoadBalancerTestUtils.getDummyInstanceInfo("dummy", HOST1, IP1, 8080);
    List<InstanceInfo> servers2 = LoadBalancerTestUtils.getDummyInstanceInfo("dummy", HOST2, IP2, 8080);
    servers.addAll(servers2);


    PowerMock.mockStatic(DiscoveryManager.class);
    PowerMock.mockStatic(DiscoveryClient.class);

    DiscoveryClient mockedDiscoveryClient = LoadBalancerTestUtils.mockDiscoveryClient();
    DiscoveryManager mockedDiscoveryManager = createMock(DiscoveryManager.class);

    expect(DiscoveryManager.getInstance()).andReturn(mockedDiscoveryManager).anyTimes();
    expect(mockedDiscoveryManager.getDiscoveryClient()).andReturn(mockedDiscoveryClient).anyTimes();

    expect(mockedDiscoveryClient.getInstancesByVipAddress("dummy", false, "region")).andReturn(servers).anyTimes();

    replay(DiscoveryManager.class);
    replay(DiscoveryClient.class);
    replay(mockedDiscoveryManager);
    replay(mockedDiscoveryClient);
}
 

@Test
public void testGetMissionToManifestMap() {
    final URL url = PowerMock.createMock(URL.class);
    PowerMock.mockStatic(JSONUtils.class);
    PowerMock.mockStatic(NetworkUtils.class);
    String manifest = null;
    try {
        manifest = WorkerTestUtils.readFile(ROOT_JSON_FILE);
    } catch (final IOException e1) {
        fail("Could not read file: " + ROOT_JSON_FILE);
    }
    try {
        NetworkUtils.getDataFromURL(url, null, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        PowerMock.expectLastCall().andReturn(manifest.getBytes());
        PowerMock.replayAll();
        final Map<String, String> missionMap = DynamoDBJSONRootWorker.getMissionToManifestMap(url, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        assertEquals(EXPECTED_MAP, missionMap);
    } catch (final IOException e) {
        fail(e.getMessage());
    }

}
 

@Test
public void testGetSolJSON() {
    final URL url = PowerMock.createMock(URL.class);
    PowerMock.mockStatic(JSONUtils.class);
    PowerMock.mockStatic(NetworkUtils.class);
    String manifest = null;
    try {
        manifest = WorkerTestUtils.readFile(IMAGE_RESOURCE_FILE);
    } catch (final IOException e1) {
        fail("Could not read file: " + IMAGE_RESOURCE_FILE);
    }
    try {
        NetworkUtils.getDataFromURL(url, null, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        PowerMock.expectLastCall().andReturn(manifest.getBytes());
        PowerMock.replayAll();
        final Map<Integer, String> actual = DynamoDBMissionWorker.getSolJSON(url, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        assertEquals(EXPECTED_MAP, actual);
    } catch (final IOException e) {
        fail(e.getMessage());
    }
}
 

@Before
public void setUp() throws Exception {
  helper.setUp();
  localUserMock = PowerMock.createMock(LocalUser.class);
  PowerMock.mockStatic(LocalUser.class);
  expect(LocalUser.getInstance()).andReturn(localUserMock).anyTimes();
  localUserMock.set(new User("1", "NonSuch", "NoName", null, 0, false, false, 0, null));
  expectLastCall().times(1);
  expect(localUserMock.getUserEmail()).andReturn("NonSuch").times(1);
  localUserInfo = PowerMock.createMock(OdeAuthFilter.UserInfo.class);
  expect(localUserInfo.buildCookie(false)).andReturn("NoCookie").anyTimes();
  expect(localUserInfo.buildCookie(true)).andReturn("NoCookie").anyTimes();
  mockFilterChain = PowerMock.createNiceMock(FilterChain.class);
  mockServletRequest = PowerMock.createNiceMock(HttpServletRequest.class);
  mockServletResponse = PowerMock.createNiceMock(HttpServletResponse.class);
}
 

@Before
public void setUp() throws Exception {
  PowerMock.mockStatic(LocalUser.class);
  localUserMock = PowerMock.createNiceMock(LocalUser.class);
  expect(LocalUser.getInstance()).andReturn(localUserMock).anyTimes();
  expect(localUserMock.getUserId()).andReturn(USER_ID).anyTimes();
  exporterMock = PowerMock.createNiceMock(FileExporterImpl.class);
  PowerMock.expectNew(FileExporterImpl.class).andReturn(exporterMock).anyTimes();

  storageIoMock = PowerMock.createNiceMock(StorageIo.class);
  StorageIoInstanceHolder.setInstance(storageIoMock);

  dummyZip = new ProjectSourceZip(DUMMY_ZIP_FILENAME, new byte[] {}, 2);
  dummyZipWithTitle = new ProjectSourceZip(DUMMY_ZIP_FILENAME_WITH_TITLE, new byte[] {}, 2);
  dummyApk = new RawFile(DUMMY_APK_FILENAME, new byte[] {});
  dummyFile = new RawFile(DUMMY_FILENAME, new byte[] {});
}
 

@Before
public void setupMock(){
    List<InstanceInfo> instances = LoadBalancerTestUtils.getDummyInstanceInfo("dummy", expected.getHost(), "127.0.0.1", expected.getPort());
    PowerMock.mockStatic(DiscoveryManager.class);
    PowerMock.mockStatic(DiscoveryClient.class);

    DiscoveryClient mockedDiscoveryClient = LoadBalancerTestUtils.mockDiscoveryClient();
    DiscoveryManager mockedDiscoveryManager = createMock(DiscoveryManager.class);

    expect(DiscoveryManager.getInstance()).andReturn(mockedDiscoveryManager).anyTimes();
    expect(mockedDiscoveryManager.getDiscoveryClient()).andReturn(mockedDiscoveryClient).anyTimes();

    expect(mockedDiscoveryClient.getInstancesByVipAddress("dummy:7001", false, null)).andReturn(instances).anyTimes();

    replay(DiscoveryManager.class);
    replay(DiscoveryClient.class);
    replay(mockedDiscoveryManager);
    replay(mockedDiscoveryClient);
}
 

@Before
public void setUp() {
    this.converter = new ObjectToCellValueConverter("USD");
    this.locale = Locale.US;
    this.stc = PowerMock.createMock(StylesContainerImpl.class);
    this.table = PowerMock.createMock(Table.class);
    final WriteUtil writeUtil = WriteUtil.create();
    this.xmlUtil = XMLUtil.create();

    this.tcc = TableColdCell.create(this.xmlUtil);
    this.ds = DataStylesBuilder.create(Locale.US).build();
    this.row = new TableRowImpl(writeUtil, this.xmlUtil, this.stc, this.ds, false, this.table,
            ROW_INDEX, 100);
    this.cell = new TableCellImpl(writeUtil, this.xmlUtil, this.stc, this.ds, false, this.row,
            COLUMN_INDEX);
    this.tcs = TableCellStyle.builder("name").build();
    PowerMock.mockStatic(TableColdCell.class);
}
 

@Before
public void setUp() {
    PowerMock.mockStatic(DiscoveryClient.class);

    EasyMock
            .expect(DiscoveryClient.getZone(EasyMock.isA(InstanceInfo.class)))
            .andReturn("zone")
            .anyTimes();

    eurekaClientMock = setUpEurekaClientMock(servers);
    eurekaClientProvider = new Provider<EurekaClient>() {
        @Override
        public EurekaClient get() {
            return eurekaClientMock;
        }
    };

    config = DefaultClientConfigImpl.getClientConfigWithDefaultValues();
    config.set(CommonClientConfigKey.DeploymentContextBasedVipAddresses, vipAddress);
    config.set(CommonClientConfigKey.ServerListUpdaterClassName, EurekaNotificationServerListUpdater.class.getName());
}
 

@org.junit.Test
public void test() throws Exception {

    PowerMock.mockStatic(StaticClass.class);
    EasyMock.expect(StaticClass.staticMethod()).andReturn(2).anyTimes();
    PowerMock.replay(StaticClass.class);

    int i = StaticClass.staticMethod();

    String xml = "<xml>" + i + "</xml>";
    SAXBuilder b = new SAXBuilder();

    Document d = b.build(new StringReader(xml));
    Assert.assertTrue(d.getRootElement().getText().equals("2"));
    PowerMock.verify(StaticClass.class);
}
 

@Test
public void shouldStartNamedTask() throws Exception {
  StartJobTask task = new StartJobTask();
  Map<String, List<String>> map = new HashMap<String, List<String>>();

  map.put("JOB_NAME", Arrays.asList("test"));
  OutputStream stream = new ByteArrayOutputStream();
  PrintWriter out = new PrintWriter(stream);

  PowerMock.mockStatic(SundialJobScheduler.class);
  SundialJobScheduler.startJob("test", Collections.singletonMap("JOB_NAME", (Object) "test"));
  EasyMock.expectLastCall();

  PowerMock.replay(SundialJobScheduler.class);

  task.execute(map, out);

  PowerMock.verify(SundialJobScheduler.class);
}
 

@Test
public void shouldPassParameters() throws Exception {
  StartJobTask task = new StartJobTask();
  Map<String, List<String>> map = new HashMap<String, List<String>>();
  map.put("JOB_NAME", Arrays.asList("test"));
  map.put("Param1", Arrays.asList("1"));
  map.put("Param2", Arrays.asList("2"));
  OutputStream stream = new ByteArrayOutputStream();
  PrintWriter out = new PrintWriter(stream);

  PowerMock.mockStatic(SundialJobScheduler.class);
  SundialJobScheduler.startJob(
      "test",
      ImmutableMap.of(
          "JOB_NAME", (Object) "test", "Param1", (Object) "1", "Param2", (Object) "2"));
  EasyMock.expectLastCall();

  PowerMock.replay(SundialJobScheduler.class);

  task.execute(map, out);

  PowerMock.verify(SundialJobScheduler.class);
}
 

@Test
public void testgetJSONFromURL() {
    PowerMock.mockStatic(NetworkUtils.class);
    try {
        final URL url = PowerMock.createMock(URL.class);
        NetworkUtils.getDataFromURL(url, null, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        PowerMock.expectLastCall().andReturn(ROOT_JSON.getBytes());
        replayAll();
        final ObjectMapper mapper = new ObjectMapper();
        final JsonNode json = mapper.readTree(ROOT_JSON);
        final JsonNode result = JSONParser.getJSONFromURL(url, ImageIngester.DEFAULT_CONNECT_TIMEOUT);
        assertEquals(json, result);
    } catch (final IOException e) {
        fail();
    }

}
 

@Test
public void testCreateResourceTable() {
    final AmazonDynamoDB dynamoDB = PowerMock.createMock(AmazonDynamoDB.class);
    PowerMock.mockStatic(DynamoDBManager.class);
    final CreateTableRequest request = new CreateTableRequest();
    request.setAttributeDefinitions(MarsDynamoDBManager.RESOURCE_TABLE_ATTRIBUTE_DEFINITIONS);
    request.setKeySchema(MarsDynamoDBManager.RESOURCE_TABLE_KEY_SCHEMA);
    request.setProvisionedThroughput(PROVISIONED_THROUGHPUT);
    request.setTableName(TABLE_NAME);

    DynamoDBManager.createTable(dynamoDB, request);
    PowerMock.expectLastCall().andReturn(null);
    PowerMock.replayAll();
    MarsDynamoDBManager.createResourceTable(dynamoDB, TABLE_NAME, PROVISIONED_THROUGHPUT);
    PowerMock.verifyAll();
}
 

@Before
public void setupCacheBean() throws IllegalAccessException, SecurityException, NoSuchMethodException {
    qlCache = new CreatedQueryLogicCacheBean();
    queryLogic = PowerMock.createMock(QueryLogic.class);
    conn = PowerMock.createMock(Connector.class);
    internalCache = new ConcurrentHashMap<>();
    
    PowerMock.field(CreatedQueryLogicCacheBean.class, "cache").set(qlCache, internalCache);
    
    PowerMock.mockStatic(System.class, System.class.getMethod("currentTimeMillis"));
}
 

@Before
public void setUp() throws InterruptedException {
    PowerMock.resetAll();
    PowerMock.mockStatic(Thread.class);
    Thread.sleep(EasyMock.anyLong());
    PowerMock.expectLastCall().andThrow(new InterruptedException()).anyTimes();
}
 

@Test
public void testCreateImageTable() {
    final AmazonDynamoDB dynamoDB = PowerMock.createMock(AmazonDynamoDB.class);
    PowerMock.mockStatic(DynamoDBManager.class);
    final CreateTableRequest request = new CreateTableRequest();
    request.setAttributeDefinitions(MarsDynamoDBManager.IMAGE_TABLE_ATTRIBUTE_DEFINITIONS);
    request.setKeySchema(MarsDynamoDBManager.IMAGE_TABLE_KEY_SCHEMA);
    final GlobalSecondaryIndex timeGSI = new GlobalSecondaryIndex();
    timeGSI.setIndexName(MarsDynamoDBManager.IMAGE_TABLE_TIME_GSI_NAME);
    timeGSI.setKeySchema(Arrays.asList(MarsDynamoDBManager.IMAGE_TABLE_TIME_GSI_HASH_KSE,
        MarsDynamoDBManager.IMAGE_TABLE_TIME_GSI_RANGE_KSE));
    timeGSI.setProjection(MarsDynamoDBManager.IMAGE_TABLE_TIME_GSI_PROJECTION);
    timeGSI.setProvisionedThroughput(PROVISIONED_THROUGHPUT);
    final GlobalSecondaryIndex voteGSI = new GlobalSecondaryIndex();
    voteGSI.setIndexName(MarsDynamoDBManager.IMAGE_TABLE_VOTE_GSI_NAME);
    voteGSI.setKeySchema(Arrays.asList(MarsDynamoDBManager.IMAGE_TABLE_VOTE_GSI_HASH_KSE,
        MarsDynamoDBManager.IMAGE_TABLE_VOTE_GSI_RANGE_KSE));
    voteGSI.setProjection(MarsDynamoDBManager.IMAGE_TABLE_VOTE_GSI_PROJECTION);
    voteGSI.setProvisionedThroughput(PROVISIONED_THROUGHPUT);
    request.setGlobalSecondaryIndexes(Arrays.asList(timeGSI, voteGSI));
    request.setProvisionedThroughput(PROVISIONED_THROUGHPUT);
    request.setTableName(TABLE_NAME);

    DynamoDBManager.createTable(dynamoDB, request);
    PowerMock.expectLastCall().andReturn(null);
    PowerMock.replayAll();
    MarsDynamoDBManager.createImageTable(dynamoDB, TABLE_NAME, PROVISIONED_THROUGHPUT, PROVISIONED_THROUGHPUT,
        PROVISIONED_THROUGHPUT);
    PowerMock.verifyAll();

}
 

private void basicExpects(Document d, Key key, Map.Entry<Key,Value> entry) throws MarkingFunctions.Exception {
    // AbstractMap.SimpleEntry<Key, Document> documentEntry = new AbstractMap.SimpleEntry<>(key, mockDocument);
    d.put("field1", new Numeric("5", key, true));
    AbstractMap.SimpleEntry<Key,Document> documentEntry = new AbstractMap.SimpleEntry<>(key, d);
    
    // Map<String,Attribute<?extends Comparable<?>>> dictionary = new HashMap<>();
    // dictionary.put("field1", mockNumeric);
    
    SimpleField simpleField = new SimpleField();
    SimpleEvent simpleEvent = new SimpleEvent();
    
    PowerMock.mockStatic(DocumentSerialization.class);
    EasyMock.expect(DocumentSerialization.getDocumentDeserializer(mockQuery)).andReturn(mockDeserializer);
    
    EasyMock.expect(mockLogic.getTableName()).andReturn("table1");
    EasyMock.expect(mockQuery.getQueryAuthorizations()).andReturn("A,B,C");
    EasyMock.expect(mockQuery.findParameter("log.timing.details")).andReturn(new QueryImpl.Parameter("", ""));
    EasyMock.expect(mockDeserializer.apply(entry)).andReturn(documentEntry);
    // EasyMock.expect(mockDocument.getDictionary()).andReturn(Collections.EMPTY_MAP);
    // mockDocument.debugDocumentSize(key);
    EasyMock.expect(mockMarkingFunctions.translateFromColumnVisibility(key.getColumnVisibilityParsed())).andReturn(Collections.EMPTY_MAP);
    // EasyMock.expect(mockDocument.getDictionary()).andReturn(dictionary);
    // EasyMock.expect(mockNumeric.getData()).andReturn("5");
    EasyMock.expect(mockResponseFactory.getField()).andReturn(simpleField);
    EasyMock.expect(mockResponseFactory.getEvent()).andReturn(simpleEvent);
    // EasyMock.expect(mockDocument.sizeInBytes()).andReturn(1l);
}
 

@Test
public void testIsExecutable() {
  PowerMock.mockStatic(Paths.class);
  PowerMock.mockStatic(Files.class);

  final Path path = mock(Path.class);

  Paths.get("test-file");
  PowerMock.expectLastCall().andReturn(path);

  Files.isRegularFile(path);
  PowerMock.expectLastCall().andReturn(false);

  PowerMock.replayAll();
  assertFalse(processLauncher.isExecutable("test-file"));
  PowerMock.verifyAll();
  PowerMock.resetAll();

  // --

  Paths.get("test-file");
  PowerMock.expectLastCall().andReturn(path);

  Files.isRegularFile(path);
  PowerMock.expectLastCall().andReturn(true);

  Files.isReadable(path);
  PowerMock.expectLastCall().andReturn(false);

  PowerMock.replayAll();
  assertFalse(processLauncher.isExecutable("test-file"));
  PowerMock.verifyAll();
  PowerMock.resetAll();

  // --

  Paths.get("test-file");
  PowerMock.expectLastCall().andReturn(path);

  Files.isRegularFile(path);
  PowerMock.expectLastCall().andReturn(true);

  Files.isReadable(path);
  PowerMock.expectLastCall().andReturn(true);

  Files.isExecutable(path);
  PowerMock.expectLastCall().andReturn(false);

  PowerMock.replayAll();
  assertFalse(processLauncher.isExecutable("test-file"));
  PowerMock.verifyAll();
  PowerMock.resetAll();

  // --

  Paths.get("test-file");
  PowerMock.expectLastCall().andReturn(path);

  Files.isRegularFile(path);
  PowerMock.expectLastCall().andReturn(true);

  Files.isReadable(path);
  PowerMock.expectLastCall().andReturn(true);

  Files.isExecutable(path);
  PowerMock.expectLastCall().andReturn(true);

  PowerMock.replayAll();
  assertTrue(processLauncher.isExecutable("test-file"));
  PowerMock.verifyAll();
}
 

@Before
public void setup() throws Exception {
    FORWARD_FIELD_MAPPING = new FieldMapping();
    FORWARD_FIELD_MAPPING.setColumnVisibility(COLVIZ);
    FORWARD_FIELD_MAPPING.setDatatype(DATATYPE);
    FORWARD_FIELD_MAPPING.setDirection(FORWARD);
    FORWARD_FIELD_MAPPING.setFieldName(FIELD_NAME);
    FORWARD_FIELD_MAPPING.setModelFieldName(MODEL_FIELD_NAME);
    REVERSE_FIELD_MAPPING = new FieldMapping();
    REVERSE_FIELD_MAPPING.setColumnVisibility(COLVIZ);
    REVERSE_FIELD_MAPPING.setDatatype(DATATYPE);
    REVERSE_FIELD_MAPPING.setDirection(REVERSE);
    REVERSE_FIELD_MAPPING.setFieldName(FIELD_NAME);
    REVERSE_FIELD_MAPPING.setModelFieldName(MODEL_FIELD_NAME);
    NULL_CV_MAPPING = new FieldMapping();
    NULL_CV_MAPPING.setColumnVisibility("");
    NULL_CV_MAPPING.setDatatype(DATATYPE);
    NULL_CV_MAPPING.setDirection(REVERSE);
    NULL_CV_MAPPING.setFieldName(FIELD_NAME);
    NULL_CV_MAPPING.setModelFieldName(MODEL_FIELD_NAME);
    FORWARD_KEY = new Key(MODEL_FIELD_NAME, MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, FIELD_NAME + ModelKeyParser.NULL_BYTE + FORWARD.getValue(),
                    COLVIZ, TIMESTAMP);
    REVERSE_KEY = new Key(FIELD_NAME, MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, MODEL_FIELD_NAME + ModelKeyParser.NULL_BYTE + REVERSE.getValue(),
                    COLVIZ, TIMESTAMP);
    NULL_CV_KEY = new Key(FIELD_NAME, MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, MODEL_FIELD_NAME + ModelKeyParser.NULL_BYTE + REVERSE.getValue(),
                    "", TIMESTAMP);
    FORWARD_MUTATION = new Mutation(MODEL_FIELD_NAME);
    FORWARD_MUTATION.put(MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, FIELD_NAME + ModelKeyParser.NULL_BYTE + FORWARD.getValue(),
                    new ColumnVisibility(COLVIZ), TIMESTAMP, ModelKeyParser.NULL_VALUE);
    FORWARD_DELETE_MUTATION = new Mutation(MODEL_FIELD_NAME);
    FORWARD_DELETE_MUTATION.putDelete(MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, FIELD_NAME + ModelKeyParser.NULL_BYTE + FORWARD.getValue(),
                    new ColumnVisibility(COLVIZ), TIMESTAMP);
    FORWARD_DELETE_MUTATION.putDelete(MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, FIELD_NAME + ModelKeyParser.NULL_BYTE + "index_only"
                    + ModelKeyParser.NULL_BYTE + FORWARD.getValue(), new ColumnVisibility(COLVIZ), TIMESTAMP);
    
    REVERSE_MUTATION = new Mutation(FIELD_NAME);
    REVERSE_MUTATION.put(MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, MODEL_FIELD_NAME + ModelKeyParser.NULL_BYTE + REVERSE.getValue(),
                    new ColumnVisibility(COLVIZ), TIMESTAMP, ModelKeyParser.NULL_VALUE);
    REVERSE_DELETE_MUTATION = new Mutation(FIELD_NAME);
    REVERSE_DELETE_MUTATION.putDelete(MODEL_NAME + ModelKeyParser.NULL_BYTE + DATATYPE, MODEL_FIELD_NAME + ModelKeyParser.NULL_BYTE + REVERSE.getValue(),
                    new ColumnVisibility(COLVIZ), TIMESTAMP);
    
    PowerMock.mockStatic(System.class, System.class.getMethod("currentTimeMillis"));
}
 

@Before
public void setupMock(){

    List<InstanceInfo> dummyII = LoadBalancerTestUtils.getDummyInstanceInfo("dummy", "http://www.host.com", "1.1.1.1", 8001);
    List<InstanceInfo> secureDummyII = LoadBalancerTestUtils.getDummyInstanceInfo("secureDummy", "http://www.host.com", "1.1.1.1", 8002);


    PowerMock.mockStatic(DiscoveryManager.class);
    PowerMock.mockStatic(DiscoveryClient.class);

    DiscoveryClient mockedDiscoveryClient = LoadBalancerTestUtils.mockDiscoveryClient();
    DiscoveryManager mockedDiscoveryManager = createMock(DiscoveryManager.class);

    expect(DiscoveryManager.getInstance()).andReturn(mockedDiscoveryManager).anyTimes();
    expect(mockedDiscoveryManager.getDiscoveryClient()).andReturn(mockedDiscoveryClient).anyTimes();

    expect(mockedDiscoveryClient.getInstancesByVipAddress("dummy", false, "region")).andReturn(dummyII).anyTimes();
    expect(mockedDiscoveryClient.getInstancesByVipAddress("secureDummy", true, "region")).andReturn(secureDummyII).anyTimes();

    replay(DiscoveryManager.class);
    replay(DiscoveryClient.class);
    replay(mockedDiscoveryManager);
    replay(mockedDiscoveryClient);

}