/**
 * Copyright (c) HashiCorp, Inc.
 * SPDX-License-Identifier: BUSL-1.1
 */

import { allFeatures } from 'vault/helpers/all-features';
import sinon from 'sinon';

/**
 * Sets up the necessary shared test context for testing sidebar nav components
 * @param {TestContext.owner} owner eg this.owner from within a test
 * @param {boolean} isEnterprise [default false]
 * @param {boolean} setCluster [default false] if true, will set the current cluster to one with replication & raft
 * @param {string[]} features if not passed, defaults to all features
 * @returns {hasNavPermission: sinon.SinonStub, features: string[]}
 */
export const stubFeaturesAndPermissions = (
  owner,
  isEnterprise = false,
  setCluster = false,
  features,
  hasPermission = true
) => {
  const permissions = owner.lookup('service:permissions');
  const hasNavPermission = sinon.stub(permissions, 'hasNavPermission');
  hasNavPermission.returns(hasPermission);
  sinon.stub(permissions, 'navPathParams');

  const version = owner.lookup('service:version');
  version.type = isEnterprise ? 'enterprise' : 'community';
  version.features = features || allFeatures();

  const auth = owner.lookup('service:auth');
  sinon.stub(auth, 'authData').value({});

  if (setCluster) {
    owner.lookup('service:currentCluster').setCluster({
      id: 'foo',
      anyReplicationEnabled: true,
      usingRaft: true,
    });
  }

  return { hasNavPermission, features };
};