Java Examples for com.squareup.okhttp.mockwebserver.MockWebServer

@BeforeMethod(alwaysRun = true)
public void setUp() throws Exception {
    app = ApplicationBuilder.newManagedApp(TestApplication.class);
    managementContext = app.getManagementContext();
    locationManager = managementContext.getLocationManager();
    loc = locationManager.createLocation(LocationSpec.create(SshMachineLocation.class).configure("address", "localhost"));
    mockWebServer = new MockWebServer();
}

@Provides
public RestAdapter provideRestAdapter(MockWebServer mockWebServer) {
    int SIZE_OF_CACHE = 1024;
    OkHttpClient ok = new OkHttpClient();
    ok.setReadTimeout(30, TimeUnit.SECONDS);
    ok.setConnectTimeout(30, TimeUnit.SECONDS);
    try {
        Cache responseCache = new Cache(DevConApplication.getInstance().getCacheDir(), SIZE_OF_CACHE);
        ok.setCache(responseCache);
    } catch (Exception e) {
        Log.d("OkHttp", "Unable to set http cache", e);
    }
    Executor executor = Executors.newCachedThreadPool();
    return new RestAdapter.Builder().setExecutors(executor, executor).setClient(new OkClient(ok)).setEndpoint(mockWebServer.getUrl("/").toString()).setRequestInterceptor(new ApiRequestInterceptor()).build();
}

public void test() throws Exception {
    // Create a MockWebServer. These are lean enough that you can create a new
    // instance for every unit test.
    MockWebServer server = new MockWebServer();
    server.enqueue(new MockResponse().setBody("hello, world!"));
    server.start();
    Bus bus = Mockito.mock(Bus.class);
    new CheckTask(bus, "michaelgruczel", "The-missing-android-guideline", new OkHttpClient()).execute();
    wait(2000);
    Mockito.verify(bus, Mockito.times(1)).post(Matchers.eq(CheckEvent.class));
    server.shutdown();
}

@Before
public void setUp() throws Exception {
    super.setUp();
    server = new MockWebServer();
    server.start();
    final Dispatcher dispatcher = new Dispatcher() {

        @Override
        public MockResponse dispatch(RecordedRequest request) throws InterruptedException {
            String p = request.getPath();
            MockResponse response = new MockResponse().setResponseCode(200);
            if (p.matches("/board/all.*") || p.matches("/board/moderator/all.*")) {
                return response.setBody(jsonReader.readString("json/bulletins.json"));
            } else if (p.equals("/board") || p.equals("/board/" + BULLETIN_ID)) {
                return response.setBody(jsonReader.readString("json/ok.json"));
            } else if (p.equals("/board/" + BULLETIN_ID + "/recipient")) {
                return response.setBody(jsonReader.readString("json/bull_recipients.json"));
            } else if (p.equals("/subdivision/" + SUBDIV_ID + "/children")) {
                return response.setBody(jsonReader.readString("json/desc_subdivision.json"));
            } else if (p.equals("/roles")) {
                return response.setBody(jsonReader.readString("json/roles" + ".json"));
            } else if (p.equals("/subdivision/" + SUBDIV_ID + "/group")) {
                return response.setBody(jsonReader.readString("json/groups.json"));
            }
            return new MockResponse().setResponseCode(404);
        }
    };
    server.setDispatcher(dispatcher);
    HttpUrl baseUrl = server.url("/");
    service = ServiceCreator.createTestService(baseUrl.toString(), BulletinService.class);
}

@BeforeClass
public static void given_mock_web_server_is_started() throws IOException {
    mockWebServer = new MockWebServer();
    InputStream resourceAsStream = Thread.currentThread().getContextClassLoader().getResourceAsStream("hystrix_circuit_breaker_opened.txt");
    String hystrixEvent = new Scanner(resourceAsStream, UTF_8.name()).useDelimiter("\\A").next();
    mockWebServer.enqueue(new MockResponse().setResponseCode(200).setBody(hystrixEvent));
    mockWebServer.play(6543);
}

