fix how the onMessage is implemented and updated the mock to reflect the change.

Author: warnerandy

app/scripts/services/angular-atmosphere.js

@@ -24,7 +24,10 @@ angular.module('ngAtmosphere', [])
 		request.fallbackTransport = 'long-polling';
 
 		function handleResponse(response) {
-			var data = angular.fromJson(response.responseBody);
+			var data = response.responseBody;
+			if (typeof data === 'string'){
+				data = angular.fromJson(data);
+			}
 			if (listeners.hasOwnProperty(data.event)) {
 				angular.forEach(listeners[data.event], function (listener) {
 					listener.fn.call(this, data);
@@ -36,9 +39,9 @@ angular.module('ngAtmosphere', [])
 		return {
 			init: function (url) {
 				request.url = url;
+				request.onMessage = handleResponse;
 
 				connection = $.atmosphere.subscribe(request);
-				connection.onMessage = handleResponse;
 			},
 			on: function (event, callbackFn) {
 

test/mock/jquery.atmosphere.mock.js

@@ -1,9 +1,6 @@
 var $ = $ || {};
 $.atmosphere = {
 	AtmosphereRequest: function () {
-
-	},
-	subscribe: function () {
 		return {
 			onError : function(response) {
 			},
@@ -20,9 +17,14 @@ $.atmosphere = {
 			onTransportFailure : function (reason, request) {
 			},
 			onLocalMessage : function (response) {
-			},
+			}
+		}
+	},
+	subscribe: function (request) {
+		var _request = request;
+		return {
 			push: function(data){
-				this.onMessage({responseBody:data});
+				_request.onMessage({responseBody:data});
 			}
 		}
 	}