From 4c190170c3043f654b1c865da5295931fc2c60a6 Mon Sep 17 00:00:00 2001
From: JimiC <jimikar@gmail.com>
Date: Fri, 11 Nov 2016 12:56:45 +0200
Subject: [PATCH] Improve stompjs typings.

---
 stompjs/index.d.ts       |  3 ++-
 stompjs/stompjs-tests.ts | 17 +++++++++++++++--
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/stompjs/index.d.ts b/stompjs/index.d.ts
index 607be77162..008b99b511 100644
--- a/stompjs/index.d.ts
+++ b/stompjs/index.d.ts
@@ -26,7 +26,8 @@ export class Client {
 
     debug(...args: string[]): any;
 
-    connect(...args: any[]): any;
+    connect(headers: { login: string, passcode: string, host?: string }, connectCallback: (frame?: Frame) => any, errorCallback?: (error: string) => any): any;
+    connect(login: string, passcode: string, connectCallback: (frame?: Frame) => any, errorCallback?: (error: string) => any, host?: string): any;
     disconnect(disconnectCallback: () => any, headers?: {}): any;
 
     send(destination: string, headers?: {}, body?: string): any;
diff --git a/stompjs/stompjs-tests.ts b/stompjs/stompjs-tests.ts
index 68320259a9..15331fc81f 100644
--- a/stompjs/stompjs-tests.ts
+++ b/stompjs/stompjs-tests.ts
@@ -28,8 +28,21 @@ client.ws = new WebSocket('url');
 
 client.debug();
 
-client.connect();
-client.connect('', () => { }, {});
+client.connect({ login: 'user', passcode: 'pass' }, () => { });
+client.connect({ login: 'user', passcode: 'pass' }, () => { }, (error) => { });
+client.connect({ login: 'user', passcode: 'pass' }, (frame) => { });
+client.connect({ login: 'user', passcode: 'pass' }, (frame) => { }, (error) => { });
+client.connect({ login: 'user', passcode: 'pass', host: 'host' }, () => { });
+client.connect({ login: 'user', passcode: 'pass', host: 'host' }, (frame) => { });
+client.connect({ login: 'user', passcode: 'pass', host: 'host' }, () => { }, (error) => { });
+client.connect({ login: 'user', passcode: 'pass', host: 'host' }, (frame) => { }, (error) => { });
+
+client.connect('user', 'pass', () => { });
+client.connect('user', 'pass', () => { }, () => { });
+client.connect('user', 'pass', () => { }, (error) => { });
+client.connect('user', 'pass', () => { }, (error) => { }, 'host');
+client.connect('user', 'pass', (frame) => { }, (error) => { });
+client.connect('user', 'pass', (frame) => { }, (error) => { }, 'host');
 
 client.disconnect(() => { });
 client.disconnect(() => { }, {});