Java tutorial
/*************************GO-LICENSE-START********************************* * Copyright 2014 ThoughtWorks, Inc. * * Licensed 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. *************************GO-LICENSE-END***********************************/ package com.thoughtworks.go.server.service; import com.thoughtworks.go.config.*; import com.thoughtworks.go.helper.PipelineConfigMother; import com.thoughtworks.go.server.dao.PipelineSqlMapDao; import com.thoughtworks.go.server.domain.Username; import com.thoughtworks.go.server.service.result.HttpLocalizedOperationResult; import org.apache.commons.httpclient.HttpStatus; import org.junit.Before; import org.junit.Test; import static org.hamcrest.Matchers.is; import static org.junit.Assert.assertThat; import static org.mockito.Matchers.any; import static org.mockito.Matchers.eq; import static org.mockito.Mockito.doThrow; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.never; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; public class PipelinePauseServiceTest { private PipelineSqlMapDao pipelineDao; public PipelinePauseService pipelinePauseService; private GoConfigService goConfigService; private GoConfigFileDao goConfigFileDao; private SecurityService securityService; private static final String VALID_PIPELINE = "some-pipeline"; private static final Username VALID_USER = new Username(new CaseInsensitiveString("admin")); private static final String INVALID_PIPELINE = "nonexistent-pipeline"; private static final Username INVALID_USER = new Username(new CaseInsensitiveString("someone-who-not-operate")); @Before public void setUp() throws Exception { pipelineDao = mock(PipelineSqlMapDao.class); goConfigFileDao = mock(GoConfigFileDao.class); goConfigService = new GoConfigService(goConfigFileDao, null, (GoConfigMigration) null, null, null, null, null, null, null); securityService = mock(SecurityService.class); pipelinePauseService = new PipelinePauseService(pipelineDao, goConfigService, securityService); } private void setUpValidPipelineWithAuth() { Authorization authorization = new Authorization( new OperationConfig(new AdminUser(VALID_USER.getUsername()))); CruiseConfig cruiseConfig = new BasicCruiseConfig(new BasicPipelineConfigs("my_group", authorization, PipelineConfigMother.pipelineConfig(VALID_PIPELINE))); when(goConfigFileDao.load()).thenReturn(cruiseConfig); when(securityService.hasOperatePermissionForGroup(eq(VALID_USER.getUsername()), any(String.class))) .thenReturn(true); } private void setUpValidPipelineWithInvalidAuth() { Authorization authorization = new Authorization( new OperationConfig(new AdminUser(INVALID_USER.getUsername()))); CruiseConfig cruiseConfig = new BasicCruiseConfig(new BasicPipelineConfigs("my_group", authorization, PipelineConfigMother.pipelineConfig(VALID_PIPELINE))); when(goConfigFileDao.load()).thenReturn(cruiseConfig); when(securityService.hasOperatePermissionForGroup(eq(INVALID_USER.getUsername()), any(String.class))) .thenReturn(false); } @Test public void shouldPausePipeline() throws Exception { setUpValidPipelineWithAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); pipelinePauseService.pause(VALID_PIPELINE, "cause", VALID_USER, result); verify(pipelineDao).pause(VALID_PIPELINE, "cause", VALID_USER.getUsername().toString()); assertThat(result.isSuccessful(), is(true)); assertThat(result.httpCode(), is(HttpStatus.SC_OK)); } @Test public void shouldUnPausePipeline() throws Exception { setUpValidPipelineWithAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); pipelinePauseService.unpause(VALID_PIPELINE, VALID_USER, result); verify(pipelineDao).unpause(VALID_PIPELINE); assertThat(result.isSuccessful(), is(true)); assertThat(result.httpCode(), is(HttpStatus.SC_OK)); } @Test public void shouldPopulateHttpResult404WhenPipelineIsNotFoundForPause() throws Exception { HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); when(goConfigFileDao.load()).thenReturn(new BasicCruiseConfig()); pipelinePauseService.pause(INVALID_PIPELINE, "cause", VALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_NOT_FOUND)); verify(pipelineDao, never()).pause(INVALID_PIPELINE, "cause", "admin"); } @Test public void shouldPopulateHttpResult404WhenPipelineIsNotFoundForUnpause() throws Exception { HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); when(goConfigFileDao.load()).thenReturn(new BasicCruiseConfig()); pipelinePauseService.unpause(INVALID_PIPELINE, VALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_NOT_FOUND)); verify(pipelineDao, never()).unpause(INVALID_PIPELINE); } @Test public void shouldPopulateHttpResult500WhenPipelinePauseResultsInAnError() throws Exception { setUpValidPipelineWithAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); doThrow(new RuntimeException("Failed to pause")).when(pipelineDao).pause(VALID_PIPELINE, "cause", VALID_USER.getUsername().toString()); pipelinePauseService.pause(VALID_PIPELINE, "cause", VALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_INTERNAL_SERVER_ERROR)); verify(pipelineDao).pause(VALID_PIPELINE, "cause", VALID_USER.getUsername().toString()); } @Test public void shouldPopulateHttpResult500WhenPipelineUnPauseResultsInAnError() throws Exception { setUpValidPipelineWithAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); doThrow(new RuntimeException("Failed to unpause")).when(pipelineDao).unpause(VALID_PIPELINE); pipelinePauseService.unpause(VALID_PIPELINE, VALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_INTERNAL_SERVER_ERROR)); verify(pipelineDao).unpause(VALID_PIPELINE); } @Test public void shouldPopulateHttpResult401WhenPipelineIsNotAuthorizedForPausing() throws Exception { setUpValidPipelineWithInvalidAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); pipelinePauseService.pause(VALID_PIPELINE, "cause", INVALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_UNAUTHORIZED)); verify(pipelineDao, never()).pause(VALID_PIPELINE, "cause", "admin"); } @Test public void shouldPopulateHttpResult401WhenPipelineIsNotAuthorizedForUnPausing() throws Exception { setUpValidPipelineWithInvalidAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); pipelinePauseService.unpause(VALID_PIPELINE, INVALID_USER, result); assertThat(result.isSuccessful(), is(false)); assertThat(result.httpCode(), is(HttpStatus.SC_UNAUTHORIZED)); verify(pipelineDao, never()).unpause(VALID_PIPELINE); } @Test public void shouldPausePipelineEvenWhenPauseCauseIsNull() throws Exception { setUpValidPipelineWithAuth(); HttpLocalizedOperationResult result = new HttpLocalizedOperationResult(); pipelinePauseService.pause(VALID_PIPELINE, null, VALID_USER, result); assertThat(result.isSuccessful(), is(true)); assertThat(result.httpCode(), is(HttpStatus.SC_OK)); verify(pipelineDao).pause(VALID_PIPELINE, "", VALID_USER.getUsername().toString()); } }