ambari/ambari-web/test/controllers/main/service_test.js

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

var App = require('app');
require('controllers/main/service');

var testHelpers = require('test/helpers');
var mainServiceController;

function getController() {
  return App.MainServiceController.create({});
}

describe('App.MainServiceController', function () {

  var tests = Em.A([
    {
      isStartStopAllClicked: true,
      content: Em.A([
        Em.Object.create({
          healthStatus: 'red',
          serviceName: 'HIVE',
          isClientsOnly: false
        }),
        Em.Object.create({
          healthStatus: 'red',
          serviceName: 'HDFS',
          isClientsOnly: false
        }),
        Em.Object.create({
          healthStatus: 'red',
          serviceName: 'TEZ',
          isClientsOnly: true
        })
      ]),
      eStart: true,
      eStop: true,
      mStart: 'mainServiceController StartAll is Disabled 2',
      mStop: 'mainServiceController StopAll is Disabled 2'
    },
    {
      isStartStopAllClicked: false,
      content: Em.A([
        Em.Object.create({
          healthStatus: 'green',
          serviceName: 'HIVE',
          isClientsOnly: false
        }),
        Em.Object.create({
          healthStatus: 'red',
          serviceName: 'HDFS',
          isClientsOnly: false
        }),
        Em.Object.create({
          healthStatus: 'red',
          serviceName: 'TEZ',
          isClientsOnly: true
        })
      ]),
      eStart: false,
      eStop: false,
      mStart: 'mainServiceController StartAll is Enabled 3',
      mStop: 'mainServiceController StopAll is Enabled 3'
    }

  ]);
  beforeEach(function() {
    mainServiceController = getController();
  });

  afterEach(function () {
    mainServiceController.destroy();
  });

  App.TestAliases.testAsComputedNotEqual(getController(), 'isStartStopAllClicked', 'App.router.backgroundOperationsController.allOperationsCount', 0);

  describe('#isStartAllDisabled', function () {
    tests.forEach(function (test) {
      it(test.mStart, function () {
        mainServiceController = App.MainServiceController.create({
          content: test.content,
          isStartStopAllClicked: test.isStartStopAllClicked
        });
        expect(mainServiceController.get('isStartAllDisabled')).to.equals(test.eStart);
      });
    });
  });

  describe('#isStopAllDisabled', function () {
    tests.forEach(function (test) {
      it(test.mStop, function () {
        mainServiceController = App.MainServiceController.create({
          content: test.content,
          isStartStopAllClicked: test.isStartStopAllClicked
        });
        expect(mainServiceController.get('isStopAllDisabled')).to.equals(test.eStop);
      });
    });
  });

  describe("#isAllServicesInstalled", function() {

    beforeEach(function() {
      sinon.stub(App.StackService, 'find').returns([
        Em.Object.create({serviceName: 'S1'})
      ]);
    });
    afterEach(function() {
      App.StackService.find.restore();
    });

    it("content is null", function() {
      mainServiceController.reopen({
        'content': null
      });
      mainServiceController.propertyDidChange('isAllServicesInstalled');
      expect(mainServiceController.get('isAllServicesInstalled')).to.be.false;
    });

    it("content is empty", function() {
      mainServiceController.reopen({
        'content': []
      });
      mainServiceController.propertyDidChange('isAllServicesInstalled');
      expect(mainServiceController.get('isAllServicesInstalled')).to.be.false;
    });

    it("content match stack services", function() {
      mainServiceController.reopen({
        'content': [Em.Object.create({serviceName: 'S1'})]
      });
      mainServiceController.propertyDidChange('isAllServicesInstalled');
      expect(mainServiceController.get('isAllServicesInstalled')).to.be.true;
    });
  });

  describe('#cluster', function() {

    Em.A([
      {
        isLoaded: true,
        cluster: [],
        m: 'cluster is loaded',
        e: {name: 'c1'}
      },
      {
        isLoaded: false,
        cluster: [],
        m: 'cluster is not loaded',
        e: null
      }
    ]).forEach(function(test) {
        describe(test.m, function() {

          beforeEach(function () {
            sinon.stub(App.router, 'get', function(k) {
              if ('clusterController.isClusterDataLoaded' === k) return test.isLoaded;
              return Em.get(App.router, k);
            });
            sinon.stub(App.Cluster, 'find', function() {
              return [test.e];
            });
          });

          afterEach(function () {
            App.router.get.restore();
            App.Cluster.find.restore();
          });

          it('cluster is valid', function () {
            var c = mainServiceController.get('cluster');
            expect(c).to.eql(test.e);
          });

        });
      });

  });

  describe('#startAllService', function() {

    beforeEach(function() {
      sinon.stub(mainServiceController, 'allServicesCall', Em.K);
    });

    afterEach(function() {
      mainServiceController.allServicesCall.restore();
    });

    it('target is disabled', function() {
      var event = {target: {className: 'disabled', nodeType: 1}};
      var r = mainServiceController.startAllService(event);
      expect(r).to.be.null;
    });

    it('parent is disabled', function() {
      var event = {target: {parentElement: {className: 'disabled', nodeType: 1}}};
      var r = mainServiceController.startAllService(event);
      expect(r).to.be.null;
    });

  });

  describe('#stopAllService', function() {

    beforeEach(function() {
      sinon.stub(mainServiceController, 'allServicesCall', Em.K);
    });

    afterEach(function() {
      mainServiceController.allServicesCall.restore();
    });

    it('target is disabled', function() {
      var event = {target: {className: 'disabled', nodeType: 1}};
      var r = mainServiceController.stopAllService(event);
      expect(r).to.be.null;
    });

    it('parent is disabled', function() {
      var event = {target: {parentElement: {className: 'disabled', nodeType: 1}}};
      var r = mainServiceController.stopAllService(event);
      expect(r).to.be.null;
    });

  });

  describe('#startStopAllService', function() {
    var event = { target: document.createElement("BUTTON") };

    beforeEach(function() {
      sinon.stub(mainServiceController, 'allServicesCall', Em.K);
      sinon.spy(Em.I18n, "t");
    });

    afterEach(function() {
      mainServiceController.allServicesCall.restore();
      Em.I18n.t.restore();
    });

    it ("should confirm stop if state is INSTALLED", function() {
      mainServiceController.startStopAllService(event, "INSTALLED");
      expect(Em.I18n.t.calledWith('services.service.stopAll.confirmMsg')).to.be.ok;
      expect(Em.I18n.t.calledWith('services.service.stop.confirmButton')).to.be.ok;
    });

    describe("should check last checkpoint for NN before confirming stop", function() {
      var mainServiceItemController;
      beforeEach(function() {
        mainServiceItemController = App.MainServiceItemController.create({});
        sinon.stub(mainServiceItemController, 'checkNnLastCheckpointTime', function() {
          return true;
        });
        sinon.stub(App.router, 'get', function(k) {
          if ('mainServiceItemController' === k) {
            return mainServiceItemController;
          }
          return Em.get(App.router, k);
        });
        sinon.stub(App.Service, 'find', function() {
          return [{
            serviceName: "HDFS",
            workStatus: "STARTED"
          }];
        });
      });

      afterEach(function () {
        mainServiceItemController.checkNnLastCheckpointTime.restore();
        App.router.get.restore();
        App.Service.find.restore();
      });

      it('checkNnLastCheckpointTime is called once', function () {
        mainServiceController.startStopAllService(event, "INSTALLED");
        expect(mainServiceItemController.checkNnLastCheckpointTime.calledOnce).to.equal(true);
      });

    });

    it ("should confirm start if state is not INSTALLED", function() {
      mainServiceController.startStopAllService(event, "STARTED");
      expect(Em.I18n.t.calledWith('services.service.startAll.confirmMsg')).to.be.ok;
      expect(Em.I18n.t.calledWith('services.service.start.confirmButton')).to.be.ok;
    });
  });

  describe('#allServicesCall', function() {

    var state = 'STARTED',
      query = 'some query';

    beforeEach(function() {
      sinon.stub(App, 'get', function(k) {
        if ('clusterName' === k) return 'tdk';
        return Em.get(App, k);
      });
      mainServiceController.allServicesCall(state, query);
      var args = testHelpers.findAjaxRequest('name', 'common.services.update');
      this.params = App.ajax.fakeGetUrl('common.services.update').format(args[0].data);
      this.data = JSON.parse(this.params.data);
    });

    afterEach(function() {
      App.get.restore();
    });

    it('PUT request is sent', function() {
      expect(this.params.type).to.equal('PUT');
    });
    it('Body.ServiceInfo.state is ' + state, function() {
      expect(this.data.Body.ServiceInfo.state).to.equal(state);
    });
    it('RequestInfo.context is ' + query, function() {
      expect(this.data.RequestInfo.context).to.equal(App.BackgroundOperationsController.CommandContexts.START_ALL_SERVICES);
    });

  });

  describe('#allServicesCallErrorCallback', function() {

    it('should set status to FAIL', function() {
      var params = {query: Em.Object.create({status: ''})};
      mainServiceController.allServicesCallErrorCallback({}, {}, '', {}, params);
      expect(params.query.get('status')).to.equal('FAIL');
    });

  });

  describe('#gotoAddService', function() {

    beforeEach(function() {
      sinon.stub(App.router, 'transitionTo', Em.K);
    });

    afterEach(function() {
      App.router.transitionTo.restore();
    });

    it('should not go to wizard', function() {
      mainServiceController.reopen({isAllServicesInstalled: true});
      mainServiceController.gotoAddService();
      expect(App.router.transitionTo.called).to.be.false;
    });

    it('should go to wizard', function() {
      mainServiceController.reopen({isAllServicesInstalled: false});
      mainServiceController.gotoAddService();
      expect(App.router.transitionTo.calledWith('main.serviceAdd')).to.be.true;
    });

  });

  App.TestAliases.testAsComputedEveryBy(getController(), 'isRestartAllRequiredDisabled', 'content', 'isRestartRequired', false);

  describe('#restartAllRequired', function () {

    beforeEach(function () {
      sinon.spy(App, 'showConfirmationPopup');
      sinon.spy(mainServiceController, 'restartHostComponents');
      sinon.stub(App.HostComponent, 'find', function() {
        return [
          Em.Object.create({
            componentName: 'componentName1',
            hostName: 'hostName1',
            service: {
              serviceName: 'serviceName1',
              displayName: 'displayName1'
            },
            staleConfigs: true
          }),
          Em.Object.create({
            componentName: 'componentName2',
            hostName: 'hostName2',
            service: {
              serviceName: 'serviceName2',
              displayName: 'displayName2'
            },
            staleConfigs: true
          }),
          Em.Object.create({
            componentName: 'componentName3',
            hostName: 'hostName3',
            service: {
              serviceName: 'serviceName3',
              displayName: 'displayName3'
            },
            staleConfigs: false
          })
        ];
      });
    });

    afterEach(function () {
      App.HostComponent.find.restore();
      App.showConfirmationPopup.restore();
      mainServiceController.restartHostComponents.restore();
    });

    it('should show confirmation popup with list of services and call restartHostComponents after confirmation', function () {
      var popup;
      mainServiceController.reopen({
        isRestartAllRequiredDisabled: false
      });
      popup = mainServiceController.restartAllRequired();
      popup.onPrimary();
      expect(App.showConfirmationPopup.args[0][1]).to.equal(Em.I18n.t('services.service.refreshAll.confirmMsg').format('displayName1, displayName2'));
      expect(mainServiceController.restartHostComponents.calledOnce).to.be.true;
    });

    it('should not open popup if isRestartAllRequiredDisabled is true', function(){
      mainServiceController.reopen({
        isRestartAllRequiredDisabled: true
      });
      expect(mainServiceController.restartAllRequired()).to.be.null;
    });

  });

  describe("#restartAllServices()", function() {

    beforeEach(function() {
      sinon.stub(mainServiceController, 'silentStopAllServices');
    });
    afterEach(function() {
      mainServiceController.silentStopAllServices.restore();
    });

    it("silentStopAllServices should be called", function() {
      mainServiceController.restartAllServices();
      expect(mainServiceController.silentStopAllServices.calledOnce).to.be.true;
    });
  });

  describe("#silentStopAllServices()", function() {

    it("App.ajax.send should be called", function() {
      mainServiceController.silentStopAllServices();
      var args = testHelpers.filterAjaxRequests('name', 'common.services.update');
      expect(args[0][0]).to.eql({
        name: 'common.services.update',
        sender: mainServiceController,
        data: {
          context: App.BackgroundOperationsController.CommandContexts.STOP_ALL_SERVICES,
          ServiceInfo: {
            state: 'INSTALLED'
          }
        },
        success: 'silentStopSuccess'
      });
    });
  });

  describe("#isStopAllServicesFailed()", function() {

    beforeEach(function() {
      this.mock = sinon.stub(App.Service, 'find');
    });
    afterEach(function() {
      this.mock.restore();
    });

    it("one INSTALLED service", function() {
      this.mock.returns([
        Em.Object.create({workStatus: 'INSTALLED'})
      ]);
      expect(mainServiceController.isStopAllServicesFailed()).to.be.false;
    });

    it("one STOPPING service", function() {
      this.mock.returns([
        Em.Object.create({workStatus: 'STOPPING'})
      ]);
      expect(mainServiceController.isStopAllServicesFailed()).to.be.false;
    });

    it("one STARTED service and one INSTALLED", function() {
      this.mock.returns([
        Em.Object.create({workStatus: 'STARTED'}),
        Em.Object.create({workStatus: 'INSTALLED'})
      ]);
      expect(mainServiceController.isStopAllServicesFailed()).to.be.true;
    });

    it("one STARTED service", function() {
      this.mock.returns([
        Em.Object.create({workStatus: 'STARTED'})
      ]);
      expect(mainServiceController.isStopAllServicesFailed()).to.be.true;
    });
  });

  describe("#silentStopSuccess()", function() {
    var mock = {
      dataLoading: function() {
        return {
          done: function(callback) {
            callback(true);
          }
        }
      },
      showPopup: Em.K
    };

    beforeEach(function() {
      sinon.stub(App.router, 'get').returns(mock);
      sinon.stub(Em.run, 'later', Em.clb);
      sinon.spy(mock, 'showPopup');
      sinon.stub(mainServiceController, 'silentStartAllServices');
      mainServiceController.silentStopSuccess();
    });
    afterEach(function() {
      App.router.get.restore();
      Em.run.later.restore();
      mock.showPopup.restore();
      mainServiceController.silentStartAllServices.restore();
    });

    it("showPopup should be called", function() {
      expect(mock.showPopup.calledOnce).to.be.true;
    });

    it("Em.run.later should be called", function() {
      expect(Em.run.later.calledOnce).to.be.true;
      expect(mainServiceController.get('shouldStart')).to.be.true;
    });
  });

  describe("#silentStartAllServices()", function() {

    beforeEach(function() {
      this.mockRouter = sinon.stub(App.router, 'get');
      this.mock = sinon.stub(mainServiceController, 'isStopAllServicesFailed');
      mainServiceController.removeObserver('shouldStart', mainServiceController, 'silentStartAllServices');
    });
    afterEach(function() {
      this.mockRouter.restore();
      this.mock.restore();
    });

    it("allOperationsCount is 1", function() {
      this.mockRouter.returns(Em.Object.create({
        allOperationsCount: 1
      }));
      mainServiceController.silentStartAllServices();
      expect(testHelpers.findAjaxRequest('name', 'common.services.update')).to.be.undefined;
    });

    it("shouldStart is false", function() {
      this.mockRouter.returns(Em.Object.create({
        allOperationsCount: 0
      }));
      mainServiceController.set('shouldStart', false);
      mainServiceController.silentStartAllServices();
      expect(testHelpers.findAjaxRequest('name', 'common.services.update')).to.be.undefined;
    });

    it("isStopAllServicesFailed returns true", function() {
      this.mockRouter.returns(Em.Object.create({
        allOperationsCount: 0
      }));
      mainServiceController.set('shouldStart', true);
      this.mock.returns(true);
      mainServiceController.silentStartAllServices();
      expect(testHelpers.findAjaxRequest('name', 'common.services.update')).to.be.undefined;
    });

    it("App.ajax.send should be called", function() {
      this.mockRouter.returns(Em.Object.create({
        allOperationsCount: 0
      }));
      mainServiceController.set('shouldStart', true);
      this.mock.returns(false);
      mainServiceController.silentStartAllServices();
      var args = testHelpers.filterAjaxRequests('name', 'common.services.update');
      expect(args[0][0]).to.be.eql({
        name: 'common.services.update',
        sender: mainServiceController,
        data: {
          context: App.BackgroundOperationsController.CommandContexts.START_ALL_SERVICES,
          ServiceInfo: {
            state: 'STARTED'
          }
        },
        success: 'silentCallSuccessCallback'
      });
      expect(mainServiceController.get('shouldStart')).to.be.false;
    });
  });

  describe("#silentCallSuccessCallback()", function () {
    var mock = {
      dataLoading: function () {
        return {
          done: function (callback) {
            callback(true);
          }
        }
      },
      showPopup: Em.K
    };

    beforeEach(function () {
      sinon.stub(App.router, 'get').returns(mock);
      sinon.spy(mock, 'showPopup');
    });
    afterEach(function () {
      App.router.get.restore();
      mock.showPopup.restore();
    });

    it("showPopup should be called", function () {
      mainServiceController.silentCallSuccessCallback();
      expect(mock.showPopup.calledOnce).to.be.true;
    });
  });

  describe("#allServicesCallSuccessCallback()", function () {
    var mock = {
      dataLoading: function () {
        return {
          done: function (callback) {
            callback(true);
          }
        }
      },
      showPopup: Em.K
    };

    beforeEach(function () {
      sinon.stub(App.router, 'get').returns(mock);
      sinon.spy(mock, 'showPopup');
    });
    afterEach(function () {
      App.router.get.restore();
      mock.showPopup.restore();
    });

    it("showPopup should be called", function () {
      var params = {
        query: Em.Object.create()
      };
      mainServiceController.allServicesCallSuccessCallback({}, {}, params);
      expect(mock.showPopup.calledOnce).to.be.true;
      expect(params.query.get('status')).to.be.equal('SUCCESS');
    });
  });

  describe("#restartAllRequiredSuccessCallback()", function () {
    var mock = {
      dataLoading: function () {
        return {
          done: function (callback) {
            callback(true);
          }
        }
      },
      showPopup: Em.K
    };

    beforeEach(function () {
      sinon.stub(App.router, 'get').returns(mock);
      sinon.spy(mock, 'showPopup');
    });
    afterEach(function () {
      App.router.get.restore();
      mock.showPopup.restore();
    });

    it("showPopup should be called", function () {
      mainServiceController.restartAllRequiredSuccessCallback();
      expect(mock.showPopup.calledOnce).to.be.true;
    });
  });

  describe('#restartHostComponents', function () {

    beforeEach(function () {
      this.mock = sinon.stub(App.HostComponent, 'find');
    });
    afterEach(function () {
      App.HostComponent.find.restore();
    });

    it('should make batch request to refresh YARN queues', function () {
      this.mock.returns([Em.Object.create({componentName: 'HIVE_SERVER_INTERACTIVE', staleConfigs: true}), Em.Object.create({componentName: 'RESOURCEMANAGER', staleConfigs: false})]);
      mainServiceController.restartHostComponents();
      expect(testHelpers.findAjaxRequest('name', 'common.batch.request_schedules')).not.to.be.undefined;
    });

    it('should make single request without refresh YARN queues', function () {
      this.mock.returns([Em.Object.create({componentName: 'HIVE_SERVER_INTERACTIVE', staleConfigs: false}), Em.Object.create({componentName: 'RESOURCEMANAGER', staleConfigs: false})]);
      mainServiceController.restartHostComponents();
      expect(testHelpers.findAjaxRequest('name', 'request.post')).not.to.be.undefined;
    });

  });
});