MaterialDatabaseDependencyUpdaterIntegrationTest.java example

Explorer
gocd-master
/*************************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.materials;

import com.thoughtworks.go.config.CaseInsensitiveString;
import com.thoughtworks.go.config.PipelineConfig;
import com.thoughtworks.go.config.materials.dependency.DependencyMaterial;
import com.thoughtworks.go.domain.Pipeline;
import com.thoughtworks.go.domain.materials.Modification;
import com.thoughtworks.go.helper.PipelineConfigMother;
import com.thoughtworks.go.server.cache.GoCache;
import com.thoughtworks.go.server.dao.DatabaseAccessHelper;
import com.thoughtworks.go.server.persistence.MaterialRepository;
import com.thoughtworks.go.server.service.GoConfigService;
import com.thoughtworks.go.util.GoConfigFileHelper;
import com.thoughtworks.go.util.TimeProvider;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import static org.hamcrest.core.Is.is;
import static org.hamcrest.core.IsNot.not;
import static org.hamcrest.core.IsNull.nullValue;
import static org.junit.Assert.assertThat;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {
        "classpath:WEB-INF/applicationContext-global.xml",
        "classpath:WEB-INF/applicationContext-dataLocalAccess.xml",
        "classpath:WEB-INF/applicationContext-acegi-security.xml"
})
public class MaterialDatabaseDependencyUpdaterIntegrationTest  {

    @Autowired private DatabaseAccessHelper dbHelper;
    @Autowired private MaterialRepository materialRepository;
    @Autowired private GoCache goCache;
    @Autowired private GoConfigService goConfigService;
    @Autowired private MaterialDatabaseUpdater updater;
    private GoConfigFileHelper configHelper = new GoConfigFileHelper();

    @Before public void setUp() throws Exception {
        dbHelper.onSetUp();
        goCache.clear();
        configHelper.onSetUp();
    }

    @After public void tearDown() throws Exception {
        dbHelper.onTearDown();
        configHelper.onTearDown();
    }

    @Test
    public void shouldInsertAllRunsAfterLastKnownOfUpstreamStage() throws Exception {
        PipelineConfig mingleConfig = PipelineConfigMother.createPipelineConfig("acceptance", "stage-name", "job");
        goConfigService.addPipeline(mingleConfig, "pipeline-group");

        Pipeline passed1 = dbHelper.schedulePipeline(mingleConfig, new TimeProvider());
        dbHelper.passStage(passed1.getStages().get(0));

        DependencyMaterial dependencyMaterial = new DependencyMaterial(new CaseInsensitiveString("acceptance"), new CaseInsensitiveString("stage-name"));
        String revision1 = String.format("acceptance/%s/stage-name/1", passed1.getCounter());

        updater.updateMaterial(dependencyMaterial);
        assertThat(materialRepository.findModificationWithRevision(dependencyMaterial, revision1), is(not(nullValue())));

        Pipeline cancelledPipeline = dbHelper.schedulePipeline(mingleConfig, new TimeProvider());
        dbHelper.cancelStage(cancelledPipeline.getStages().get(0));

        Pipeline passed2 = dbHelper.schedulePipeline(mingleConfig, new TimeProvider());
        dbHelper.passStage(passed2.getStages().get(0));

        updater.updateMaterial(dependencyMaterial);
        assertThat(materialRepository.findModificationWithRevision(dependencyMaterial, String.format("acceptance/%s/stage-name/1", cancelledPipeline.getCounter())), is(nullValue()));

        Modification modification1 = materialRepository.findModificationWithRevision(dependencyMaterial, revision1);
        assertThat(modification1, is(not(nullValue())));
        assertThat(modification1.getRevision(), is(revision1));
        assertThat(modification1.getPipelineLabel(), is(passed1.getCounter().toString()));
        assertThat(modification1.getPipelineId(), is(passed1.getId()));

        String revision2 = String.format("acceptance/%s/stage-name/1", passed2.getCounter());
        Modification modification2 = materialRepository.findModificationWithRevision(dependencyMaterial, revision2);
        assertThat(modification2, is(not(nullValue())));
        assertThat(modification2.getRevision(), is(revision2));
        assertThat(modification2.getPipelineLabel(), is(passed2.getCounter().toString()));
        assertThat(modification2.getPipelineId(), is(passed2.getId()));
    }
}