cps-rest/src/test/groovy/org/onap/cps/rest/utils/MultipartFileUtilSpec.groovy

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2020 Pantheon.tech
   4  *  ================================================================================
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License.
  15  *
  16  *  SPDX-License-Identifier: Apache-2.0
  17  *  ============LICENSE_END=========================================================
  18  */
  19
  20 package org.onap.cps.rest.utils
  21
  22 import org.onap.cps.spi.exceptions.ModelValidationException
  23 import org.springframework.mock.web.MockMultipartFile
  24 import spock.lang.Specification
  25
  26 class MultipartFileUtilSpec extends Specification {
  27
  28     def 'Extract yang resource from multipart file'() {
  29         given:
  30             def multipartFile = new MockMultipartFile("file", "filename.yang", "text/plain", "content".getBytes())
  31         when:
  32             def result = MultipartFileUtil.extractYangResourcesMap(multipartFile)
  33         then:
  34             assert result != null
  35             assert result.size() == 1
  36             assert result.get("filename.yang") == "content"
  37     }
  38
  39     def 'Extract yang resource from  file with invalid filename extension'() {
  40         given:
  41             def multipartFile = new MockMultipartFile("file", "filename.doc", "text/plain", "content".getBytes())
  42         when:
  43             MultipartFileUtil.extractYangResourcesMap(multipartFile)
  44         then:
  45             thrown(ModelValidationException)
  46     }
  47
  48 }