Support authentication

src/core/client/stream/local/initLocalState.spec.ts

@@ -2,6 +2,7 @@ import { Environment, RecordSource } from "relay-runtime";
 
 import { timeout } from "talk-common/utils";
 import { LOCAL_ID } from "talk-framework/lib/relay";
+import { createInMemoryStorage } from "talk-framework/lib/storage";
 import { createRelayEnvironment } from "talk-framework/testHelpers";
 
 import initLocalState from "./initLocalState";
@@ -18,7 +19,7 @@ beforeEach(() => {
 });
 
 it("init local state", async () => {
-  initLocalState(environment);
+  initLocalState(environment, { localStorage: createInMemoryStorage() } as any);
   await timeout();
   expect(JSON.stringify(source.toJSON(), null, 2)).toMatchSnapshot();
 });
@@ -32,7 +33,7 @@ it("set assetID from query", () => {
     document.title,
     `http://localhost/?assetID=${assetID}`
   );
-  initLocalState(environment);
+  initLocalState(environment, { localStorage: createInMemoryStorage() } as any);
   expect(source.get(LOCAL_ID)!.assetID).toBe(assetID);
   window.history.replaceState(previousState, document.title, previousLocation);
 });
@@ -46,7 +47,16 @@ it("set commentID from query", () => {
     document.title,
     `http://localhost/?commentID=${commentID}`
   );
-  initLocalState(environment);
+  initLocalState(environment, { localStorage: createInMemoryStorage() } as any);
   expect(source.get(LOCAL_ID)!.commentID).toBe(commentID);
   window.history.replaceState(previousState, document.title, previousLocation);
 });
+
+it("set authToken from localStorage", () => {
+  const authToken = "auth-token";
+  const localStorage = createInMemoryStorage();
+  localStorage.setItem("authToken", authToken);
+  initLocalState(environment, { localStorage } as any);
+  expect(source.get(LOCAL_ID)!.authToken).toBe(authToken);
+  localStorage.removeItem("authToken");
+});