You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
3.3 KiB
78 lines
3.3 KiB
/* |
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
|
* contributor license agreements. See the NOTICE file distributed with |
|
* this work for additional information regarding copyright ownership. |
|
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.service.utils; |
|
|
|
import static org.mockito.ArgumentMatchers.anyString; |
|
|
|
import org.apache.dolphinscheduler.common.constants.Constants; |
|
import org.apache.dolphinscheduler.common.utils.OSUtils; |
|
import org.apache.dolphinscheduler.common.utils.PropertyUtils; |
|
|
|
import org.junit.jupiter.api.Assertions; |
|
import org.junit.jupiter.api.BeforeEach; |
|
import org.junit.jupiter.api.Test; |
|
import org.junit.jupiter.api.extension.ExtendWith; |
|
import org.mockito.MockedStatic; |
|
import org.mockito.Mockito; |
|
import org.mockito.MockitoAnnotations; |
|
import org.mockito.junit.jupiter.MockitoExtension; |
|
import org.slf4j.Logger; |
|
import org.slf4j.LoggerFactory; |
|
|
|
@ExtendWith(MockitoExtension.class) |
|
public class ProcessUtilsTest { |
|
|
|
private static final Logger logger = LoggerFactory.getLogger(ProcessUtils.class); |
|
|
|
@BeforeEach |
|
public void setUp() { |
|
MockitoAnnotations.initMocks(this); |
|
} |
|
|
|
@Test |
|
public void getPidsStr() throws Exception { |
|
int processId = 1; |
|
|
|
try (MockedStatic<OSUtils> mockedStaticOSUtils = Mockito.mockStatic(OSUtils.class)) { |
|
mockedStaticOSUtils.when(() -> OSUtils.exeCmd(anyString())).thenReturn(null); |
|
String pidList = ProcessUtils.getPidsStr(processId); |
|
Assertions.assertEquals("", pidList); |
|
} |
|
} |
|
|
|
@Test |
|
public void testGetKerberosInitCommand() { |
|
try (MockedStatic<PropertyUtils> mockedStaticPropertyUtils = Mockito.mockStatic(PropertyUtils.class)) { |
|
mockedStaticPropertyUtils |
|
.when(() -> PropertyUtils.getBoolean(Constants.HADOOP_SECURITY_AUTHENTICATION_STARTUP_STATE, false)) |
|
.thenReturn(true); |
|
mockedStaticPropertyUtils.when(() -> PropertyUtils.getString(Constants.JAVA_SECURITY_KRB5_CONF_PATH)) |
|
.thenReturn("/etc/krb5.conf"); |
|
mockedStaticPropertyUtils.when(() -> PropertyUtils.getString(Constants.LOGIN_USER_KEY_TAB_PATH)) |
|
.thenReturn("/etc/krb5.keytab"); |
|
mockedStaticPropertyUtils.when(() -> PropertyUtils.getString(Constants.LOGIN_USER_KEY_TAB_USERNAME)) |
|
.thenReturn("test@DS.COM"); |
|
Assertions.assertNotEquals("", ProcessUtils.getKerberosInitCommand()); |
|
mockedStaticPropertyUtils |
|
.when(() -> PropertyUtils.getBoolean(Constants.HADOOP_SECURITY_AUTHENTICATION_STARTUP_STATE, false)) |
|
.thenReturn(false); |
|
Assertions.assertEquals("", ProcessUtils.getKerberosInitCommand()); |
|
} |
|
} |
|
|
|
}
|
|
|