模拟 Websocket 服务器进行单元测试

Question

我在客户端使用 sockJS 来实现我的项目的 websockets（套接字服务器由 Java Spring 实例运行）。我无法弄清楚如何为我的前端套接字单元测试模拟 websockets 服务器。

我有一些函数依赖于从服务器流式传输的数据。但我不知道如何模拟这些数据以调用 sock.onmessage。

我的客户端一般是这样设置的：

websocket-client.js (sockJS)

sock = new SockJS('/server/uri');
    
sock.onopen = function() {

};
    
sock.onmessage = function(e) {
  var message = e.data.message;
  displayMessage(message);
};
    
sock.onclose = function() {
  // clean up stuff
};

我将 Karma 和 Chai 用于我的测试运行程序/断言库。

由于这是一个 Angular 应用程序，通常我可以使用 $httpBackend 模拟我的 API 服务。但在这种情况下，我认为这不是一个选择。

Answer 1

最简单的方法是将 SockJS 包装在工厂/服务中：

angular
    .module('app')
    .factory('sockJS', function () {
        return {
            create: function (uri) {
                return new SockJS(uri);
            }
        };
    });

然后你可以注入 sockJS 并模拟它。

angular
    .module('app')
    .controller('MainCtrl', function($scope, sockJS) {
        var sock = sockJS.create('/server/uri');

        sock.onopen = function() {

        };

        sock.onmessage = function(e) {
        // do stuff with e data
        };

        sock.onclose = function() {
        // clean up stuff
        };
    });