public void testList() throws Exception {
    MockWebServer server = mockOpenStackServer();
    server.enqueue(addCommonHeaders(new MockResponse().setBody(stringFromResource("/access.json"))));
    server.enqueue(addCommonHeaders(containerResponse().addHeader(CONTAINER_READ, CONTAINER_ACL_ANYBODY_READ).setBody(stringFromResource("/object_list.json"))));
    try {
        SwiftApi api = api(server.getUrl("/").toString(), "openstack-swift");
        ObjectList objects = api.getObjectApi("DFW", "myContainer").list();
        assertEquals(objects, parsedObjectsForUrl(server.getUrl("/").toString()));
        assertEquals(objects.getContainer().getName(), "myContainer");
        assertTrue(objects.getContainer().getAnybodyRead().get());
        // Check MD5 is parsed from the ETag header.
        SwiftObject object1 = objects.get(1);
        assertEquals(base16().lowerCase().decode(object1.getETag()), object1.getPayload().getContentMetadata().getContentMD5AsHashCode().asBytes());
        assertEquals(server.getRequestCount(), 2);
        assertAuthentication(server);
        assertRequest(server.takeRequest(), "GET", "/v1/MossoCloudFS_5bcf396e-39dd-45ff-93a1-712b9aba90a9/myContainer?format=json");
    } finally {
        server.shutdown();
    }
}

public void testWhenNamespaceInExtensionsLBaaSPresent() throws IOException {
    MockWebServer server = mockOpenStackServer();
    server.enqueue(addCommonHeaders(new MockResponse().setBody(stringFromResource("/access.json"))));
    server.enqueue(addCommonHeaders(new MockResponse().setResponseCode(200).setBody(stringFromResource("/extension_list_with_lbaas_v1_response.json"))));
    try {
        NeutronApi neutronApi = api(server.getUrl("/").toString(), "openstack-neutron", overrides);
        Optional<LBaaSApi> lbaasApiExtension = neutronApi.getLBaaSApi("RegionOne");
        assertAuthentication(server);
        /*
          * Check response
          */
        assertNotNull(lbaasApiExtension);
        assertEquals(lbaasApiExtension.isPresent(), true, "LBaaS API Version 1 is expected to be available");
    } finally {
        server.shutdown();
    }
}

@Before
public void before() throws Throwable {
    super.before();
    // Create a MockWebServer. These are lean enough that you can create a new
    // instance for every unit test.
    server = new MockWebServer();
    try {
        server.start();
    } catch (IOException e) {
        logger.log(Level.INFO, "Failed to start MockWebServer");
        throw new RuntimeException(e);
    }
}

@Before
public void setUp() throws Exception {
    ShadowLog.stream = System.out;
    mockWebServer = new MockWebServer();
    mockWebServer.start(9999);
    baseRetryHandler = BaseRetryHandler_.getInstance_(RuntimeEnvironment.application);
    connectivityAwareUrlClient = ConnectivityAwareUrlClient_.getInstance_(RuntimeEnvironment.application);
    connectivityAwareUrlClient.setWrappedClient(new OkClient(new OkHttpClient()));
    connectivityAwareUrlClient.setRetryHandler(baseRetryHandler);
    RestAdapter restAdapter = new RestAdapter.Builder().setClient(connectivityAwareUrlClient).setLogLevel(RestAdapter.LogLevel.FULL).setEndpoint(mockWebServer.getUrl("/").toString()).build();
    sodaService = restAdapter.create(SodaService.class);
}

@Before
public void setup() throws Exception {
    //setup mocServer
    mServer = new MockWebServer();
    mServer.setDispatcher(NCMBDispatcher.dispatcher);
    mServer.start();
    String mocServerUrl = mServer.getUrl("/").toString();
    //initialization
    NCMB.initialize(RuntimeEnvironment.application, "appKey", "clientKKey", mocServerUrl, null);
    ShadowLog.stream = System.out;
    Robolectric.getBackgroundThreadScheduler().pause();
    Robolectric.getForegroundThreadScheduler().pause();
    callbackFlag = false;
}

@Test
public final void testSuccess() throws Throwable {
    MockWebServer server = new MockWebServer();
    server.enqueue(new MockResponse().setResponseCode(200));
    server.play();
    String path = "/foo/bar/data/";
    String callbackUrl = server.getUrl(path).toString();
    ObjectNode sentData = new ObjectNode(JsonNodeFactory.instance);
    sentData.put("sourceId", SOURCE_ID);
    ArrayNode data = sentData.putArray("data");
    data.addObject().put("hello", "world");
    data.addObject().put("and", "again");
    setupSender(new HttpClient(SOURCE_ID, callbackUrl, true));
    sender.onNewDataStart();
    for (JsonNode item : data) sender.onNewDataItem((ObjectNode) item);
    sender.onNewDataComplete();
    assertEquals(SenderResult.Status.SUCCESS, sender.getSenderResult().getStatus());
    RecordedRequest request = server.takeRequest();
    assertEquals(path, request.getPath());
    assertEquals(sentData.toString(), request.getUtf8Body());
    server.shutdown();
}

static void setupLogging() {
    final long start = currentTimeMillis();
    ConsoleHandler handler = new ConsoleHandler();
    handler.setLevel(Level.FINE);
    handler.setFormatter(new Formatter() {

        @Override
        public String format(LogRecord record) {
            return String.format("%7d - %s%n", record.getMillis() - start, record.getMessage());
        }
    });
    Logger[] loggers = { Logger.getLogger(DenominatorD.class.getPackage().getName()), Logger.getLogger(feign.Logger.class.getName()), Logger.getLogger(MockWebServer.class.getName()) };
    for (Logger logger : loggers) {
        logger.setLevel(Level.FINE);
        logger.setUseParentHandlers(false);
        logger.addHandler(handler);
    }
}

@Before
public void preTestSetup() throws IOException, URISyntaxException {
    server = new MockWebServer();
    // Start the server.
    server.start();
    // Ask the server for its URL. You'll need this to make HTTP requests.
    URL baseUrl = server.getUrl("");
    port = baseUrl.getPort();
    // Exercise your application code, which should make those HTTP requests.
    // Responses are returned in the same order that they are enqueued.
    fs = new TestableAdlFileSystem();
    conf.setClass(AZURE_AD_TOKEN_PROVIDER_CLASS_KEY, CustomMockTokenProvider.class, AzureADTokenProvider.class);
    URI uri = new URI("adl://localhost:" + port);
    fs.initialize(uri, conf);
}

@Before
public void setUp() throws Exception {
    mockResponse = new MockResponse();
    mockResponse.setBody("TEST RESPONSE".getBytes());
    server = new MockWebServer();
    server.enqueue(mockResponse);
    server.play();
    engine = (OkHttpEngine) new OkHttpEngine.OkHttpFactory().create(new OSciMap4TileSource(server.getUrl("/tiles/vtm").toString()));
}

@Before
public void setUp() throws Exception {
    Robolectric.getFakeHttpLayer().interceptHttpRequests(false);
    Robolectric.getFakeHttpLayer().interceptResponseContent(false);
    webserver = new MockWebServer();
    MockResponse response = new MockResponse();
    InputStream xmlStream = this.getClass().getResourceAsStream("/resources/index.xml");
    response.setBody(xmlStream, 1198l);
    webserver.enqueue(response);
    webserver.play();
}

@Before
public void setup() throws IOException {
    Logger.getLogger(MockWebServer.class.getName()).setLevel(Level.WARNING);
    server = new MockWebServer();
    server.start();
    new MockUp<HttpUrl>() {

        @Mock
        @SuppressWarnings("unused")
        HttpUrl parse(String input) {
            return new HttpUrl.Builder().scheme("http").host(server.getHostName()).port(server.getPort()).encodedPath("/shrink").build();
        }
    };
}

@Test
public void setEndpoint_shouldChangeServiceEndpoint() throws Exception {
    final MockWebServer server = new MockWebServer();
    MockResponse response = new MockResponse();
    server.enqueue(response);
    server.play();
    Pelias pelias = new Pelias(server.getUrl("/").toString());
    pelias.setRequestHandler(new PeliasRequestHandler() {

        @Override
        public Map<String, String> headersForRequest() {
            HashMap<String, String> headers = new HashMap();
            headers.put("TEST_HEADER", "TEST_HEADER_VALUE");
            return headers;
        }

        @Override
        public Map<String, String> queryParamsForRequest() {
            HashMap<String, String> params = new HashMap();
            params.put("TEST_PARAM", "TEST_PARAM_VALUE");
            return params;
        }
    });
    pelias.suggest("test", 1.0, 2.0, callback);
    RecordedRequest request = server.takeRequest();
    assertThat(request.getPath()).contains("/autocomplete");
    assertThat(request.getHeader("TEST_HEADER")).isEqualTo("TEST_HEADER_VALUE");
    assertThat(request.getPath()).contains("TEST_PARAM");
    server.shutdown();
}

@Test
public void testWithRequestCompression() throws IOException {
    AtomicReference<NetworkEventReporter.InspectorRequest> capturedRequest = hookAlmostRealRequestWillBeSent(mMockEventReporter);
    MockWebServer server = new MockWebServer();
    server.start();
    server.enqueue(new MockResponse().setBody("Success!"));
    final byte[] decompressed = "Request text".getBytes();
    final byte[] compressed = compress(decompressed);
    assertNotEquals("Bogus test: decompressed and compressed lengths match", compressed.length, decompressed.length);
    RequestBody compressedBody = RequestBody.create(MediaType.parse("text/plain"), compress(decompressed));
    Request request = new Request.Builder().url(server.getUrl("/")).addHeader("Content-Encoding", "gzip").post(compressedBody).build();
    Response response = mClientWithInterceptor.newCall(request).execute();
    // Force a read to complete the flow.
    response.body().string();
    assertArrayEquals(decompressed, capturedRequest.get().body());
    Mockito.verify(mMockEventReporter).dataSent(anyString(), eq(decompressed.length), eq(compressed.length));
    server.shutdown();
}

protected ContextBuilder builder(Properties overrides) {
    MockWebServer defaultServer = regionToServers.get(DEFAULT_REGION);
    overrides.setProperty(Constants.PROPERTY_MAX_RETRIES, "1");
    overrides.setProperty(ComputeServiceProperties.TIMEOUT_CLEANUP_INCIDENTAL_RESOURCES, "0");
    return ContextBuilder.newBuilder(new AWSEC2ProviderMetadata()).credentials(ACCESS_KEY, SECRET_KEY).endpoint(defaultServer.getUrl("").toString()).overrides(overrides).modules(modules);
}

private void runXmlrpcDiscovery(String testCaseComment, JSONObject testSetup) throws JSONException, IOException {
    final MockWebServer server = new MockWebServer();
    testSetup = new JSONObject(replaceServerMagicName(server, testSetup.toString()));
    final JSONObject input = testSetup.getJSONObject("input");
    setupMockHttpServer(server, input);
    final String inputUrl = input.isNull("siteUrl") ? server.url("").toString() : input.getString("siteUrl");
    final JSONObject output = testSetup.getJSONObject("output");
    final String outputUrl = output.optString("xmlrpcEndpoint", null);
    final JSONObject error = output.optJSONObject("error");
    String xmlrpcUrl = null;
    try {
        xmlrpcUrl = XMLRPCUtils.verifyOrDiscoverXmlRpcUrl(inputUrl, input.optString("username", null), input.optString("username", null));
        // if we reached this point, it means that no error occurred
        assertNull(testCaseMessage("Testcase defines an error but no error occurred!", testCaseComment), error);
    } catch (XMLRPCUtils.XMLRPCUtilsException hce) {
        assertNotNull(testCaseMessage("Error occurred but testcase does not define an error!", testCaseComment), error);
        assertEquals(testCaseMessage("Error message does not match the defined one!", testCaseComment), error.getString("message"), getInstrumentation().getTargetContext().getString(hce.errorMsgId));
    }
    assertEquals(testCaseMessage("XMLRPC URL does not match the defined one!", testCaseComment), outputUrl, xmlrpcUrl);
    server.shutdown();
}

@Test
public void json_schema_validator_supports_matching_uri_json_schema_as_json_node() throws Exception {
    // Given
    String schema = IOUtils.toString(Thread.currentThread().getContextClassLoader().getResourceAsStream("products-schema.json"));
    MockWebServer server = new MockWebServer();
    server.enqueue(new MockResponse().setResponseCode(200).setBody(schema));
    server.play();
    try {
        get("/products").then().assertThat().body(JsonSchemaValidator.matchesJsonSchema(new URI("http://localhost:" + server.getPort())).using(settings().parseUriAndUrlsAsJsonNode(true)));
    } finally {
        server.shutdown();
    }
}

@Test
public void testMoreItems() throws IOException, InterruptedException, ZooniverseClient.RequestMoreItemsException {
    final MockWebServer server = new MockWebServer();
    final String strResponse = getStringFromStream(MoreItemsJsonParserTest.class.getClassLoader().getResourceAsStream("test_more_items_response.json"));
    assertNotNull(strResponse);
    server.enqueue(new MockResponse().setBody(strResponse));
    server.start();
    final ZooniverseClient client = createZooniverseClient(server);
    final int COUNT = 5;
    final List<ZooniverseClient.Subject> subjects = client.requestMoreItemsSync(COUNT);
    assertNotNull(subjects);
    assertTrue(subjects.size() == COUNT);
    final ZooniverseClient.Subject subject = subjects.get(0);
    assertNotNull(subject);
    assertNotNull(subject.getSubjectId());
    assertEquals(subject.getSubjectId(), "5500684569736d5964271400");
    assertNotNull(subject.getZooniverseId());
    assertEquals(subject.getZooniverseId(), "AGZ00081ls");
    assertNotNull(subject.getGroupId());
    assertEquals(subject.getGroupId(), TEST_GROUP_ID);
    assertNotNull(subject.getLocationStandard());
    assertEquals(subject.getLocationStandard(), "http://www.galaxyzoo.org.s3.amazonaws.com/subjects/standard/goods_full_n_27820_standard.jpg");
    assertNotNull(subject.getLocationThumbnail());
    assertEquals(subject.getLocationThumbnail(), "http://www.galaxyzoo.org.s3.amazonaws.com/subjects/thumbnail/goods_full_n_27820_thumbnail.jpg");
    assertNotNull(subject.getLocationInverted());
    assertEquals(subject.getLocationInverted(), "http://www.galaxyzoo.org.s3.amazonaws.com/subjects/inverted/goods_full_n_27820_inverted.jpg");
    //Test what the server received:
    assertEquals(1, server.getRequestCount());
    final RecordedRequest request = server.takeRequest();
    assertEquals("GET", request.getMethod());
    //ZooniverseClient uses one of several possible group IDs at random:
    //See com.murrayc.galaxyzoo.app.provider.Config
    final String possiblePath1 = "/groups/" + TEST_GROUP_ID + "/subjects?limit=5";
    final String possiblePath2 = "/groups/" + Config.SUBJECT_GROUP_ID_DECALS_DR2 + "/subjects?limit=5";
    final String possiblePath3 = "/groups/" + Config.SUBJECT_GROUP_ID_FERENGI2 + "/subjects?limit=5";
    final String possiblePath4 = "/groups/" + Config.SUBJECT_GROUP_ID_SDSS_LOST_SET + "/subjects?limit=5";
    //TODO: Can we use this?
    // assertThat(request.getPath(), anyOf(is(possiblePath1), is(possiblePath2)));
    final String path = request.getPath();
    assertTrue(path.equals(possiblePath1) || path.equals(possiblePath2) || path.equals(possiblePath3) || path.equals(possiblePath4));
    server.shutdown();
}

@Test
public void redirectWithProxySelector() throws Exception {
    final List<URI> proxySelectionRequests = new ArrayList<URI>();
    client.client().setProxySelector(new ProxySelector() {

        @Override
        public List<Proxy> select(URI uri) {
            proxySelectionRequests.add(uri);
            MockWebServer proxyServer = (uri.getPort() == server.getPort()) ? server : server2;
            return Arrays.asList(proxyServer.toProxyAddress());
        }

        @Override
        public void connectFailed(URI uri, SocketAddress address, IOException failure) {
            throw new AssertionError();
        }
    });
    server2.enqueue(new MockResponse().setBody("This is the 2nd server!"));
    server.enqueue(new MockResponse().setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP).addHeader("Location: " + server2.getUrl("/b").toString()).setBody("This page has moved!"));
    assertContent("This is the 2nd server!", client.open(server.getUrl("/a")));
    assertEquals(Arrays.asList(server.getUrl("/a").toURI(), server2.getUrl("/b").toURI()), proxySelectionRequests);
}

public MockWebServer getMockWebServer() {
    return mockWebServer;
}

@Before
public void setUp() throws Exception {
    super.setUp();
    MockitoAnnotations.initMocks(this);
    mMockWebServer = new MockWebServer();
}

@Before
public void setUp() throws Exception {
    mMockWebServer = new MockWebServer();
    mMockWebServer.start();
    RestDataSource.END_POINT = mMockWebServer.getUrl("/").toString();
}

@Before
public void setUp() throws Exception {
    mockWebServer = new MockWebServer();
    mockWebServer.start(9000);
    // this will load fhconfig.local.properties file
    FH.init(getContext(), null);
    FH.setLogLevel(FH.LOG_LEVEL_VERBOSE);
}

@BeforeMethod
public void start() throws IOException {
    server = new MockWebServer();
    server.play();
    api = ContextBuilder.newBuilder("elb").credentials("", MOCK_BEARER_TOKEN).endpoint(url("")).modules(modules).overrides(overrides()).buildApi(ELBApi.class);
}

private void runMockServer() throws IOException {
    server = new MockWebServer();
    server.setDispatcher(dispatcher);
    server.start(0);
}

@BeforeMethod
protected void setup() {
    server = new MockWebServer();
    json = new Json();
}

@BeforeClass
public static void setUpClass() throws Exception {
    mockWebServer = new MockWebServer();
    mockWebServer.start();
}

@BeforeClass
public static void setUpClass() throws Exception {
    mockWebServer = new MockWebServer();
    mockWebServer.start();
}

@Before
public void setUp() throws Exception {
    server = new MockWebServer();
}

@Before
public void setup() throws IOException {
    server = new MockWebServer();
    server.start();
    shapeShift = new ShapeShift();
    shapeShift.baseUrl = server.getUrl("/").toString();
    shapeShift.client.setConnectionSpecs(Collections.singletonList(ConnectionSpec.CLEARTEXT));
}