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.
173 lines
7.9 KiB
173 lines
7.9 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.api.controller; |
|
|
|
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get; |
|
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post; |
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content; |
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
|
|
|
import org.apache.dolphinscheduler.api.enums.Status; |
|
import org.apache.dolphinscheduler.api.utils.Result; |
|
import org.apache.dolphinscheduler.common.utils.JSONUtils; |
|
import org.apache.dolphinscheduler.dao.entity.User; |
|
|
|
import org.junit.jupiter.api.Assertions; |
|
import org.junit.jupiter.api.Test; |
|
import org.slf4j.Logger; |
|
import org.slf4j.LoggerFactory; |
|
import org.springframework.http.MediaType; |
|
import org.springframework.test.web.servlet.MvcResult; |
|
import org.springframework.util.LinkedMultiValueMap; |
|
import org.springframework.util.MultiValueMap; |
|
|
|
/** |
|
* k8s namespace controller test |
|
*/ |
|
public class K8sNamespaceControllerTest extends AbstractControllerTest { |
|
|
|
private static final String NAMESPACE_CREATE_STRING = "namespace1"; |
|
private static final Logger logger = LoggerFactory.getLogger(K8sNamespaceControllerTest.class); |
|
protected User user; |
|
|
|
@Test |
|
public void queryProjectListPaging() throws Exception { |
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("searchVal", ""); |
|
paramsMap.add("pageNo", "1"); |
|
paramsMap.add("pageSize", "20"); |
|
|
|
MvcResult mvcResult = mockMvc.perform(get("/k8s-namespace") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue()); |
|
} |
|
|
|
@Test |
|
public void createNamespace() throws Exception { |
|
|
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("namespace", NAMESPACE_CREATE_STRING); |
|
paramsMap.add("clusterCode", "0"); |
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/k8s-namespace") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isCreated()) // it can |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
|
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue());// because we not have a k8s |
|
// cluster in test env |
|
logger.info("create queue return result:{}", mvcResult.getResponse().getContentAsString()); |
|
} |
|
|
|
@Test |
|
public void verifyNamespace() throws Exception { |
|
// queue value exist |
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("namespace", "NAMESPACE_CREATE_STRING"); |
|
paramsMap.add("clusterCode", "100"); |
|
|
|
// success |
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/k8s-namespace/verify") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue()); |
|
logger.info(mvcResult.getResponse().getContentAsString()); |
|
logger.info("verify namespace return result:{}", mvcResult.getResponse().getContentAsString()); |
|
|
|
// error |
|
paramsMap.clear(); |
|
paramsMap.add("namespace", null); |
|
paramsMap.add("clusterCode", "100"); |
|
mvcResult = mockMvc.perform(post("/k8s-namespace/verify") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.VERIFY_K8S_NAMESPACE_ERROR.getCode(), result.getCode().intValue()); |
|
logger.info(mvcResult.getResponse().getContentAsString()); |
|
logger.info("verify namespace return result:{}", mvcResult.getResponse().getContentAsString()); |
|
} |
|
|
|
@Test |
|
public void deleteNamespaceById() throws Exception { |
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("id", "1"); |
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/k8s-namespace/delete") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
|
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue());// there is no k8s cluster in |
|
// test env |
|
logger.info(mvcResult.getResponse().getContentAsString()); |
|
} |
|
|
|
@Test |
|
public void testQueryUnauthorizedNamespace() throws Exception { |
|
|
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("userId", "1"); |
|
|
|
MvcResult mvcResult = mockMvc.perform(get("/k8s-namespace/unauth-namespace") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
|
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue()); |
|
logger.info(mvcResult.getResponse().getContentAsString()); |
|
} |
|
|
|
@Test |
|
public void testQueryAuthorizedNamespace() throws Exception { |
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>(); |
|
paramsMap.add("userId", "1"); |
|
|
|
MvcResult mvcResult = mockMvc.perform(get("/k8s-namespace/authed-namespace") |
|
.header(SESSION_ID, sessionId) |
|
.params(paramsMap)) |
|
.andExpect(status().isOk()) |
|
.andExpect(content().contentType(MediaType.APPLICATION_JSON)) |
|
.andReturn(); |
|
|
|
Result result = JSONUtils.parseObject(mvcResult.getResponse().getContentAsString(), Result.class); |
|
Assertions.assertEquals(Status.SUCCESS.getCode(), result.getCode().intValue()); |
|
logger.info(mvcResult.getResponse().getContentAsString()); |
|
} |
|
}
|
|
|