code: /*******************************************************************************
   * Copyright 2018 572682
   * 
   * 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.
   ******************************************************************************/
  package us.dot.its.jpo.ode.upload;
  
  import static org.assertj.core.api.Assertions.assertThat;
  import static org.mockito.BDDMockito.given;
  import static org.mockito.BDDMockito.then;
  import static org.mockito.Matchers.any;
  
  import org.junit.Test;
  import org.junit.Ignore;
  import org.junit.runner.RunWith;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.boot.test.context.SpringBootTest;
  import org.springframework.boot.test.mock.mockito.MockBean;
  import org.springframework.boot.test.web.client.TestRestTemplate;
  import org.springframework.boot.web.server.LocalServerPort;
  import org.springframework.core.io.ClassPathResource;
  import org.springframework.http.HttpHeaders;
  import org.springframework.http.HttpStatus;
  import org.springframework.http.ResponseEntity;
  import org.springframework.test.context.junit4.SpringRunner;
  import org.springframework.util.LinkedMultiValueMap;
  import org.springframework.util.MultiValueMap;
  import org.springframework.web.multipart.MultipartFile;
  
  import us.dot.its.jpo.ode.storage.StorageService;
  
  @RunWith(SpringRunner.class)
  @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
  public class FileUploadIntegrationTests {
  
      @Autowired
      private TestRestTemplate restTemplate;
  
      @MockBean
      private StorageService storageService;
  
      @LocalServerPort
      private int port;
  
      @Ignore
      @Test
      public void shouldUploadFile() throws Exception {
          ClassPathResource resource = new ClassPathResource("testupload.txt", getClass());
  
          MultiValueMap<String, Object> map = new LinkedMultiValueMap<String, Object>();
          map.add("file", resource);
          ResponseEntity<String> response = this.restTemplate.postForEntity("/", map, String.class);
  
          assertThat(response.getStatusCode()).isEqualByComparingTo(HttpStatus.FOUND);
          assertThat(response.getHeaders().getLocation().toString()).startsWith("http://localhost:" + this.port + "/");
          then(storageService).should().store(any(MultipartFile.class), "obulog");
      }
  
      @Ignore
      @Test
      public void shouldDownloadFile() throws Exception {
          ClassPathResource resource = new ClassPathResource("testupload.txt", getClass());
          given(this.storageService.loadAsResource("testupload.txt")).willReturn(resource);
  
          ResponseEntity<String> response = this.restTemplate
                  .getForEntity("/files/{filename}", String.class, "testupload.txt");
  
          assertThat(response.getStatusCodeValue()).isEqualTo(200);
          assertThat(response.getHeaders().getFirst(HttpHeaders.CONTENT_DISPOSITION))
                  .isEqualTo("attachment; filename=\"testupload.txt\"");
          assertThat(response.getBody()).isEqualTo("Spring Framework");
      }
  
  }
  