我找到了这个问题的解决方案。它适用于我的Selenium GRID和远程Chrome浏览器。
首先,我将未打包的ModHeader(版本1.2.4)扩展程序存储在我的项目资源中。它看起来是这样的:
![enter image description here](https://istack.dev59.com/1gNsr.webp)
如果我需要在Chrome中修改标题,我会执行以下步骤:
1)将资源中的扩展文件夹解压缩到临时文件夹中
2)在header.json中设置标题键和值
3)使用Java将此扩展打包为zip文件
4)将zip文件添加到ChromeOptions中
public static IDriver getDriverWithCustomHeader(List<HeaderElement> headerList) {
Logger.info(StringUtils.buildString("Create new instance of Driver with header."));
IDriver driver;
DesiredCapabilities capabilities;
switch (GlobalConfig.getInstance().getDriverType()) {
case CHROME:
String unpackedExtensionPath = FileUtils.getResourcePath("chrome_extension", true);
FileUtils.writeToJson(StringUtils.buildString(unpackedExtensionPath, File.separator, "header.json"), headerList);
String crxExtensionPath = ZipUtils.packZipWithNameOfFolder(unpackedExtensionPath, "crx");
capabilities = CapabilityFactory.getChromeCapabilitiesWithExtension(crxExtensionPath);
driver = new AppiumDriver(GlobalConfig.getInstance().getHost(), GlobalConfig.getInstance().getPort(),
capabilities);
break;
default:
throw new CommonTestRuntimeException("Unsupported Driver Type for changing head args.");
}
drivers.add(driver);
if (defaultDriver != null) {
closeDefaultDriver();
}
defaultDriver.set(driver);
return driver;
}
文件工具类
public static String getResourcePath(String resourceName, boolean isDir) {
String jarFileName = new File(FileUtils.class.getClassLoader().getResource(resourceName).getPath()).getAbsolutePath()
.replaceAll("(!|file:\\\\)", "");
if (!(jarFileName.contains(".jar"))) {
return getResourcePath(resourceName);
}
if (isDir) {
return getDirPath(resourceName);
}
return getFilePath(resourceName);
}
private static String getResourcePath(String resourceName) {
String resourcePath = FileUtils.class.getClassLoader().getResource(resourceName).getPath();
if (platformIsWindows()) {
resourcePath = resourcePath.substring(1);
}
return resourcePath;
}
private static boolean platformIsWindows() {
boolean platformIsWindows = (File.separatorChar == '\\') ? true : false;
return platformIsWindows;
}
private static String getDirPath(String dirName) {
JarFile jarFile = null;
if (directoryPath.get() == null) {
directoryPath.set(Files.createTempDir().getAbsolutePath());
}
if (!new File(directoryPath.get() + File.separator + dirName.replaceAll("/", "")).exists()) {
try {
List<JarEntry> dirEntries = new ArrayList<JarEntry>();
File directory = null;
String jarFileName = new File(FileUtils.class.getClassLoader().getResource(dirName).getPath()).getParent()
.replaceAll("(!|file:\\\\)", "").replaceAll("(!|file:)", "");
jarFile = new JarFile(URLDecoder.decode(jarFileName, "UTF-8"));
Enumeration<JarEntry> entries = jarFile.entries();
while (entries.hasMoreElements()) {
JarEntry jarEntry = entries.nextElement();
if (jarEntry.getName().startsWith(dirName)) {
if (jarEntry.getName().replaceAll("/", "").equals(dirName.replaceAll("/", ""))) {
directory = new File(directoryPath.get() + File.separator + dirName.replaceAll("/", ""));
directory.mkdirs();
} else
dirEntries.add(jarEntry);
}
}
if (directory == null) {
throw new CommonTestRuntimeException(StringUtils.buildString("There is no directory ", dirName,
"in the jar file"));
}
for (JarEntry dirEntry : dirEntries) {
if (!dirEntry.isDirectory()) {
File dirFile = new File(directory.getParent() + File.separator + dirEntry.getName());
dirFile.createNewFile();
convertStreamToFile(dirEntry.getName(), dirFile);
} else {
File dirFile = new File(directory.getParent() + File.separator + dirEntry.getName());
dirFile.mkdirs();
}
}
return directory.getAbsolutePath();
} catch (IOException ex) {
ex.printStackTrace();
} finally {
try {
jarFile.close();
} catch (Exception e) {
e.printStackTrace();
}
}
throw new CommonTestRuntimeException("There are problems in creation files in directory " + directoryPath);
} else {
return directoryPath.get() + File.separator + dirName.replaceAll("/", "");
}
}
private static String getFilePath(String fileName) {
try {
String[] fileType = fileName.split("\\.");
int typeIndex = fileType.length;
File file = File.createTempFile(StringUtils.generateRandomString("temp"),
StringUtils.buildString(".", fileType[typeIndex - 1]));
file.deleteOnExit();
convertStreamToFile(fileName, file);
return file.getAbsolutePath();
} catch (IOException e) {
e.printStackTrace();
}
throw new CommonTestRuntimeException("Impossible to get file path");
}
private static void convertStreamToFile(String resourceFileName, File file) throws IOException {
try (InputStream in = FileUtils.class.getClassLoader().getResourceAsStream(resourceFileName);
BufferedReader reader = new BufferedReader(new InputStreamReader(in, "UTF8"));
FileOutputStream fos = new FileOutputStream(file);
OutputStreamWriter fileOutputStreamWriter = new OutputStreamWriter(fos, "UTF8");
BufferedWriter fileWriter = new BufferedWriter(fileOutputStreamWriter);
) {
String line = null;
while ((line = reader.readLine()) != null) {
fileWriter.write(line + "\n");
}
}
}
public static void writeToJson(String jsonFilePath, Object object) {
try {
Gson gson = new Gson();
FileWriter fileWriter = new FileWriter(jsonFilePath);
fileWriter.write(gson.toJson(object));
fileWriter.close();
} catch (IOException e) {
e.printStackTrace();
}
}
ZipUtils
public static String packZipWithNameOfFolder(String folder, String extension) {
String outZipPath = folder + "." + extension;
try {
try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(outZipPath))) {
File file = new File(folder);
doZip(file, zos);
}
} catch (IOException e) {
throw new CommonTestRuntimeException("Fail of packaging of folder. ", e);
}
return outZipPath;
}
private static void doZip(File dir, ZipOutputStream out) throws IOException {
for (File f: dir.listFiles()) {
if (f.isDirectory()) {
doZip(f, out);
} else {
out.putNextEntry(new ZipEntry(f.getName()));
try (FileInputStream in = new FileInputStream(f)) {
write(in, out);
}
}
}
}
private static void write (InputStream in, OutputStream out) throws IOException {
byte[] buffer = new byte[1024];
int len;
while ((len = in.read(buffer)) >= 0) {
out.write(buffer, 0, len);
}
}
CapabilityFactory.getChromeCapabilitiesWithExtension(...)
public static DesiredCapabilities getChromeCapabilitiesWithExtension(String crxExtensionPath) {
DesiredCapabilities chromeCapabilities = getChromeCapabilities();
Logger.info("Extension path: " + crxExtensionPath);
ChromeOptions options = new ChromeOptions();
options.addExtensions(new File(crxExtensionPath));
options.addArguments("--start-maximized");
chromeCapabilities.setCapability(ChromeOptions.CAPABILITY, options);
return chromeCapabilities;
}
Tools/Chrome_ModHeader_2_0_6.crx
。这可能是因为该文件在您的远程机器上不可用,因此安装失败。本地可以工作,因为它在您的本地系统中可用。 - Prasanta Biswas