energylink/public/app/core/specs/org_switcher.jest.ts

import { OrgSwitchCtrl } from "../components/org_switcher";
import q from "q";

jest.mock("app/core/services/context_srv", () => ({
  contextSrv: {
    user: { orgId: 1 }
  }
}));

describe("OrgSwitcher", () => {
  describe("when switching org", () => {
    let expectedHref;
    let expectedUsingUrl;

    beforeEach(() => {
      const backendSrvStub: any = {
        get: url => {
          return q.resolve([]);
        },
        post: url => {
          expectedUsingUrl = url;
          return q.resolve({});
        }
      };

      const orgSwitcherCtrl = new OrgSwitchCtrl(backendSrvStub);

      orgSwitcherCtrl.getWindowLocationHref = () =>
        "http://localhost:3000?orgId=1&from=now-3h&to=now";
      orgSwitcherCtrl.setWindowLocationHref = href => (expectedHref = href);

      return orgSwitcherCtrl.setUsingOrg({ orgId: 2 });
    });

    it("should switch orgId in call to backend", () => {
      expect(expectedUsingUrl).toBe("/api/user/using/2");
    });

    it("should switch orgId in url", () => {
      expect(expectedHref).toBe(
        "http://localhost:3000?orgId=2&from=now-3h&to=now"
      );
    });
  });
});