cypress/packages/server/test/unit/iframes_spec.js

require('../spec_helper')

const { iframesController } = require('../../lib/controllers/iframes')
const files = require('../../lib/controllers/files')

describe('controllers/iframes', () => {
  describe('e2e', () => {
    it('sets Origin-Agent-Cluster response header to false', () => {
      sinon.stub(files, 'handleIframe')

      const mockReq = {}
      const mockRes = {
        setHeader: sinon.stub(),
      }

      const controllerOptions = {
        getSpec: sinon.stub(),
        remoteStates: sinon.stub(),
        config: {},
      }

      iframesController.e2e(controllerOptions, mockReq, mockRes)

      expect(mockRes.setHeader).to.have.been.calledWith('Origin-Agent-Cluster', '?0')
      expect(files.handleIframe).to.have.been.calledWith(
        mockReq, mockRes, controllerOptions.config, controllerOptions.remoteStates, sinon.match({
          specFilter: undefined, specType: 'integration',
        }),
      )
    })
  })
})