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');
    });
  });
});