xiangzihao
3 years ago
committed by
GitHub
6 changed files with 338 additions and 5 deletions
@ -0,0 +1,187 @@ |
|||||||
|
/* |
||||||
|
* 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.e2e.cases; |
||||||
|
|
||||||
|
import lombok.SneakyThrows; |
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
import static org.awaitility.Awaitility.await; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.e2e.core.Constants; |
||||||
|
import org.apache.dolphinscheduler.e2e.core.DolphinScheduler; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.LoginPage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.resource.FunctionManagePage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.resource.ResourcePage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.resource.UdfManagePage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.security.SecurityPage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.security.TenantPage; |
||||||
|
import org.apache.dolphinscheduler.e2e.pages.security.UserPage; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.io.FileOutputStream; |
||||||
|
import java.io.InputStream; |
||||||
|
import java.net.URL; |
||||||
|
import java.net.URLConnection; |
||||||
|
import java.nio.file.Files; |
||||||
|
import java.nio.file.Path; |
||||||
|
import java.time.Duration; |
||||||
|
import java.util.Comparator; |
||||||
|
|
||||||
|
import org.junit.jupiter.api.AfterAll; |
||||||
|
import org.junit.jupiter.api.BeforeAll; |
||||||
|
import org.junit.jupiter.api.Order; |
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
import org.openqa.selenium.By; |
||||||
|
import org.openqa.selenium.WebElement; |
||||||
|
import org.openqa.selenium.remote.RemoteWebDriver; |
||||||
|
import org.openqa.selenium.support.ui.ExpectedConditions; |
||||||
|
import org.openqa.selenium.support.ui.WebDriverWait; |
||||||
|
|
||||||
|
@DolphinScheduler(composeFiles = "docker/file-manage/docker-compose.yaml") |
||||||
|
public class FunctionManageE2ETest { |
||||||
|
private static RemoteWebDriver browser; |
||||||
|
|
||||||
|
private static final String tenant = System.getProperty("user.name"); |
||||||
|
|
||||||
|
private static final String user = "admin"; |
||||||
|
|
||||||
|
private static final String password = "dolphinscheduler123"; |
||||||
|
|
||||||
|
private static final String email = "admin@gmail.com"; |
||||||
|
|
||||||
|
private static final String phone = "15800000000"; |
||||||
|
|
||||||
|
private static final String testUdfFunctionName = "test_function"; |
||||||
|
|
||||||
|
private static final String testRenameUdfFunctionName = "test_rename_function"; |
||||||
|
|
||||||
|
private static final String testUploadUdfFileName = "hive-jdbc-3.1.2.jar"; |
||||||
|
|
||||||
|
private static final String testClassName = "org.dolphinscheduler.UdfTest"; |
||||||
|
|
||||||
|
private static final String testDescription = "test_description"; |
||||||
|
|
||||||
|
private static final Path testUploadUdfFilePath = Constants.HOST_TMP_PATH.resolve(testUploadUdfFileName); |
||||||
|
|
||||||
|
@BeforeAll |
||||||
|
@SneakyThrows |
||||||
|
public static void setup() { |
||||||
|
TenantPage tenantPage = new LoginPage(browser) |
||||||
|
.login(user, password) |
||||||
|
.create(tenant); |
||||||
|
|
||||||
|
await().untilAsserted(() -> assertThat(tenantPage.tenantList()) |
||||||
|
.as("Tenant list should contain newly-created tenant") |
||||||
|
.extracting(WebElement::getText) |
||||||
|
.anyMatch(it -> it.contains(tenant))); |
||||||
|
|
||||||
|
downloadFile("https://repo1.maven.org/maven2/org/apache/hive/hive-jdbc/3.1.2/hive-jdbc-3.1.2.jar", testUploadUdfFilePath.toFile().getAbsolutePath()); |
||||||
|
|
||||||
|
UdfManagePage udfManagePage = tenantPage.goToNav(SecurityPage.class) |
||||||
|
.goToTab(UserPage.class) |
||||||
|
.update(user, user, password, email, phone) |
||||||
|
.goToNav(ResourcePage.class) |
||||||
|
.goToTab(UdfManagePage.class) |
||||||
|
.uploadFile(testUploadUdfFilePath.toFile().getAbsolutePath()); |
||||||
|
|
||||||
|
new WebDriverWait(browser, 10).until(ExpectedConditions.invisibilityOfElementLocated(By.id("fileUpdateDialog"))); |
||||||
|
|
||||||
|
udfManagePage.goToNav(ResourcePage.class) |
||||||
|
.goToTab(FunctionManagePage.class); |
||||||
|
} |
||||||
|
|
||||||
|
@AfterAll |
||||||
|
@SneakyThrows |
||||||
|
public static void cleanup() { |
||||||
|
Files.walk(Constants.HOST_CHROME_DOWNLOAD_PATH) |
||||||
|
.sorted(Comparator.reverseOrder()) |
||||||
|
.map(Path::toFile) |
||||||
|
.forEach(File::delete); |
||||||
|
|
||||||
|
Files.deleteIfExists(testUploadUdfFilePath); |
||||||
|
} |
||||||
|
|
||||||
|
static void downloadFile(String downloadUrl, String filePath) throws Exception { |
||||||
|
int byteRead; |
||||||
|
|
||||||
|
URL url = new URL(downloadUrl); |
||||||
|
|
||||||
|
URLConnection conn = url.openConnection(); |
||||||
|
InputStream inStream = conn.getInputStream(); |
||||||
|
FileOutputStream fs = new FileOutputStream(filePath); |
||||||
|
|
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
while ((byteRead = inStream.read(buffer)) != -1) { |
||||||
|
fs.write(buffer, 0, byteRead); |
||||||
|
} |
||||||
|
|
||||||
|
inStream.close(); |
||||||
|
fs.close(); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
@Order(10) |
||||||
|
void testCreateUdfFunction() { |
||||||
|
FunctionManagePage page = new FunctionManagePage(browser); |
||||||
|
|
||||||
|
new WebDriverWait(page.driver(), 10) |
||||||
|
.until(ExpectedConditions.urlContains("/#/resource/func")); |
||||||
|
|
||||||
|
page.createUdfFunction(testUdfFunctionName, testClassName, testUploadUdfFileName, testDescription); |
||||||
|
|
||||||
|
await().untilAsserted(() -> assertThat(page.functionList()) |
||||||
|
.as("Function list should contain newly-created file") |
||||||
|
.extracting(WebElement::getText) |
||||||
|
.anyMatch(it -> it.contains(testUdfFunctionName))); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
@Order(20) |
||||||
|
void testRenameUdfFunction() { |
||||||
|
FunctionManagePage page = new FunctionManagePage(browser); |
||||||
|
|
||||||
|
browser.navigate().refresh(); |
||||||
|
|
||||||
|
page.renameUdfFunction(testUdfFunctionName, testRenameUdfFunctionName); |
||||||
|
|
||||||
|
await().pollDelay(Duration.ofSeconds(2)).untilAsserted(() -> assertThat(page.functionList()) |
||||||
|
.as("Function list should contain newly-created file") |
||||||
|
.extracting(WebElement::getText) |
||||||
|
.anyMatch(it -> it.contains(testRenameUdfFunctionName))); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
@Order(30) |
||||||
|
void testDeleteUdfFunction() { |
||||||
|
FunctionManagePage page = new FunctionManagePage(browser); |
||||||
|
|
||||||
|
page.deleteUdfFunction(testRenameUdfFunctionName); |
||||||
|
|
||||||
|
await().untilAsserted(() -> { |
||||||
|
browser.navigate().refresh(); |
||||||
|
|
||||||
|
assertThat( |
||||||
|
page.functionList() |
||||||
|
).noneMatch( |
||||||
|
it -> it.getText().contains(testRenameUdfFunctionName) |
||||||
|
); |
||||||
|
}); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue