Add test to verify that JS responses work (marcuswestin#184)

marcuswestin · marcuswestin · commit 5e40fefddd79 · 2016-02-02T10:58:52.000-05:00
diff --git a/Tests/WebViewJavascriptBridgeTests/BridgeTests.m b/Tests/WebViewJavascriptBridgeTests/BridgeTests.m
@@ -121,6 +121,23 @@ - (void)classSpecificTestObjectEncoding:(Class)cls webView:(id)webView {
     loadEchoSample(webView);
 }
 
+- (void)testJavascriptReceiveResponse {
+    [self classSpecificTestJavascriptReceiveResponse:[WebViewJavascriptBridge class] webView:_uiWebView];
+    [self classSpecificTestJavascriptReceiveResponse:[WKWebViewJavascriptBridge class] webView:_wkWebView];
+    [self waitForExpectationsWithTimeout:3 handler:NULL];
+}
+- (void)classSpecificTestJavascriptReceiveResponse:(Class)cls webView:(id)webView {
+    WebViewJavascriptBridge *bridge = [self bridgeForCls:cls webView:webView];
+    loadEchoSample(webView);
+    XCTestExpectation *callbackInvocked = [self expectationWithDescription:@"Callback invoked"];
+    [bridge registerHandler:@"objcEchoToJs" handler:^(id data, WVJBResponseCallback responseCallback) {
+        responseCallback(data);
+    }];
+    [bridge callHandler:@"jsRcvResponseTest" data:nil responseCallback:^(id responseData) {
+        XCTAssertEqualObjects(responseData, @"Response from JS");
+        [callbackInvocked fulfill];
+    }];
+}
 
 - (WebViewJavascriptBridge*)bridgeForCls:(Class)cls webView:(id)webView {
     if (cls == [WebViewJavascriptBridge class]) {
diff --git a/Tests/WebViewJavascriptBridgeTests/echo.html b/Tests/WebViewJavascriptBridgeTests/echo.html
@@ -15,10 +15,19 @@
 	}
 
 	setupWebViewJavascriptBridge(function(bridge) {
-		WebViewJavascriptBridge.callHandler('Greet', 'Hello world');
-		WebViewJavascriptBridge.registerHandler('echoHandler', function(data, responseCallback) {
+		bridge.callHandler('Greet', 'Hello world');
+		bridge.registerHandler('echoHandler', function(data, responseCallback) {
 			responseCallback(data)
 		})
+		bridge.registerHandler('jsRcvResponseTest', function(data, responseCallback) {
+			bridge.callHandler('objcEchoToJs', { foo:'bar' }, function(response) {
+				if (response && response.foo == 'bar') {
+					responseCallback("Response from JS")
+				} else {
+					responseCallback("Failed")
+				}
+			})
+		})
 	})
 	</script>
 </body></html>
