|
@@ -0,0 +1,73 @@
|
|
|
+/*
|
|
|
+ * 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.common.utils;
|
|
|
+
|
|
|
+import org.apache.dolphinscheduler.spi.enums.ResourceType;
|
|
|
+
|
|
|
+import org.junit.Assert;
|
|
|
+import org.junit.Test;
|
|
|
+import org.junit.runner.RunWith;
|
|
|
+import org.powermock.api.mockito.PowerMockito;
|
|
|
+import org.powermock.core.classloader.annotations.PrepareForTest;
|
|
|
+import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
|
|
|
+import org.powermock.modules.junit4.PowerMockRunner;
|
|
|
+import org.slf4j.Logger;
|
|
|
+import org.slf4j.LoggerFactory;
|
|
|
+
|
|
|
+/**
|
|
|
+ * hadoop utils test
|
|
|
+ */
|
|
|
+@RunWith(PowerMockRunner.class)
|
|
|
+@PrepareForTest(value = {HadoopUtils.class})
|
|
|
+@SuppressStaticInitializationFor("org.apache.dolphinscheduler.common.utils.HttpUtils")
|
|
|
+public class HadoopUtilsTest {
|
|
|
+ private static final Logger logger = LoggerFactory.getLogger(HadoopUtilsTest.class);
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void getHdfsTenantDir() {
|
|
|
+ logger.info(HadoopUtils.getHdfsTenantDir("1234"));
|
|
|
+ Assert.assertTrue(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void getHdfsUdfFileName() {
|
|
|
+ logger.info(HadoopUtils.getHdfsUdfFileName("admin", "file_name"));
|
|
|
+ Assert.assertTrue(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void getHdfsResourceFileName() {
|
|
|
+ logger.info(HadoopUtils.getHdfsResourceFileName("admin", "file_name"));
|
|
|
+ Assert.assertTrue(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void getHdfsFileName() {
|
|
|
+ logger.info(HadoopUtils.getHdfsFileName(ResourceType.FILE, "admin", "file_name"));
|
|
|
+ Assert.assertTrue(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void getAppAddress() {
|
|
|
+ PowerMockito.mockStatic(HttpUtils.class);
|
|
|
+ PowerMockito.when(HttpUtils.get("http://ds1:8088/ws/v1/cluster/info")).thenReturn("{\"clusterInfo\":{\"state\":\"STARTED\",\"haState\":\"ACTIVE\"}}");
|
|
|
+ logger.info(HadoopUtils.getAppAddress("http://ds1:8088/ws/v1/cluster/apps/%s", "ds1,ds2"));
|
|
|
+ Assert.assertTrue(true);
|
|
|
+ }
|
|
|
+
|
|
|
+}
|