wfenginemgrservice/src/test/java/org/onap/workflow/common/RestClientUtilsTest.java

   1 /**\r
   2  * Copyright 2017 ZTE Corporation.\r
   3  *\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  *     http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 \r
  17 package org.onap.workflow.common;\r
  18 \r
  19 import java.io.File;\r
  20 import java.io.FileInputStream;\r
  21 import java.io.InputStream;\r
  22 \r
  23 import org.apache.http.HttpEntity;\r
  24 import org.apache.http.entity.ContentType;\r
  25 import org.junit.Assert;\r
  26 import org.junit.Test;\r
  27 \r
  28 /**\r
  29  *\r
  30  */\r
  31 public class RestClientUtilsTest {\r
  32 \r
  33   @Test\r
  34   public void testbuildMultipartRequest() {\r
  35     String filePath = System.getProperty("java.io.tmpdir");\r
  36     String fileName = "testfile";\r
  37     File file = new File(filePath + File.separator + fileName);\r
  38     InputStream is = null;\r
  39     try {\r
  40       if (!file.exists()) {\r
  41         file.createNewFile();\r
  42       }\r
  43       is = new FileInputStream(file);\r
  44       HttpEntity httpentity = RestClientUtils.buildMultipartRequest(is, fileName);\r
  45       Assert.assertTrue(httpentity.getContentType().getValue()\r
  46           .indexOf(ContentType.MULTIPART_FORM_DATA.getMimeType()) > -1);\r
  47     } catch (Exception e) {\r
  48       Assert.assertFalse (false);\r
  49     } finally {\r
  50       ToolUtil.closeInputStream(is);\r
  51       if (file.exists()) {\r
  52         file.delete();\r
  53       }\r
  54     }\r
  55   }\r
  56 }\r