From e3eec6e93e1397630c4d63ba9f26d96385ecb5b9 Mon Sep 17 00:00:00 2001
From: simojenki <simojenki@users.noreply.github.com>
Date: Fri, 26 Feb 2021 12:31:46 +1100
Subject: [PATCH] Scenario test fetching auth token

---
 tests/scenarios.test.ts | 76 +++++++++++++++++++++++++----------------
 1 file changed, 46 insertions(+), 30 deletions(-)

diff --git a/tests/scenarios.test.ts b/tests/scenarios.test.ts
index 4a8ec52..6f3535b 100644
--- a/tests/scenarios.test.ts
+++ b/tests/scenarios.test.ts
@@ -1,9 +1,9 @@
-import { createClientAsync } from "soap";
+import { createClientAsync, Client } from "soap";
 import { Express } from "express";
 
 import request from "supertest";
 
-import { GetAppLinkResult } from "../src/smapi";
+import { GetAppLinkResult, GetDeviceAuthTokenResult } from "../src/smapi";
 import { getAppLinkMessage } from "./builders";
 import { InMemoryMusicService } from "./in_memory_music_service";
 import { InMemoryLinkCodes } from "../src/link_codes";
@@ -12,6 +12,16 @@ import makeServer from "../src/server";
 import { Service, bonobService, SONOS_DISABLED } from "../src/sonos";
 import supersoap from "./supersoap";
 
+class FooDriver {
+  client: Client;
+  token: GetDeviceAuthTokenResult;
+
+  constructor(client: Client, token: GetDeviceAuthTokenResult) {
+    this.client = client;
+    this.token = token;
+  }
+}
+
 class SonosDriver {
   server: Express;
   rootUrl: string;
@@ -36,37 +46,43 @@ class SonosDriver {
       .get(this.stripServiceRoot(this.service.presentation.uri!))
       .expect(200);
 
-    return createClientAsync(`${this.service.uri}?wsdl`, {
+    const client = await createClientAsync(`${this.service.uri}?wsdl`, {
       endpoint: this.service.uri,
       httpClient: supersoap(this.server, this.rootUrl),
-    }).then((client) =>
-      client
-        .getAppLinkAsync(getAppLinkMessage())
-        .then(
-          ([result]: [GetAppLinkResult]) =>
-            result.getAppLinkResult.authorizeAccount.deviceLink
-        )
-        .then(({ regUrl, linkCode }: { regUrl: string; linkCode: string }) => ({
-          login: async ({ username, password }: Credentials) => {
-            await request(this.server).get(this.stripServiceRoot(regUrl)).expect(200);
+    });
 
-            return request(this.server)
-              .post(this.stripServiceRoot(regUrl))
-              .type("form")
-              .send({ username, password, linkCode })
-              .then(response => ({
-                expectSuccess: () => {
-                  expect(response.status).toEqual(200);
-                  expect(response.text).toContain("Login successful")
-                },
-                expectFailure: () => {
-                  expect(response.status).toEqual(403);
-                  expect(response.text).toContain("Login failed")
-                },
-              }));
-          },
-        }))
-    );
+    return client
+      .getAppLinkAsync(getAppLinkMessage())
+      .then(
+        ([result]: [GetAppLinkResult]) =>
+          result.getAppLinkResult.authorizeAccount.deviceLink
+      )
+      .then(({ regUrl, linkCode }: { regUrl: string; linkCode: string }) => ({
+        login: async ({ username, password }: Credentials) => {
+          await request(this.server)
+            .get(this.stripServiceRoot(regUrl))
+            .expect(200);
+
+          return request(this.server)
+            .post(this.stripServiceRoot(regUrl))
+            .type("form")
+            .send({ username, password, linkCode })
+            .then((response) => ({
+              expectSuccess: async () => {
+                expect(response.status).toEqual(200);
+                expect(response.text).toContain("Login successful");
+
+                return client
+                  .getDeviceAuthTokenAsync({ linkCode })
+                  .then((authToken: [GetDeviceAuthTokenResult, any]) => new FooDriver(client, authToken[0]));
+              },
+              expectFailure: () => {
+                expect(response.status).toEqual(403);
+                expect(response.text).toContain("Login failed");
+              },
+            }));
+        },
+      }));
   }
 }