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

import { OrgSwitchCtrl } from '../components/org_switcher';
// @ts-ignore
import q from 'q';

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

jest.mock('app/core/config', () => {
  return {
    appSubUrl: '/subUrl',
  };
});

describe('OrgSwitcher', () => {
  describe('when switching org', () => {
    let expectedHref: string;
    let expectedUsingUrl: string;

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

      const orgSwitcherCtrl = new OrgSwitchCtrl(backendSrvStub);

      orgSwitcherCtrl.setWindowLocation = 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 and redirect to home page', () => {
      expect(expectedHref).toBe('/subUrl/?orgId=2');
    });
  });
});