-
Notifications
You must be signed in to change notification settings - Fork 766
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #70 from ethereumjs/tests
Unit tests for all components
- Loading branch information
Showing
49 changed files
with
2,596 additions
and
556 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,150 @@ | ||
| 'use strict' | ||
|
|
||
| const EventEmitter = require('events') | ||
|
|
||
| /** | ||
| * Binds a protocol implementation to the specified peer | ||
| * @memberof module:net/protocol | ||
| */ | ||
| class BoundProtocol extends EventEmitter { | ||
| /** | ||
| * Create bound protocol | ||
| * @param {Object} options constructor parameters | ||
| * @param {Protocol} options.protocol protocol to bind | ||
| * @param {Peer} options.peer peer that protocol is bound to | ||
| * @param {Sender} options.sender message sender | ||
| */ | ||
| constructor (options) { | ||
| super() | ||
|
|
||
| this.protocol = options.protocol | ||
| this.peer = options.peer | ||
| this.sender = options.sender | ||
| this.name = this.protocol.name | ||
| this.versions = this.protocol.versions | ||
| this.timeout = this.protocol.timeout | ||
| this.logger = this.protocol.logger | ||
| this._status = {} | ||
| this.resolvers = new Map() | ||
| this.sender.on('message', message => { | ||
| try { | ||
| this.handle(message) | ||
| } catch (error) { | ||
| this.emit('error', error) | ||
| } | ||
| }) | ||
| this.sender.on('error', error => this.emit('error', error)) | ||
| this.addMethods() | ||
| } | ||
|
|
||
| get status () { | ||
| return this._status | ||
| } | ||
|
|
||
| set status (status) { | ||
| Object.assign(this._status, status) | ||
| } | ||
|
|
||
| async handshake (sender) { | ||
| this._status = await this.protocol.handshake(sender) | ||
| } | ||
|
|
||
| /** | ||
| * Handle incoming message | ||
| * @private | ||
| * @param {Object} message message object | ||
| * @emits message | ||
| */ | ||
| handle (incoming) { | ||
| const messages = this.protocol.messages | ||
| const message = messages.find(m => m.code === incoming.code) | ||
| if (!message) { | ||
| return | ||
| } | ||
|
|
||
| let data | ||
| let error | ||
| try { | ||
| data = this.protocol.decode(message, incoming.payload) | ||
| } catch (e) { | ||
| error = new Error(`Could not decode message ${message.name}: ${e}`) | ||
| } | ||
| const resolver = this.resolvers.get(incoming.code) | ||
| if (resolver) { | ||
| clearTimeout(resolver.timeout) | ||
| this.resolvers.delete(incoming.code) | ||
| if (error) { | ||
| resolver.reject(error) | ||
| } else { | ||
| resolver.resolve(data) | ||
| } | ||
| } else { | ||
| if (error) { | ||
| this.emit('error', error) | ||
| } else { | ||
| this.emit('message', { name: message.name, data: data }) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Send message with name and the specified args | ||
| * @param {string} name message name | ||
| * @param {object} args message arguments | ||
| */ | ||
| send (name, args) { | ||
| const messages = this.protocol.messages | ||
| const message = messages.find(m => m.name === name) | ||
| if (message) { | ||
| const encoded = this.protocol.encode(message, args) | ||
| this.sender.sendMessage(message.code, encoded) | ||
| return message | ||
| } else { | ||
| throw new Error(`Unknown message: ${name}`) | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Returns a promise that resolves with the message payload when a response | ||
| * to the specified message is received | ||
| * @param {string} name message to wait for | ||
| * @param {object} args message arguments | ||
| * @return {Promise} | ||
| */ | ||
| async request (name, args) { | ||
| const message = this.send(name, args) | ||
| const resolver = { | ||
| timeout: null, | ||
| resolve: null, | ||
| reject: null | ||
| } | ||
| if (this.resolvers.get(message.response)) { | ||
| throw new Error(`Only one active request allowed per message type (${name})`) | ||
| } | ||
| this.resolvers.set(message.response, resolver) | ||
| return new Promise((resolve, reject) => { | ||
| resolver.timeout = setTimeout(() => { | ||
| resolver.timeout = null | ||
| this.resolvers.delete(message.response) | ||
| reject(new Error(`Request timed out after ${this.timeout}ms`)) | ||
| }, this.timeout) | ||
| resolver.resolve = resolve | ||
| resolver.reject = reject | ||
| }) | ||
| } | ||
|
|
||
| /** | ||
| * Add a methods to the bound protocol for each protocol message that has a | ||
| * corresponding response message | ||
| */ | ||
| addMethods () { | ||
| const messages = this.protocol.messages.filter(m => m.response) | ||
| for (let message of messages) { | ||
| const name = message.name | ||
| const camel = name[0].toLowerCase() + name.slice(1) | ||
| this[name] = this[camel] = async (args) => this.request(name, args) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| module.exports = BoundProtocol |
Oops, something went wrong.