datafilecollector/src/test/java/org/oran/datafile/model/SchemeTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2019-2023 Nordix Foundation. All rights reserved.
   4  * Copyright (C) 2020 Nokia. All rights reserved.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  * ============LICENSE_END=========================================================
  18  */
  19
  20 package org.oran.datafile.model;
  21
  22 import static org.junit.jupiter.api.Assertions.assertEquals;
  23 import static org.junit.jupiter.api.Assertions.assertThrows;
  24
  25 import org.junit.jupiter.api.Test;
  26 import org.oran.datafile.exceptions.DatafileTaskException;
  27 import org.oran.datafile.model.FileData.Scheme;
  28
  29 public class SchemeTest {
  30
  31     @Test
  32     public void shouldReturnSchemeForSupportedProtocol() throws DatafileTaskException {
  33         assertEquals(Scheme.FTPES, Scheme.getSchemeFromString("FTPES"));
  34         assertEquals(Scheme.SFTP, Scheme.getSchemeFromString("SFTP"));
  35         assertEquals(Scheme.HTTP, Scheme.getSchemeFromString("HTTP"));
  36         assertEquals(Scheme.HTTPS, Scheme.getSchemeFromString("HTTPS"));
  37     }
  38
  39     @Test
  40     public void shouldThrowExceptionForUnsupportedProtocol() {
  41         assertThrows(DatafileTaskException.class, () -> Scheme.getSchemeFromString("FTPS"));
  42     }
  43
  44     @Test
  45     public void shouldThrowExceptionForInvalidProtocol() {
  46         assertThrows(DatafileTaskException.class, () -> Scheme.getSchemeFromString("invalid"));
  47     }
  48 }