org.eclipse.hawkbit.mgmt.rest.resource.SMRessourceMisingMongoDbConnectionTest.java Source code

Introduction

Here is the source code for org.eclipse.hawkbit.mgmt.rest.resource.SMRessourceMisingMongoDbConnectionTest.java

Source

/**
 * Copyright (c) 2015 Bosch Software Innovations GmbH and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package org.eclipse.hawkbit.mgmt.rest.resource;

import static org.fest.assertions.api.Assertions.assertThat;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.fileUpload;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

import org.apache.commons.lang3.RandomStringUtils;
import org.eclipse.hawkbit.exception.SpServerError;
import org.eclipse.hawkbit.repository.model.SoftwareModule;
import org.eclipse.hawkbit.rest.AbstractRestIntegrationTestWithMongoDB;
import org.eclipse.hawkbit.rest.json.model.ExceptionInfo;
import org.eclipse.hawkbit.rest.util.MockMvcResultPrinter;
import org.junit.After;
import org.junit.Test;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.web.servlet.MvcResult;

import ru.yandex.qatools.allure.annotations.Description;
import ru.yandex.qatools.allure.annotations.Features;
import ru.yandex.qatools.allure.annotations.Stories;

/**
 * Tests {@link MgmtSoftwareModuleResource} in case of missing MongoDB
 * connection.
 *
 */
@Features("Component Tests - Management API")
@Stories("Download Resource")
public class SMRessourceMisingMongoDbConnectionTest extends AbstractRestIntegrationTestWithMongoDB {

    @Test
    @Description("Ensures that the correct error code is returned in case MongoDB unavailable.")
    public void missingMongoDbConnectionResultsInErrorAtUpload() throws Exception {
        mongodExecutable.stop();

        assertThat(softwareManagement.findSoftwareModulesAll(pageReq)).hasSize(0);
        assertThat(artifactManagement.countLocalArtifactsAll()).isEqualTo(0);
        SoftwareModule sm = entityFactory.generateSoftwareModule(
                softwareManagement.findSoftwareModuleTypeByKey("os"), "name 1", "version 1", null, null);
        sm = softwareManagement.createSoftwareModule(sm);
        assertThat(artifactManagement.countLocalArtifactsAll()).isEqualTo(0);

        // create test file
        final byte random[] = RandomStringUtils.random(5 * 1024).getBytes();
        final MockMultipartFile file = new MockMultipartFile("file", "origFilename", null, random);

        // upload
        final MvcResult mvcResult = mvc
                .perform(fileUpload("/rest/v1/softwaremodules/{smId}/artifacts", sm.getId()).file(file))
                .andDo(MockMvcResultPrinter.print()).andExpect(status().isInternalServerError()).andReturn();

        // check error result
        final ExceptionInfo exceptionInfo = ResourceUtility
                .convertException(mvcResult.getResponse().getContentAsString());
        assertThat(exceptionInfo.getErrorCode()).isEqualTo(SpServerError.SP_ARTIFACT_UPLOAD_FAILED.getKey());
        assertThat(exceptionInfo.getMessage()).isEqualTo(SpServerError.SP_ARTIFACT_UPLOAD_FAILED.getMessage());

        // ensure that the JPA transaction was rolled back
        assertThat(artifactManagement.countLocalArtifactsAll()).isEqualTo(0);

    }

    @After
    public void cleanCurrentCollection() {
        // not needed, mongodb is stopped already
    }

}