How to use socket method in autotest

Best Python code snippet using autotest_python

socket_test.js

Source:socket_test.js Github

copy

Full Screen

1import assert from "assert"2import jsdom from "jsdom"3import sinon from "sinon"4import {WebSocket, Server as WebSocketServer} from "mock-socket"5import {encode, decode} from "./serializer"6import {Socket, LongPoll} from "../js/phoenix"7let socket8describe("constructor", () => {9 before(() => {10 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()11 })12 after(() => {13 window.XMLHttpRequest = null14 })15 it("sets defaults", () => {16 socket = new Socket("/socket")17 assert.equal(socket.channels.length, 0)18 assert.equal(socket.sendBuffer.length, 0)19 assert.equal(socket.ref, 0)20 assert.equal(socket.endPoint, "/socket/websocket")21 assert.deepEqual(socket.stateChangeCallbacks, {open: [], close: [], error: [], message: []})22 assert.equal(socket.transport, LongPoll)23 assert.equal(socket.timeout, 10000)24 assert.equal(socket.longpollerTimeout, 20000)25 assert.equal(socket.heartbeatIntervalMs, 30000)26 assert.equal(typeof socket.logger, "function")27 assert.equal(typeof socket.reconnectAfterMs, "function")28 })29 it("overrides some defaults with options", () => {30 const customTransport = function transport() {}31 const customLogger = function logger() {}32 const customReconnect = function reconnect() {}33 socket = new Socket("/socket", {34 timeout: 40000,35 longpollerTimeout: 50000,36 heartbeatIntervalMs: 60000,37 transport: customTransport,38 logger: customLogger,39 reconnectAfterMs: customReconnect,40 params: {one: "two"},41 })42 assert.equal(socket.timeout, 40000)43 assert.equal(socket.longpollerTimeout, 50000)44 assert.equal(socket.heartbeatIntervalMs, 60000)45 assert.equal(socket.transport, customTransport)46 assert.equal(socket.logger, customLogger)47 assert.equal(socket.reconnectAfterMs, customReconnect)48 assert.deepEqual(socket.params, {one: "two"})49 })50 describe("with Websocket", () => {51 let mockServer52 before(() => {53 mockServer = new WebSocketServer("wss://example.com/")54 })55 after((done) => {56 mockServer.stop(() => {57 window.WebSocket = null58 done()59 })60 })61 it("defaults to Websocket transport if available", () => {62 socket = new Socket("/socket")63 assert.equal(socket.transport, WebSocket)64 })65 })66})67describe("protocol", () => {68 beforeEach(() => {69 socket = new Socket("/socket")70 })71 it("returns wss when location.protocol is https", () => {72 jsdom.changeURL(window, "https://example.com/");73 assert.equal(socket.protocol(), "wss")74 })75 it("returns ws when location.protocol is http", () => {76 jsdom.changeURL(window, "http://example.com/");77 assert.equal(socket.protocol(), "ws")78 })79})80describe("endpointURL", () => {81 it("returns endpoint for given full url", () => {82 jsdom.changeURL(window, "https://example.com/");83 socket = new Socket("wss://example.org/chat")84 assert.equal(socket.endPointURL(), "wss://example.org/chat/websocket?vsn=2.0.0")85 })86 it("returns endpoint for given protocol-relative url", () => {87 jsdom.changeURL(window, "https://example.com/");88 socket = new Socket("//example.org/chat")89 assert.equal(socket.endPointURL(), "wss://example.org/chat/websocket?vsn=2.0.0")90 })91 it("returns endpoint for given path on https host", () => {92 jsdom.changeURL(window, "https://example.com/");93 socket = new Socket("/socket")94 assert.equal(socket.endPointURL(), "wss://example.com/socket/websocket?vsn=2.0.0")95 })96 it("returns endpoint for given path on http host", () => {97 jsdom.changeURL(window, "http://example.com/");98 socket = new Socket("/socket")99 assert.equal(socket.endPointURL(), "ws://example.com/socket/websocket?vsn=2.0.0")100 })101})102describe("connect with WebSocket", () => {103 let mockServer104 before(() => {105 mockServer = new WebSocketServer("wss://example.com/")106 })107 after((done) => {108 mockServer.stop(() => {109 window.WebSocket = null110 done()111 })112 })113 beforeEach(() => {114 socket = new Socket("/socket")115 })116 it("establishes websocket connection with endpoint", () => {117 socket.connect()118 let conn = socket.conn119 assert.ok(conn instanceof WebSocket)120 assert.equal(conn.url, socket.endPointURL())121 })122 it("sets callbacks for connection", () => {123 let opens = 0124 socket.onOpen(() => ++opens)125 let closes = 0126 socket.onClose(() => ++closes)127 let lastError128 socket.onError((error) => lastError = error)129 let lastMessage130 socket.onMessage((message) => lastMessage = message.payload)131 socket.connect()132 socket.conn.onopen[0]()133 assert.equal(opens, 1)134 socket.conn.onclose[0]()135 assert.equal(closes, 1)136 socket.conn.onerror[0]("error")137 assert.equal(lastError, "error")138 const data = {"topic":"topic","event":"event","payload":"payload","status":"ok"}139 socket.conn.onmessage[0]({data: encode(data)})140 assert.equal(lastMessage, "payload")141 })142 it("is idempotent", () => {143 socket.connect()144 let conn = socket.conn145 socket.connect()146 assert.deepStrictEqual(conn, socket.conn)147 })148})149describe("connect with long poll", () => {150 before(() => {151 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()152 })153 after(() => {154 window.XMLHttpRequest = null155 })156 beforeEach(() => {157 socket = new Socket("/socket")158 })159 it("establishes long poll connection with endpoint", () => {160 socket.connect()161 let conn = socket.conn162 assert.ok(conn instanceof LongPoll)163 assert.equal(conn.pollEndpoint, "http://example.com/socket/longpoll?vsn=2.0.0")164 assert.equal(conn.timeout, 20000)165 })166 it("sets callbacks for connection", () => {167 let opens = 0168 socket.onOpen(() => ++opens)169 let closes = 0170 socket.onClose(() => ++closes)171 let lastError172 socket.onError((error) => lastError = error)173 let lastMessage174 socket.onMessage((message) => lastMessage = message.payload)175 socket.connect()176 socket.conn.onopen()177 assert.equal(opens, 1)178 socket.conn.onclose()179 assert.equal(closes, 1)180 socket.conn.onerror("error")181 assert.equal(lastError, "error")182 socket.connect()183 const data = {"topic":"topic","event":"event","payload":"payload","status":"ok"}184 socket.conn.onmessage({data: encode(data)})185 assert.equal(lastMessage, "payload")186 })187 it("is idempotent", () => {188 socket.connect()189 let conn = socket.conn190 socket.connect()191 assert.deepStrictEqual(conn, socket.conn)192 })193})194describe("disconnect", () => {195 let mockServer196 before(() => {197 mockServer = new WebSocketServer('wss://example.com/')198 })199 after((done) => {200 mockServer.stop(() => {201 window.WebSocket = null202 done()203 })204 })205 beforeEach(() => {206 socket = new Socket("/socket")207 })208 it("removes existing connection", () => {209 socket.connect()210 socket.disconnect()211 assert.equal(socket.conn, null)212 })213 it("calls callback", () => {214 let count = 0215 socket.connect()216 socket.disconnect(() => count++)217 assert.equal(count, 1)218 })219 it("calls connection close callback", () => {220 socket.connect()221 const spy = sinon.spy(socket.conn, "close")222 socket.disconnect(null, "code", "reason")223 assert(spy.calledWith("code", "reason"))224 })225 it("does not throw when no connection", () => {226 assert.doesNotThrow(() => {227 socket.disconnect()228 })229 })230})231describe("connectionState", () => {232 before(() => {233 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()234 })235 after(() => {236 window.XMLHttpRequest = null237 })238 beforeEach(() => {239 socket = new Socket("/socket")240 })241 it("defaults to closed", () => {242 assert.equal(socket.connectionState(), "closed")243 })244 it("returns closed if readyState unrecognized", () => {245 socket.connect()246 socket.conn.readyState = 5678247 assert.equal(socket.connectionState(), "closed")248 })249 it("returns connecting", () => {250 socket.connect()251 socket.conn.readyState = 0252 assert.equal(socket.connectionState(), "connecting")253 assert.ok(!socket.isConnected(), "is not connected")254 })255 it("returns open", () => {256 socket.connect()257 socket.conn.readyState = 1258 assert.equal(socket.connectionState(), "open")259 assert.ok(socket.isConnected(), "is connected")260 })261 it("returns closing", () => {262 socket.connect()263 socket.conn.readyState = 2264 assert.equal(socket.connectionState(), "closing")265 assert.ok(!socket.isConnected(), "is not connected")266 })267 it("returns closed", () => {268 socket.connect()269 socket.conn.readyState = 3270 assert.equal(socket.connectionState(), "closed")271 assert.ok(!socket.isConnected(), "is not connected")272 })273})274describe("channel", () => {275 let channel276 beforeEach(() => {277 socket = new Socket("/socket")278 })279 it("returns channel with given topic and params", () => {280 channel = socket.channel("topic", {one: "two"})281 assert.deepStrictEqual(channel.socket, socket)282 assert.equal(channel.topic, "topic")283 assert.deepEqual(channel.params, {one: "two"})284 })285 it("adds channel to sockets channels list", () => {286 assert.equal(socket.channels.length, 0)287 channel = socket.channel("topic", {one: "two"})288 assert.equal(socket.channels.length, 1)289 const [foundChannel] = socket.channels290 assert.deepStrictEqual(foundChannel, channel)291 })292})293describe("remove", () => {294 beforeEach(() => {295 socket = new Socket("/socket")296 })297 it("removes given channel from channels", () => {298 const channel1 = socket.channel("topic-1")299 const channel2 = socket.channel("topic-2")300 sinon.stub(channel1, "joinRef").returns(1)301 sinon.stub(channel2, "joinRef").returns(2)302 socket.remove(channel1)303 assert.equal(socket.channels.length, 1)304 const [foundChannel] = socket.channels305 assert.deepStrictEqual(foundChannel, channel2)306 })307})308describe("push", () => {309 const data = {topic: "topic", event: "event", payload: "payload", ref: "ref"}310 const json = encode(data)311 before(() => {312 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()313 })314 after(() => {315 window.XMLHttpRequest = null316 })317 beforeEach(() => {318 socket = new Socket("/socket")319 })320 it("sends data to connection when connected", () => {321 socket.connect()322 socket.conn.readyState = 1 // open323 const spy = sinon.spy(socket.conn, "send")324 socket.push(data)325 assert.ok(spy.calledWith(json))326 })327 it("buffers data when not connected", () => {328 socket.connect()329 socket.conn.readyState = 0 // connecting330 const spy = sinon.spy(socket.conn, "send")331 assert.equal(socket.sendBuffer.length, 0)332 socket.push(data)333 assert.ok(spy.neverCalledWith(json))334 assert.equal(socket.sendBuffer.length, 1)335 const [callback] = socket.sendBuffer336 callback()337 assert.ok(spy.calledWith(json))338 })339})340describe("makeRef", () => {341 beforeEach(() => {342 socket = new Socket("/socket")343 })344 it("returns next message ref", () => {345 assert.strictEqual(socket.ref, 0)346 assert.strictEqual(socket.makeRef(), "1")347 assert.strictEqual(socket.ref, 1)348 assert.strictEqual(socket.makeRef(), "2")349 assert.strictEqual(socket.ref, 2)350 })351 it("restarts for overflow", () => {352 socket.ref = Number.MAX_SAFE_INTEGER + 1353 assert.strictEqual(socket.makeRef(), "0")354 assert.strictEqual(socket.ref, 0)355 })356})357describe("sendHeartbeat", () => {358 before(() => {359 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()360 })361 after(() => {362 window.XMLHttpRequest = null363 })364 beforeEach(() => {365 socket = new Socket("/socket")366 socket.connect()367 })368 it("closes socket when heartbeat is not ack'd within heartbeat window", () => {369 let closed = false370 socket.conn.readyState = 1 // open371 socket.conn.onclose = () => closed = true372 socket.sendHeartbeat()373 assert.equal(closed, false)374 socket.sendHeartbeat()375 assert.equal(closed, true)376 })377 it("pushes heartbeat data when connected", () => {378 socket.conn.readyState = 1 // open379 const spy = sinon.spy(socket.conn, "send")380 const data = "[null,\"1\",\"phoenix\",\"heartbeat\",{}]"381 socket.sendHeartbeat()382 assert.ok(spy.calledWith(data))383 })384 it("no ops when not connected", () => {385 socket.conn.readyState = 0 // connecting386 const spy = sinon.spy(socket.conn, "send")387 const data = encode({topic: "phoenix", event: "heartbeat", payload: {},ref: "1"})388 socket.sendHeartbeat()389 assert.ok(spy.neverCalledWith(data))390 })391})392describe("flushSendBuffer", () => {393 before(() => {394 window.XMLHttpRequest = sinon.useFakeXMLHttpRequest()395 })396 after(() => {397 window.XMLHttpRequest = null398 })399 beforeEach(() => {400 socket = new Socket("/socket")401 socket.connect()402 })403 it("calls callbacks in buffer when connected", () => {404 socket.conn.readyState = 1 // open405 const spy1 = sinon.spy()406 const spy2 = sinon.spy()407 const spy3 = sinon.spy()408 socket.sendBuffer.push(spy1)409 socket.sendBuffer.push(spy2)410 socket.flushSendBuffer()411 assert.ok(spy1.calledOnce)412 assert.ok(spy2.calledOnce)413 assert.equal(spy3.callCount, 0)414 })415 it("empties sendBuffer", () => {416 socket.conn.readyState = 1 // open417 socket.sendBuffer.push(() => {})418 socket.flushSendBuffer()419 assert.deepEqual(socket.sendBuffer.length, 0)420 })421})422describe("onConnOpen", () => {423 let mockServer424 before(() => {425 mockServer = new WebSocketServer('wss://example.com/')426 })427 after((done) => {428 mockServer.stop(() => {429 window.WebSocket = null430 done()431 })432 })433 beforeEach(() => {434 socket = new Socket("/socket", {435 reconnectAfterMs: () => 100000436 })437 socket.connect()438 })439 it("flushes the send buffer", () => {440 socket.conn.readyState = 1 // open441 const spy = sinon.spy()442 socket.sendBuffer.push(spy)443 socket.onConnOpen()444 assert.ok(spy.calledOnce)445 })446 it("resets reconnectTimer", () => {447 const spy = sinon.spy(socket.reconnectTimer, "reset")448 socket.onConnOpen()449 assert.ok(spy.calledOnce)450 })451 it("triggers onOpen callback", () => {452 const spy = sinon.spy()453 socket.onOpen(spy)454 socket.onConnOpen()455 assert.ok(spy.calledOnce)456 })457})458describe("onConnClose", () => {459 let mockServer460 before(() => {461 mockServer = new WebSocketServer('wss://example.com/')462 })463 after((done) => {464 mockServer.stop(() => {465 window.WebSocket = null466 done()467 })468 })469 beforeEach(() => {470 socket = new Socket("/socket", {471 reconnectAfterMs: () => 100000472 })473 socket.connect()474 })475 it("schedules reconnectTimer timeout", () => {476 const spy = sinon.spy(socket.reconnectTimer, "scheduleTimeout")477 socket.onConnClose()478 assert.ok(spy.calledOnce)479 })480 it("triggers onClose callback", () => {481 const spy = sinon.spy()482 socket.onClose(spy)483 socket.onConnClose("event")484 assert.ok(spy.calledWith("event"))485 })486 it("triggers channel error", () => {487 const channel = socket.channel("topic")488 const spy = sinon.spy(channel, "trigger")489 socket.onConnClose()490 assert.ok(spy.calledWith("phx_error"))491 })492})493describe("onConnError", () => {494 let mockServer495 before(() => {496 mockServer = new WebSocketServer('wss://example.com/')497 })498 after((done) => {499 mockServer.stop(() => {500 window.WebSocket = null501 done()502 })503 })504 beforeEach(() => {505 socket = new Socket("/socket", {506 reconnectAfterMs: () => 100000507 })508 socket.connect()509 })510 it("triggers onClose callback", () => {511 const spy = sinon.spy()512 socket.onError(spy)513 socket.onConnError("error")514 assert.ok(spy.calledWith("error"))515 })516 it("triggers channel error", () => {517 const channel = socket.channel("topic")518 const spy = sinon.spy(channel, "trigger")519 socket.onConnError("error")520 assert.ok(spy.calledWith("phx_error"))521 })522})523describe("onConnMessage", () => {524 let mockServer525 before(() => {526 mockServer = new WebSocketServer('wss://example.com/')527 })528 after((done) => {529 mockServer.stop(() => {530 window.WebSocket = null531 done()532 })533 })534 beforeEach(() => {535 socket = new Socket("/socket", {536 reconnectAfterMs: () => 100000537 })538 socket.connect()539 })540 it("parses raw message and triggers channel event", () => {541 const message = encode({topic: "topic", event: "event", payload: "payload", ref: "ref"})542 const data = {data: message}543 const targetChannel = socket.channel("topic")544 const otherChannel = socket.channel("off-topic")545 const targetSpy = sinon.spy(targetChannel, "trigger")546 const otherSpy = sinon.spy(otherChannel, "trigger")547 socket.onConnMessage(data)548 assert.ok(targetSpy.calledWith("event", "payload", "ref"))549 assert.equal(targetSpy.callCount, 1)550 assert.equal(otherSpy.callCount, 0)551 })552 it("triggers onMessage callback", () => {553 const message = {"topic":"topic","event":"event","payload":"payload","ref":"ref"}554 const spy = sinon.spy()555 socket.onMessage(spy)556 socket.onConnMessage({data: encode(message)})557 assert.ok(spy.calledWith({558 "topic": "topic",559 "event": "event",560 "payload": "payload",561 "ref": "ref",562 "join_ref": null563 }))564 })565})566describe("custom encoder and decoder", () => {567 it("encodes to JSON array by default", () => {568 socket = new Socket("/socket")569 let payload = {topic: "topic", ref: "2", join_ref: "1", event: "join", payload: {foo: "bar"}}570 socket.encode(payload, encoded => {571 assert.deepStrictEqual(encoded, '["1","2","topic","join",{"foo":"bar"}]')572 })573 })574 it("allows custom encoding when using WebSocket transport", () => {575 let encoder = (payload, callback) => callback("encode works")576 socket = new Socket("/socket", {transport: WebSocket, encode: encoder})577 socket.encode({foo: "bar"}, encoded => {578 assert.deepStrictEqual(encoded, "encode works")579 })580 })581 it("forces JSON encoding when using LongPoll transport", () => {582 let encoder = (payload, callback) => callback("encode works")583 socket = new Socket("/socket", {transport: LongPoll, encode: encoder})584 let payload = {topic: "topic", ref: "2", join_ref: "1", event: "join", payload: {foo: "bar"}}585 socket.encode(payload, encoded => {586 assert.deepStrictEqual(encoded, '["1","2","topic","join",{"foo":"bar"}]')587 })588 })589 it("decodes JSON by default", () => {590 socket = new Socket("/socket")591 let encoded = '["1","2","topic","join",{"foo":"bar"}]'592 socket.decode(encoded, decoded => {593 assert.deepStrictEqual(decoded, {topic: "topic", ref: "2", join_ref: "1", event: "join", payload: {foo: "bar"}})594 })595 })596 it("allows custom decoding when using WebSocket transport", () => {597 let decoder = (payload, callback) => callback("decode works")598 socket = new Socket("/socket", {transport: WebSocket, decode: decoder})599 socket.decode("...esoteric format...", decoded => {600 assert.deepStrictEqual(decoded, "decode works")601 })602 })603 it("forces JSON decoding when using LongPoll transport", () => {604 let decoder = (payload, callback) => callback("decode works")605 socket = new Socket("/socket", {transport: LongPoll, decode: decoder})606 let payload = {topic: "topic", ref: "2", join_ref: "1", event: "join", payload: {foo: "bar"}}607 socket.decode('["1","2","topic","join",{"foo":"bar"}]', decoded => {608 assert.deepStrictEqual(decoded, payload)609 })610 })...

Full Screen

Full Screen

connection.js

Source:connection.js Github

copy

Full Screen

...81 socket.emit('getUtf8');82 });83 it('should connect to a namespace after connection established', function (done) {84 var manager = io.Manager();85 var socket = manager.socket('/');86 socket.on('connect', function () {87 var foo = manager.socket('/foo');88 foo.on('connect', function () {89 foo.close();90 socket.close();91 manager.close();92 done();93 });94 });95 });96 it('should open a new namespace after connection gets closed', function (done) {97 var manager = io.Manager();98 var socket = manager.socket('/');99 socket.on('connect', function () {100 socket.disconnect();101 }).on('disconnect', function () {102 var foo = manager.socket('/foo');103 foo.on('connect', function () {104 foo.disconnect();105 manager.close();106 done();107 });108 });109 });110 it('should reconnect by default', function (done) {111 var socket = io({ forceNew: true });112 socket.io.on('reconnect', function () {113 socket.disconnect();114 done();115 });116 setTimeout(function () {117 socket.io.engine.close();118 }, 500);119 });120 it('should reconnect manually', function (done) {121 var socket = io({ forceNew: true });122 socket.once('connect', function () {123 socket.disconnect();124 }).once('disconnect', function () {125 socket.once('connect', function () {126 socket.disconnect();127 done();128 });129 socket.connect();130 });131 });132 it('should reconnect automatically after reconnecting manually', function (done) {133 var socket = io({ forceNew: true });134 socket.once('connect', function () {135 socket.disconnect();136 }).once('disconnect', function () {137 socket.on('reconnect', function () {138 socket.disconnect();139 done();140 });141 socket.connect();142 setTimeout(function () {143 socket.io.engine.close();144 }, 500);145 });146 });147 it('should attempt reconnects after a failed reconnect', function (done) {148 var manager = io.Manager({ reconnection: true, timeout: 0, reconnectionAttempts: 2, reconnectionDelay: 10 });149 var socket = manager.socket('/timeout');150 socket.once('reconnect_failed', function () {151 var reconnects = 0;152 var reconnectCb = function () {153 reconnects++;154 };155 manager.on('reconnect_attempt', reconnectCb);156 manager.on('reconnect_failed', function failed () {157 expect(reconnects).to.be(2);158 socket.close();159 manager.close();160 done();161 });162 socket.connect();163 });164 });165 it('reconnect delay should increase every time', function (done) {166 var manager = io.Manager({ reconnection: true, timeout: 0, reconnectionAttempts: 3, reconnectionDelay: 100, randomizationFactor: 0.2 });167 var socket = manager.socket('/timeout');168 var reconnects = 0;169 var increasingDelay = true;170 var startTime;171 var prevDelay = 0;172 socket.on('connect_error', function () {173 startTime = new Date().getTime();174 });175 socket.on('reconnect_attempt', function () {176 reconnects++;177 var currentTime = new Date().getTime();178 var delay = currentTime - startTime;179 if (delay <= prevDelay) {180 increasingDelay = false;181 }182 prevDelay = delay;183 });184 socket.on('reconnect_failed', function failed () {185 expect(reconnects).to.be(3);186 expect(increasingDelay).to.be.ok();187 socket.close();188 manager.close();189 done();190 });191 });192 it('reconnect event should fire in socket', function (done) {193 var socket = io({ forceNew: true });194 socket.on('reconnect', function () {195 socket.disconnect();196 done();197 });198 setTimeout(function () {199 socket.io.engine.close();200 }, 500);201 });202 it('should not reconnect when force closed', function (done) {203 var socket = io('/invalid', { forceNew: true, timeout: 0, reconnectionDelay: 10 });204 socket.on('connect_error', function () {205 socket.on('reconnect_attempt', function () {206 expect().fail();207 });208 socket.disconnect();209 // set a timeout to let reconnection possibly fire210 setTimeout(function () {211 done();212 }, 500);213 });214 });215 it('should stop reconnecting when force closed', function (done) {216 var socket = io('/invalid', { forceNew: true, timeout: 0, reconnectionDelay: 10 });217 socket.once('reconnect_attempt', function () {218 socket.on('reconnect_attempt', function () {219 expect().fail();220 });221 socket.disconnect();222 // set a timeout to let reconnection possibly fire223 setTimeout(function () {224 done();225 }, 500);226 });227 });228 it('should reconnect after stopping reconnection', function (done) {229 var socket = io('/invalid', { forceNew: true, timeout: 0, reconnectionDelay: 10 });230 socket.once('reconnect_attempt', function () {231 socket.on('reconnect_attempt', function () {232 socket.disconnect();233 done();234 });235 socket.disconnect();236 socket.connect();237 });238 });239 it('should stop reconnecting on a socket and keep to reconnect on another', function (done) {240 var manager = io.Manager();241 var socket1 = manager.socket('/');242 var socket2 = manager.socket('/asd');243 manager.on('reconnect_attempt', function () {244 socket1.on('connect', function () {245 expect().fail();246 });247 socket2.on('connect', function () {248 setTimeout(function () {249 socket2.disconnect();250 manager.disconnect();251 done();252 }, 500);253 });254 socket1.disconnect();255 });256 setTimeout(function () {257 manager.engine.close();258 }, 1000);259 });260 it('should try to reconnect twice and fail when requested two attempts with immediate timeout and reconnect enabled', function (done) {261 var manager = io.Manager({ reconnection: true, timeout: 0, reconnectionAttempts: 2, reconnectionDelay: 10 });262 var socket;263 var reconnects = 0;264 var reconnectCb = function () {265 reconnects++;266 };267 manager.on('reconnect_attempt', reconnectCb);268 manager.on('reconnect_failed', function failed () {269 expect(reconnects).to.be(2);270 socket.close();271 manager.close();272 done();273 });274 socket = manager.socket('/timeout');275 });276 it('should fire reconnect_* events on socket', function (done) {277 var manager = io.Manager({ reconnection: true, timeout: 0, reconnectionAttempts: 2, reconnectionDelay: 10 });278 var socket = manager.socket('/timeout_socket');279 var reconnects = 0;280 var reconnectCb = function (attempts) {281 reconnects++;282 expect(attempts).to.be(reconnects);283 };284 socket.on('reconnect_attempt', reconnectCb);285 socket.on('reconnect_failed', function failed () {286 expect(reconnects).to.be(2);287 socket.close();288 manager.close();289 done();290 });291 });292 it('should fire error on socket', function (done) {293 var manager = io.Manager({ reconnection: true });294 var socket = manager.socket('/timeout_socket');295 socket.on('error', function (data) {296 expect(data.code).to.be('test');297 socket.close();298 manager.close();299 done();300 });301 socket.on('connect', function () {302 manager.engine.onPacket({ type: 'error', data: 'test' });303 });304 });305 it('should fire reconnecting (on socket) with attempts number when reconnecting twice', function (done) {306 var manager = io.Manager({ reconnection: true, timeout: 0, reconnectionAttempts: 2, reconnectionDelay: 10 });307 var socket = manager.socket('/timeout_socket');308 var reconnects = 0;309 var reconnectCb = function (attempts) {310 reconnects++;311 expect(attempts).to.be(reconnects);312 };313 socket.on('reconnecting', reconnectCb);314 socket.on('reconnect_failed', function failed () {315 expect(reconnects).to.be(2);316 socket.close();317 manager.close();318 done();319 });320 });321 it('should not try to reconnect and should form a connection when connecting to correct port with default timeout', function (done) {322 var manager = io.Manager({ reconnection: true, reconnectionDelay: 10 });323 var cb = function () {324 socket.close();325 expect().fail();326 };327 manager.on('reconnect_attempt', cb);328 var socket = manager.socket('/valid');329 socket.on('connect', function () {330 // set a timeout to let reconnection possibly fire331 setTimeout(function () {332 socket.close();333 manager.close();334 done();335 }, 1000);336 });337 });338 it('should connect while disconnecting another socket', function (done) {339 var manager = io.Manager();340 var socket1 = manager.socket('/foo');341 socket1.on('connect', function () {342 var socket2 = manager.socket('/asd');343 socket2.on('connect', done);344 socket1.disconnect();345 });346 });347 // Ignore incorrect connection test for old IE due to no support for348 // `script.onerror` (see: http://requirejs.org/docs/api.html#ieloadfail)349 if (!global.document || hasCORS) {350 it('should try to reconnect twice and fail when requested two attempts with incorrect address and reconnect enabled', function (done) {351 var manager = io.Manager('http://localhost:3940', { reconnection: true, reconnectionAttempts: 2, reconnectionDelay: 10 });352 var socket = manager.socket('/asd');353 var reconnects = 0;354 var cb = function () {355 reconnects++;356 };357 manager.on('reconnect_attempt', cb);358 manager.on('reconnect_failed', function () {359 expect(reconnects).to.be(2);360 socket.disconnect();361 manager.close();362 done();363 });364 });365 it('should not try to reconnect with incorrect port when reconnection disabled', function (done) {366 var manager = io.Manager('http://localhost:9823', { reconnection: false });367 var cb = function () {368 socket.close();369 expect().fail();370 };371 manager.on('reconnect_attempt', cb);372 manager.on('connect_error', function () {373 // set a timeout to let reconnection possibly fire374 setTimeout(function () {375 socket.disconnect();376 manager.close();377 done();378 }, 1000);379 });380 var socket = manager.socket('/invalid');381 });382 }383 it('should emit date as string', function (done) {384 var socket = io({ forceNew: true });385 socket.on('takeDate', function (data) {386 socket.close();387 expect(data).to.be.a('string');388 done();389 });390 socket.emit('getDate');391 });392 it('should emit date in object', function (done) {393 var socket = io({ forceNew: true });394 socket.on('takeDateObj', function (data) {...

Full Screen

Full Screen

agent.js

Source:agent.js Github

copy

Full Screen

1'use strict';2const OriginalAgent = require('http').Agent;3const ms = require('humanize-ms');4const debug = require('debug')('agentkeepalive');5const deprecate = require('depd')('agentkeepalive');6const {7 INIT_SOCKET,8 CURRENT_ID,9 CREATE_ID,10 SOCKET_CREATED_TIME,11 SOCKET_NAME,12 SOCKET_REQUEST_COUNT,13 SOCKET_REQUEST_FINISHED_COUNT,14} = require('./constants');15// OriginalAgent come from16// - https://github.com/nodejs/node/blob/v8.12.0/lib/_http_agent.js17// - https://github.com/nodejs/node/blob/v10.12.0/lib/_http_agent.js18// node <= 1019let defaultTimeoutListenerCount = 1;20const majorVersion = parseInt(process.version.split('.', 1)[0].substring(1));21if (majorVersion >= 11 && majorVersion <= 12) {22 defaultTimeoutListenerCount = 2;23} else if (majorVersion >= 13) {24 defaultTimeoutListenerCount = 3;25}26class Agent extends OriginalAgent {27 constructor(options) {28 options = options || {};29 options.keepAlive = options.keepAlive !== false;30 // default is keep-alive and 4s free socket timeout31 // see https://medium.com/ssense-tech/reduce-networking-errors-in-nodejs-23b4eb9f2d8332 if (options.freeSocketTimeout === undefined) {33 options.freeSocketTimeout = 4000;34 }35 // Legacy API: keepAliveTimeout should be rename to `freeSocketTimeout`36 if (options.keepAliveTimeout) {37 deprecate('options.keepAliveTimeout is deprecated, please use options.freeSocketTimeout instead');38 options.freeSocketTimeout = options.keepAliveTimeout;39 delete options.keepAliveTimeout;40 }41 // Legacy API: freeSocketKeepAliveTimeout should be rename to `freeSocketTimeout`42 if (options.freeSocketKeepAliveTimeout) {43 deprecate('options.freeSocketKeepAliveTimeout is deprecated, please use options.freeSocketTimeout instead');44 options.freeSocketTimeout = options.freeSocketKeepAliveTimeout;45 delete options.freeSocketKeepAliveTimeout;46 }47 // Sets the socket to timeout after timeout milliseconds of inactivity on the socket.48 // By default is double free socket timeout.49 if (options.timeout === undefined) {50 // make sure socket default inactivity timeout >= 8s51 options.timeout = Math.max(options.freeSocketTimeout * 2, 8000);52 }53 // support humanize format54 options.timeout = ms(options.timeout);55 options.freeSocketTimeout = ms(options.freeSocketTimeout);56 options.socketActiveTTL = options.socketActiveTTL ? ms(options.socketActiveTTL) : 0;57 super(options);58 this[CURRENT_ID] = 0;59 // create socket success counter60 this.createSocketCount = 0;61 this.createSocketCountLastCheck = 0;62 this.createSocketErrorCount = 0;63 this.createSocketErrorCountLastCheck = 0;64 this.closeSocketCount = 0;65 this.closeSocketCountLastCheck = 0;66 // socket error event count67 this.errorSocketCount = 0;68 this.errorSocketCountLastCheck = 0;69 // request finished counter70 this.requestCount = 0;71 this.requestCountLastCheck = 0;72 // including free socket timeout counter73 this.timeoutSocketCount = 0;74 this.timeoutSocketCountLastCheck = 0;75 this.on('free', socket => {76 // https://github.com/nodejs/node/pull/3200077 // Node.js native agent will check socket timeout eqs agent.options.timeout.78 // Use the ttl or freeSocketTimeout to overwrite.79 const timeout = this.calcSocketTimeout(socket);80 if (timeout > 0 && socket.timeout !== timeout) {81 socket.setTimeout(timeout);82 }83 });84 }85 get freeSocketKeepAliveTimeout() {86 deprecate('agent.freeSocketKeepAliveTimeout is deprecated, please use agent.options.freeSocketTimeout instead');87 return this.options.freeSocketTimeout;88 }89 get timeout() {90 deprecate('agent.timeout is deprecated, please use agent.options.timeout instead');91 return this.options.timeout;92 }93 get socketActiveTTL() {94 deprecate('agent.socketActiveTTL is deprecated, please use agent.options.socketActiveTTL instead');95 return this.options.socketActiveTTL;96 }97 calcSocketTimeout(socket) {98 /**99 * return <= 0: should free socket100 * return > 0: should update socket timeout101 * return undefined: not find custom timeout102 */103 let freeSocketTimeout = this.options.freeSocketTimeout;104 const socketActiveTTL = this.options.socketActiveTTL;105 if (socketActiveTTL) {106 // check socketActiveTTL107 const aliveTime = Date.now() - socket[SOCKET_CREATED_TIME];108 const diff = socketActiveTTL - aliveTime;109 if (diff <= 0) {110 return diff;111 }112 if (freeSocketTimeout && diff < freeSocketTimeout) {113 freeSocketTimeout = diff;114 }115 }116 // set freeSocketTimeout117 if (freeSocketTimeout) {118 // set free keepalive timer119 // try to use socket custom freeSocketTimeout first, support headers['keep-alive']120 // https://github.com/node-modules/urllib/blob/b76053020923f4d99a1c93cf2e16e0c5ba10bacf/lib/urllib.js#L498121 const customFreeSocketTimeout = socket.freeSocketTimeout || socket.freeSocketKeepAliveTimeout;122 return customFreeSocketTimeout || freeSocketTimeout;123 }124 }125 keepSocketAlive(socket) {126 const result = super.keepSocketAlive(socket);127 // should not keepAlive, do nothing128 if (!result) return result;129 const customTimeout = this.calcSocketTimeout(socket);130 if (typeof customTimeout === 'undefined') {131 return true;132 }133 if (customTimeout <= 0) {134 debug('%s(requests: %s, finished: %s) free but need to destroy by TTL, request count %s, diff is %s',135 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT], customTimeout);136 return false;137 }138 if (socket.timeout !== customTimeout) {139 socket.setTimeout(customTimeout);140 }141 return true;142 }143 // only call on addRequest144 reuseSocket(...args) {145 // reuseSocket(socket, req)146 super.reuseSocket(...args);147 const socket = args[0];148 const req = args[1];149 req.reusedSocket = true;150 const agentTimeout = this.options.timeout;151 if (getSocketTimeout(socket) !== agentTimeout) {152 // reset timeout before use153 socket.setTimeout(agentTimeout);154 debug('%s reset timeout to %sms', socket[SOCKET_NAME], agentTimeout);155 }156 socket[SOCKET_REQUEST_COUNT]++;157 debug('%s(requests: %s, finished: %s) reuse on addRequest, timeout %sms',158 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT],159 getSocketTimeout(socket));160 }161 [CREATE_ID]() {162 const id = this[CURRENT_ID]++;163 if (this[CURRENT_ID] === Number.MAX_SAFE_INTEGER) this[CURRENT_ID] = 0;164 return id;165 }166 [INIT_SOCKET](socket, options) {167 // bugfix here.168 // https on node 8, 10 won't set agent.options.timeout by default169 // TODO: need to fix on node itself170 if (options.timeout) {171 const timeout = getSocketTimeout(socket);172 if (!timeout) {173 socket.setTimeout(options.timeout);174 }175 }176 if (this.options.keepAlive) {177 // Disable Nagle's algorithm: http://blog.caustik.com/2012/04/08/scaling-node-js-to-100k-concurrent-connections/178 // https://fengmk2.com/benchmark/nagle-algorithm-delayed-ack-mock.html179 socket.setNoDelay(true);180 }181 this.createSocketCount++;182 if (this.options.socketActiveTTL) {183 socket[SOCKET_CREATED_TIME] = Date.now();184 }185 // don't show the hole '-----BEGIN CERTIFICATE----' key string186 socket[SOCKET_NAME] = `sock[${this[CREATE_ID]()}#${options._agentKey}]`.split('-----BEGIN', 1)[0];187 socket[SOCKET_REQUEST_COUNT] = 1;188 socket[SOCKET_REQUEST_FINISHED_COUNT] = 0;189 installListeners(this, socket, options);190 }191 createConnection(options, oncreate) {192 let called = false;193 const onNewCreate = (err, socket) => {194 if (called) return;195 called = true;196 if (err) {197 this.createSocketErrorCount++;198 return oncreate(err);199 }200 this[INIT_SOCKET](socket, options);201 oncreate(err, socket);202 };203 const newSocket = super.createConnection(options, onNewCreate);204 if (newSocket) onNewCreate(null, newSocket);205 }206 get statusChanged() {207 const changed = this.createSocketCount !== this.createSocketCountLastCheck ||208 this.createSocketErrorCount !== this.createSocketErrorCountLastCheck ||209 this.closeSocketCount !== this.closeSocketCountLastCheck ||210 this.errorSocketCount !== this.errorSocketCountLastCheck ||211 this.timeoutSocketCount !== this.timeoutSocketCountLastCheck ||212 this.requestCount !== this.requestCountLastCheck;213 if (changed) {214 this.createSocketCountLastCheck = this.createSocketCount;215 this.createSocketErrorCountLastCheck = this.createSocketErrorCount;216 this.closeSocketCountLastCheck = this.closeSocketCount;217 this.errorSocketCountLastCheck = this.errorSocketCount;218 this.timeoutSocketCountLastCheck = this.timeoutSocketCount;219 this.requestCountLastCheck = this.requestCount;220 }221 return changed;222 }223 getCurrentStatus() {224 return {225 createSocketCount: this.createSocketCount,226 createSocketErrorCount: this.createSocketErrorCount,227 closeSocketCount: this.closeSocketCount,228 errorSocketCount: this.errorSocketCount,229 timeoutSocketCount: this.timeoutSocketCount,230 requestCount: this.requestCount,231 freeSockets: inspect(this.freeSockets),232 sockets: inspect(this.sockets),233 requests: inspect(this.requests),234 };235 }236}237// node 8 don't has timeout attribute on socket238// https://github.com/nodejs/node/pull/21204/files#diff-e6ef024c3775d787c38487a6309e491dR408239function getSocketTimeout(socket) {240 return socket.timeout || socket._idleTimeout;241}242function installListeners(agent, socket, options) {243 debug('%s create, timeout %sms', socket[SOCKET_NAME], getSocketTimeout(socket));244 // listener socket events: close, timeout, error, free245 function onFree() {246 // create and socket.emit('free') logic247 // https://github.com/nodejs/node/blob/master/lib/_http_agent.js#L311248 // no req on the socket, it should be the new socket249 if (!socket._httpMessage && socket[SOCKET_REQUEST_COUNT] === 1) return;250 socket[SOCKET_REQUEST_FINISHED_COUNT]++;251 agent.requestCount++;252 debug('%s(requests: %s, finished: %s) free',253 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT]);254 // should reuse on pedding requests?255 const name = agent.getName(options);256 if (socket.writable && agent.requests[name] && agent.requests[name].length) {257 // will be reuse on agent free listener258 socket[SOCKET_REQUEST_COUNT]++;259 debug('%s(requests: %s, finished: %s) will be reuse on agent free event',260 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT]);261 }262 }263 socket.on('free', onFree);264 function onClose(isError) {265 debug('%s(requests: %s, finished: %s) close, isError: %s',266 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT], isError);267 agent.closeSocketCount++;268 }269 socket.on('close', onClose);270 // start socket timeout handler271 function onTimeout() {272 // onTimeout and emitRequestTimeout(_http_client.js)273 // https://github.com/nodejs/node/blob/v12.x/lib/_http_client.js#L711274 const listenerCount = socket.listeners('timeout').length;275 // node <= 10, default listenerCount is 1, onTimeout276 // 11 < node <= 12, default listenerCount is 2, onTimeout and emitRequestTimeout277 // node >= 13, default listenerCount is 3, onTimeout,278 // onTimeout(https://github.com/nodejs/node/pull/32000/files#diff-5f7fb0850412c6be189faeddea6c5359R333)279 // and emitRequestTimeout280 const timeout = getSocketTimeout(socket);281 const req = socket._httpMessage;282 const reqTimeoutListenerCount = req && req.listeners('timeout').length || 0;283 debug('%s(requests: %s, finished: %s) timeout after %sms, listeners %s, defaultTimeoutListenerCount %s, hasHttpRequest %s, HttpRequest timeoutListenerCount %s',284 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT],285 timeout, listenerCount, defaultTimeoutListenerCount, !!req, reqTimeoutListenerCount);286 if (debug.enabled) {287 debug('timeout listeners: %s', socket.listeners('timeout').map(f => f.name).join(', '));288 }289 agent.timeoutSocketCount++;290 const name = agent.getName(options);291 if (agent.freeSockets[name] && agent.freeSockets[name].indexOf(socket) !== -1) {292 // free socket timeout, destroy quietly293 socket.destroy();294 // Remove it from freeSockets list immediately to prevent new requests295 // from being sent through this socket.296 agent.removeSocket(socket, options);297 debug('%s is free, destroy quietly', socket[SOCKET_NAME]);298 } else {299 // if there is no any request socket timeout handler,300 // agent need to handle socket timeout itself.301 //302 // custom request socket timeout handle logic must follow these rules:303 // 1. Destroy socket first304 // 2. Must emit socket 'agentRemove' event tell agent remove socket305 // from freeSockets list immediately.306 // Otherise you may be get 'socket hang up' error when reuse307 // free socket and timeout happen in the same time.308 if (reqTimeoutListenerCount === 0) {309 const error = new Error('Socket timeout');310 error.code = 'ERR_SOCKET_TIMEOUT';311 error.timeout = timeout;312 // must manually call socket.end() or socket.destroy() to end the connection.313 // https://nodejs.org/dist/latest-v10.x/docs/api/net.html#net_socket_settimeout_timeout_callback314 socket.destroy(error);315 agent.removeSocket(socket, options);316 debug('%s destroy with timeout error', socket[SOCKET_NAME]);317 }318 }319 }320 socket.on('timeout', onTimeout);321 function onError(err) {322 const listenerCount = socket.listeners('error').length;323 debug('%s(requests: %s, finished: %s) error: %s, listenerCount: %s',324 socket[SOCKET_NAME], socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT],325 err, listenerCount);326 agent.errorSocketCount++;327 if (listenerCount === 1) {328 // if socket don't contain error event handler, don't catch it, emit it again329 debug('%s emit uncaught error event', socket[SOCKET_NAME]);330 socket.removeListener('error', onError);331 socket.emit('error', err);332 }333 }334 socket.on('error', onError);335 function onRemove() {336 debug('%s(requests: %s, finished: %s) agentRemove',337 socket[SOCKET_NAME],338 socket[SOCKET_REQUEST_COUNT], socket[SOCKET_REQUEST_FINISHED_COUNT]);339 // We need this function for cases like HTTP 'upgrade'340 // (defined by WebSockets) where we need to remove a socket from the341 // pool because it'll be locked up indefinitely342 socket.removeListener('close', onClose);343 socket.removeListener('error', onError);344 socket.removeListener('free', onFree);345 socket.removeListener('timeout', onTimeout);346 socket.removeListener('agentRemove', onRemove);347 }348 socket.on('agentRemove', onRemove);349}350module.exports = Agent;351function inspect(obj) {352 const res = {};353 for (const key in obj) {354 res[key] = obj[key].length;355 }356 return res;...

Full Screen

Full Screen

chatServer.js

Source:chatServer.js Github

copy

Full Screen

1// Require the packages we will use:2const http = require("http"),3fs = require("fs");4const { SocketAddress } = require("net");5const port = 3456;6const file = "chatClient.html";7// Listen for HTTP connections. This is essentially a miniature static file server that only serves our one file, client.html, on port 3456:8const server = http.createServer(function (req, res) {9 // This callback runs when a new connection is made to our HTTP server.10 fs.readFile(file, function (err, data) {11 // This callback runs when the client.html file has been read from the filesystem.12 if (err) return res.writeHead(500);13 res.writeHead(200);14 res.end(data);15 });16});17server.listen(port);18// Import Socket.IO and pass our HTTP server object to it.19const socketio = require("socket.io")(http, {20 wsEngine: 'ws'21});22// Attach our Socket.IO server to our HTTP server to listen23const io = socketio.listen(server);24let starterRooms = [{"name" : "Home"}, {"name" : "Room2"}];25let joinedSocketId = {};26let socketIds = {};27let currUsers = {};28let listOfRooms = {};29let setSocketIds = new Set();30io.sockets.on("connection", function (socket) {31 // This callback runs when a new Socket.IO connection is established.32 socket.on("newUser", function(username){33 if(username == ""){34 socket.emit("displayInformation", "You must enter a username. Refresh the page and try again.")35 }36 let newUser = {};37 //let newUser = username["userName"]38 newUser.username = username; 39 newUser.room = "Home";40 //console.log("Current Users:", currUsers);41 socket.room = "Home";42 socket.join("Home");43 setSocketIds.add(socket.id);44 socket.username = username;45 socket.bannedFrom = new Set();46 socketIds[username] = socket.id;47 currUsers[username] = newUser;48 socket.to("Home").emit('displayInfo', username + ' has joined to this room');49 newUser.room= socket.room;50 currUsers[username] = newUser;51 io.to(socket.id).emit("updateRoomDisplay", socket.room, starterRooms);52 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));53 });54 socket.on('message_to_server', function (data) {55 // This callback runs when the server receives a new message from the client.56 console.log("message123: " + data["message"]);57 console.log("Room: " + socket.room)// log it to the Node.JS output58 //this function iterates through every socket that exists and checks to see if it is in the current room59 //not very efficient but idrc60 if(data["message"] == "/currentusers"){61 inRoom = getWhoInRoom(socket.room)62 io.to(socket.id).emit("displayInfo", "Current Occupents of Room: " + inRoom);63 return;64 }65 //basic broadcast66 io.to(socket.room).emit("message_to_client", socket.username, {message: data["message"]});67 });68 socket.on("changeRoom", function(roomToChangeTo){69 if(socket.bannedFrom.has(roomToChangeTo)){70 io.sockets.in(socket.room).emit("displayInfo", "You may not enter that room, you have been banned.");71 return;72 }73 console.log("Changing rooms to: " + roomToChangeTo);74 socket.to(socket.room).emit("displayInfo", socket.username + " has left the room.");75 socket.leave(socket.room)76 socket.join(roomToChangeTo);77 socket.room = roomToChangeTo;78 currUsers[socket.username].room = roomToChangeTo;79 socket.to(roomToChangeTo).emit("displayInfo", socket.username + " has joined the room.");80 socket.emit('updateRoomDisplay', roomToChangeTo, starterRooms);81 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));82 });83 socket.on("addRoom", function(roomName){84 listOfRooms[roomName] = socket.username;85 starterRooms.push({name:roomName, creator: socket.username});86 io.emit("updateRoomDisplay", socket.room, starterRooms);87 });88 socket.on("kick", function(usernameToKick){89 if(socket.room == "Home"){90 console.log("You are in room: " + socket.room);91 socket.emit("improperPermission");92 return;93 }94 if(socket.username != listOfRooms[socket.room]){95 console.log("Recorded Owner Of Room: " + listOfRooms[socket.room]);96 console.log("User that tried to kick: " + socket.username)97 socket.emit("improperPermission");98 return;99 }100 let kickedUser = io.sockets.connected[socketIds[usernameToKick]]101 if(kickedUser == null){102 io.to(socket.id).emit("displayInfo", "User does not exist."); 103 return; 104 }105 if(kickedUser.room != socket.room){106 io.to(socket.id).emit("displayInfo", "That user is not in your room"); 107 return; 108 }109 console.log("Username to Kick: " + usernameToKick)110 console.log("Socket ID: " + socketIds[usernameToKick])111 kickedUser.leave(kickedUser.room);112 kickedUser.join("Home");113 kickedUser.room = "Home";114 console.log("Kicked Users new room: " + kickedUser.room + " (should be home)")115 io.to(socketIds[usernameToKick]).emit("updateRoomDisplay", "Home", starterRooms)116 io.to(socket.room).emit("displayInfo", "User has been kicked.");117 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));118 io.to(kickedUser.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));119 io.to(socketIds[usernameToKick]).emit("displayInfo", "You have been kicked from the room.");120 });121 socket.on("ban", function(usernameToBan){122 if(socket.room == "Home"){123 console.log("You are in room: " + socket.room);124 socket.emit("improperPermission");125 return;126 }127 if(socket.username != listOfRooms[socket.room]){128 console.log("Recorded Owner Of Room: " + listOfRooms[socket.room]);129 console.log("User that tried to kick: " + socket.username)130 socket.emit("improperPermission");131 return;132 } 133 let kickedUser = io.sockets.connected[socketIds[usernameToBan]]134 if(kickedUser == null){135 io.to(socket.id).emit("displayInfo", "User does not exist."); 136 return; 137 }138 if(kickedUser.room != socket.room){139 io.to(socket.id).emit("displayInfo", "That user is not in your room"); 140 return; 141 }142 console.log("Username to Ban: " + usernameToBan)143 console.log("Socket ID: " + socketIds[usernameToBan])144 kickedUser.leave(kickedUser.room);145 kickedUser.join("Home");146 kickedUser.room = "Home";147 kickedUser.bannedFrom.add(socket.room);148 io.to(socketIds[usernameToBan]).emit("updateRoomDisplay", "Home", starterRooms)149 io.to(socket.room).emit("displayInfo", "User has been banned.");150 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));151 io.to(kickedUser.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));152 io.to(socketIds[usernameToBan]).emit("displayInfo", "You have been banned from that room.");153 })154 socket.on("disconnect", function(){155 setSocketIds.delete(socket.id);156 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));157 console.log(socket.username + " has disconnected.");158 })159 function getWhoInRoom(currentRoom){160 let inRoom = "";161 setSocketIds.forEach(id =>{162 let randomSocket = io.sockets.connected[id];163 if(randomSocket.room == currentRoom){164 inRoom = inRoom + " " + randomSocket.username;165 } 166 })167 inRoom.replace("null", "");168 return inRoom;169 }170 socket.on("joinPrivateRoom", function(privateRoom, guess){171 //every private room should open a prompt onclick, redirect to that function in the html172 console.log("in joinPrivateRoom (js)")173 for(var i=0; i<starterRooms.length; i++){174 console.log(starterRooms[i].hasOwnProperty('password'));175 if(starterRooms[i].name == privateRoom){176 if(starterRooms[i].hasOwnProperty('password')){177 //check password guess with real password178 if(starterRooms[i].password == guess){179 //currUsers[socket.user].room = room_name;180 console.log("password approved");181 socket.broadcast.to(socket.room).emit("displayInfo", socket.username + ' has left this room');182 socket.leave(socket.room);183 console.log(socket.user + " left room: " + socket.room);184 socket.join(privateRoom);185 socket.room = privateRoom;186 currUsers[socket.username].room = privateRoom;187 //socket.to(socket.room).emit("displayInfo", socket.username + " has joined this room.");188 189 socket.to(privateRoom).emit("displayInfo", socket.username + " has joined the room.");190 socket.emit('updateRoomDisplay', privateRoom, starterRooms);191 io.to(socket.room).emit("displayWhoInRoom", getWhoInRoom(socket.room));192 //socket.broadcast.to(privateRoom).emit("displayInfo", socket.username + ' has joined this room');193 return;194 } else {195 socket.emit("displayInfo", 'wrong password for ' +privateRoom);196 return;197 }198 }199 }200 201 }202 });203 socket.on("addWithPassword", function(roomName, password){204 starterRooms.push({name:roomName, creator: socket.username, password: password});205 listOfRooms[roomName] = socket.username;206 //io.emit('addRoom', starterRooms, socket.room);207 //socket.emit("updateRoomDisplay", socket.room, starterRooms);208 io.emit("updateRoomDisplay", socket.room, starterRooms);209 });210 211 socket.on("pm", function(to, msg){212 let from = socket.username;213 io.to(socketIds[to]).emit('sendPM', from, msg);214 });215 socket.on("passOnCreator", function(newCreator){216 listOfRooms[socket.room] = newCreator;217 for(var i = 0; i < starterRooms.length; i++){218 if(starterRooms[i].name == socket.room){219 if(starterRooms[i].creator == socket.username){220 starterRooms[i].creator = newCreator;221 }222 else {223 console.log("Recorded creator Of Room: " + starterRooms[i].creator);224 console.log("User that tried to give privileges: " + socket.username)225 socket.emit("improperPermission");226 }227 } 228 }229 console.log("New creator for room " + socket.room + " : " + newCreator);230 });231 socket.on("deleteRoom", function(room_name){232 if(room_name == "Home"){233 console.log("You are in the home room");234 io.to(socket.id).emit("displayInfo", "You can't delete the home room"); 235 return;236 }237 for(var i=0; i<starterRooms.length; i++){238 if(starterRooms[i].name == room_name){239 //find room to be deleted in the array of rooms240 if(socket.username != starterRooms[i].creator){241 console.log("Recorded Owner Of Room: " + starterRooms[i].creator);242 console.log("User that tried to delete room: " + socket.username)243 socket.emit("improperPermission");244 return; 245 }246 else{247 //iterate through connected sockets and move them to the home room.248 setSocketIds.forEach(id =>{249 if(io.sockets.connected[id].room == room_name){250 io.sockets.connected[id].leave(room_name);251 io.sockets.connected[id].join("Home");252 io.to(socket.id).emit("displayInfo", "The room you were in has been deleted. Click a room to join."); 253 }254 })255 starterRooms.splice(i,1);256 io.sockets.emit("updateRoomDisplay", socket.room, starterRooms);257 }258 } 259 }260 //go through all the users in this room and leave room and put them back in home261 262 //run thru the chat rooms array to remove room263 /*264 for (let i in starterRooms){265 if(starterRooms[i].name==room_name){266 starterRooms.splice(i,1);267 }268 }269 */270 io.sockets.emit("updateRoomDisplay", socket.room, starterRooms);271 });...

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run autotest automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

... View sample repo... View DocumentationSign up Free
›
_

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

...
Sign up with GoogleSign up with Email
Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful