package org.openecomp.sdc.ci.tests.utilities;
-import static org.testng.AssertJUnit.assertTrue;
-
-import java.io.BufferedOutputStream;
-import java.io.BufferedWriter;
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.FileOutputStream;
-import java.io.FileWriter;
-import java.io.FilenameFilter;
-import java.io.IOException;
-import java.io.InputStream;
-import java.nio.file.Paths;
-import java.util.ArrayList;
-import java.util.Enumeration;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Properties;
-import java.util.zip.ZipEntry;
-import java.util.zip.ZipException;
-import java.util.zip.ZipFile;
-import java.util.zip.ZipInputStream;
-
+import com.aventstack.extentreports.Status;
+import com.clearspring.analytics.util.Pair;
import org.apache.commons.io.FileUtils;
import org.openecomp.sdc.be.model.DataTypeDefinition;
import org.openecomp.sdc.ci.tests.config.Config;
import org.openecomp.sdc.common.util.GeneralUtility;
import org.yaml.snakeyaml.Yaml;
-import com.aventstack.extentreports.Status;
-import com.clearspring.analytics.util.Pair;
+import java.io.*;
+import java.nio.file.Paths;
+import java.util.*;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipException;
+import java.util.zip.ZipFile;
+import java.util.zip.ZipInputStream;
+
+import static org.testng.AssertJUnit.assertTrue;
public class FileHandling {
-// ------------------yaml parser methods----------------------------
- public static Map<?, ?> parseYamlFile(String filePath) throws Exception {
- Yaml yaml = new Yaml();
- File file = new File(filePath);
- InputStream inputStream = new FileInputStream(file);
- Map<?, ?> map = new HashMap<>();
- map = (Map<?, ?>) yaml.load(inputStream);
- return map;
- }
-
- /**
- * The method return map fetched objects by pattern from yaml file
- * @param yamlFile
- * @param pattern
- * @return
- * @throws Exception
- */
- public static Map<String, Object> parseYamlFileToMapByPattern(File yamlFile, String pattern) throws Exception {
- Map<?, ?> yamlFileToMap = FileHandling.parseYamlFile(yamlFile.toString());
- Map<String, Object> objectMap = getObjectMapByPattern(yamlFileToMap, pattern);
- return objectMap;
- }
-
- @SuppressWarnings("unchecked")
- public static Map<String, Object> getObjectMapByPattern(Map<?, ?> parseUpdetedEnvFile, String pattern) {
- Map<String, Object> objectMap = null;
-
- Object objectUpdetedEnvFile = parseUpdetedEnvFile.get(pattern);
- if(objectUpdetedEnvFile instanceof HashMap){
- objectMap = (Map<String, Object>) objectUpdetedEnvFile;
- }
- return objectMap;
- }
-
-
- public static Map<String, DataTypeDefinition> parseDataTypesYaml(String filePath) throws Exception {
- @SuppressWarnings("unchecked")
- Map<String, DataTypeDefinition> dataTypesMap = (Map<String, DataTypeDefinition>) parseYamlFile(filePath);
- return dataTypesMap;
- }
+ // ------------------yaml parser methods----------------------------
+ public static Map<?, ?> parseYamlFile(String filePath) throws Exception {
+ Yaml yaml = new Yaml();
+ File file = new File(filePath);
+ InputStream inputStream = new FileInputStream(file);
+ Map<?, ?> map = new HashMap<>();
+ map = (Map<?, ?>) yaml.load(inputStream);
+ return map;
+ }
+
+ /**
+ * The method return map fetched objects by pattern from yaml file
+ *
+ * @param yamlFile
+ * @param pattern
+ * @return
+ * @throws Exception
+ */
+ public static Map<String, Object> parseYamlFileToMapByPattern(File yamlFile, String pattern) throws Exception {
+ Map<?, ?> yamlFileToMap = FileHandling.parseYamlFile(yamlFile.toString());
+ Map<String, Object> objectMap = getObjectMapByPattern(yamlFileToMap, pattern);
+ return objectMap;
+ }
+
+ @SuppressWarnings("unchecked")
+ public static Map<String, Object> getObjectMapByPattern(Map<?, ?> parseUpdetedEnvFile, String pattern) {
+ Map<String, Object> objectMap = null;
+
+ Object objectUpdetedEnvFile = parseUpdetedEnvFile.get(pattern);
+ if (objectUpdetedEnvFile instanceof HashMap) {
+ objectMap = (Map<String, Object>) objectUpdetedEnvFile;
+ }
+ return objectMap;
+ }
+
+
+ public static Map<String, DataTypeDefinition> parseDataTypesYaml(String filePath) throws Exception {
+ @SuppressWarnings("unchecked")
+ Map<String, DataTypeDefinition> dataTypesMap = (Map<String, DataTypeDefinition>) parseYamlFile(filePath);
+ return dataTypesMap;
+ }
// -------------------------------------------------------------------------------------------------
-
-
- /**
- * @param folder, folder name under "Files" folder
- * @return path to given folder from perspective of working directory or sdc-vnfs repository
- */
- public static String getFilePath(String folder) {
- String filepath = System.getProperty("filePath");
- boolean isFilePathEmptyOrNull = (filepath == null || filepath.isEmpty());
-
- // return folder from perspective of sdc-vnfs repository
- if (isFilePathEmptyOrNull && ( System.getProperty("os.name").contains("Windows") || System.getProperty("os.name").contains("Mac"))) {
- return FileHandling.getResourcesFilesPath() + folder + File.separator;
- }
-
- // return folder from perspective of working directory ( in general for nightly run from Linux, should already contain "Files" directory )
- return FileHandling.getBasePath() + "Files" + File.separator + folder + File.separator;
- }
-
- public static String getBasePath() {
- return System.getProperty("user.dir") + File.separator;
- }
-
- public static String getSdcVnfsPath() {
- return getBasePath() + Paths.get("..", "..", "sdc-vnfs").toString();
- }
-
- public static String getDriversPath() {
- return getBasePath() + "src" + File.separator + "main" + File.separator + "resources"
- + File.separator + "ci" + File.separator + "drivers" + File.separator;
- }
-
- public static String getResourcesFilesPath() {
+
+
+ /**
+ * @param folder, folder name under "Files" folder
+ * @return path to given folder from perspective of working directory or sdc-vnfs repository
+ */
+ public static String getFilePath(String folder) {
+ String filepath = System.getProperty("filePath");
+ boolean isFilePathEmptyOrNull = (filepath == null || filepath.isEmpty());
+
+ // return folder from perspective of sdc-vnfs repository
+ if (isFilePathEmptyOrNull && (System.getProperty("os.name").contains("Windows") || System.getProperty("os.name").contains("Mac"))) {
+ return FileHandling.getResourcesFilesPath() + folder + File.separator;
+ }
+
+ // return folder from perspective of working directory ( in general for nightly run from Linux, should already contain "Files" directory )
+ return FileHandling.getBasePath() + "Files" + File.separator + folder + File.separator;
+ }
+
+ public static String getBasePath() {
+ return System.getProperty("user.dir") + File.separator;
+ }
+
+ public static String getSdcVnfsPath() {
+ String vnfsPath = System.getProperty("vnfs.path");
+ if (vnfsPath != null && !vnfsPath.isEmpty()) {
+ return vnfsPath;
+ }
+ return getBasePath() + Paths.get("..", "..", "sdc-vnfs").toString();
+ }
+
+ public static String getDriversPath() {
+ return getBasePath() + "src" + File.separator + "main" + File.separator + "resources"
+ + File.separator + "ci" + File.separator + "drivers" + File.separator;
+ }
+
+ public static String getResourcesFilesPath() {
// return getBasePath() + "src" + File.separator + "main" + File.separator + "resources"
// + File.separator + "Files" + File.separator;
- return getSdcVnfsPath()+ File.separator + "ui-tests" + File.separator + "Files" + File.separator;
- }
-
- public static String getResourcesEnvFilesPath() {
- return getBasePath() + File.separator + "src" + File.separator + "main" + File.separator + "resources"
- + File.separator + "Files" + File.separator + "ResourcesEnvFiles" +File.separator;
- }
-
- public static String getCiFilesPath() {
- return getBasePath() + "src" + File.separator + "main" + File.separator + "resources"
- + File.separator + "ci";
- }
-
- public static String getConfFilesPath() {
- return getCiFilesPath() + File.separator + "conf" + File.separator;
- }
-
- public static String getTestSuitesFilesPath() {
- return getCiFilesPath() + File.separator + "testSuites" + File.separator;
- }
-
- public static String getVnfRepositoryPath() {
- return getFilePath("VNFs");
- }
-
- public static String getUpdateVSPVnfRepositoryPath() {
- return getFilePath("UpdateVSP");
- }
-
- public static File getConfigFile(String configFileName) throws Exception {
- File configFile = new File(FileHandling.getBasePath() + File.separator + "conf" + File.separator + configFileName);
- if (!configFile.exists()) {
- configFile = new File(FileHandling.getConfFilesPath() + configFileName);
- }
- return configFile;
- }
-
- public static Object[] filterFileNamesFromFolder(String filepath, String extension) {
- try {
- File dir = new File(filepath);
- List<String> filenames = new ArrayList<String>();
-
- FilenameFilter extensionFilter = new FilenameFilter() {
- public boolean accept(File dir, String name) {
- return name.endsWith(extension);
- }
- };
-
- if (dir.isDirectory()) {
- for (File file : dir.listFiles(extensionFilter)) {
- filenames.add(file.getName());
- }
- return filenames.toArray();
- }
-
- } catch (Exception e) {
- e.printStackTrace();
- }
- return null;
- }
-
- public static List<String> filterFileNamesListFromFolder(String filepath, String extension) {
- try {
- File dir = new File(filepath);
- List<String> filenames = new ArrayList<String>();
-
- FilenameFilter extensionFilter = new FilenameFilter() {
- public boolean accept(File dir, String name) {
- return name.endsWith(extension);
- }
- };
-
- if (dir.isDirectory()) {
- for (File file : dir.listFiles(extensionFilter)) {
- filenames.add(file.getName());
- }
-
- filenames.removeAll(OnboardingUtils.exludeVnfList);
-
- return filenames;
- }
-
- } catch (Exception e) {
- e.printStackTrace();
- }
- return null;
- }
-
- public static String[] getArtifactsFromZip(String filepath, String zipFilename){
- try{
- ZipFile zipFile = new ZipFile(filepath + File.separator + zipFilename);
- Enumeration<? extends ZipEntry> entries = zipFile.entries();
-
- List<String> artifactNames = new ArrayList<String>();
-
- while(entries.hasMoreElements()){
- ZipEntry nextElement = entries.nextElement();
- if (!nextElement.isDirectory()){
- if (!nextElement.getName().equals("MANIFEST.json")){
- String name = nextElement.getName();
- artifactNames.add(name);
- }
- }
- }
- zipFile.close();
- // convert list to array
- return artifactNames.toArray(new String[0]);
- }
- catch(ZipException zipEx){
- System.err.println("Error in zip file named : " + zipFilename);
- zipEx.printStackTrace();
- } catch (IOException e) {
- System.err.println("Unhandled exception : ");
- e.printStackTrace();
- }
-
- return null;
-
- }
+ return getSdcVnfsPath() + File.separator + "ui-tests" + File.separator + "Files" + File.separator;
+ }
+
+ public static String getResourcesEnvFilesPath() {
+ return getBasePath() + File.separator + "src" + File.separator + "main" + File.separator + "resources"
+ + File.separator + "Files" + File.separator + "ResourcesEnvFiles" + File.separator;
+ }
+
+ public static String getCiFilesPath() {
+ return getBasePath() + "src" + File.separator + "main" + File.separator + "resources"
+ + File.separator + "ci";
+ }
+
+ public static String getConfFilesPath() {
+ return getCiFilesPath() + File.separator + "conf" + File.separator;
+ }
+
+ public static String getTestSuitesFilesPath() {
+ return getCiFilesPath() + File.separator + "testSuites" + File.separator;
+ }
+
+ public static String getVnfRepositoryPath() {
+ return getFilePath("VNFs");
+ }
+
+ public static String getUpdateVSPVnfRepositoryPath() {
+ return getFilePath("UpdateVSP");
+ }
+
+ public static File getConfigFile(String configFileName) throws Exception {
+ File configFile = new File(FileHandling.getBasePath() + File.separator + "conf" + File.separator + configFileName);
+ if (!configFile.exists()) {
+ configFile = new File(FileHandling.getConfFilesPath() + configFileName);
+ }
+ return configFile;
+ }
+
+ public static Object[] filterFileNamesFromFolder(String filepath, String extension) {
+ try {
+ File dir = new File(filepath);
+ List<String> filenames = new ArrayList<String>();
+
+ FilenameFilter extensionFilter = new FilenameFilter() {
+ public boolean accept(File dir, String name) {
+ return name.endsWith(extension);
+ }
+ };
+
+ if (dir.isDirectory()) {
+ for (File file : dir.listFiles(extensionFilter)) {
+ filenames.add(file.getName());
+ }
+ return filenames.toArray();
+ }
+
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ public static List<String> filterFileNamesListFromFolder(String filepath, String extension) {
+ try {
+ File dir = new File(filepath);
+ List<String> filenames = new ArrayList<String>();
+
+ FilenameFilter extensionFilter = new FilenameFilter() {
+ public boolean accept(File dir, String name) {
+ return name.endsWith(extension);
+ }
+ };
+
+ if (dir.isDirectory()) {
+ for (File file : dir.listFiles(extensionFilter)) {
+ filenames.add(file.getName());
+ }
+
+ filenames.removeAll(OnboardingUtils.exludeVnfList);
+
+ return filenames;
+ }
+
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ public static String[] getArtifactsFromZip(String filepath, String zipFilename) {
+ try {
+ ZipFile zipFile = new ZipFile(filepath + File.separator + zipFilename);
+ Enumeration<? extends ZipEntry> entries = zipFile.entries();
+
+ List<String> artifactNames = new ArrayList<String>();
+
+ while (entries.hasMoreElements()) {
+ ZipEntry nextElement = entries.nextElement();
+ if (!nextElement.isDirectory()) {
+ if (!nextElement.getName().equals("MANIFEST.json")) {
+ String name = nextElement.getName();
+ artifactNames.add(name);
+ }
+ }
+ }
+ zipFile.close();
+ // convert list to array
+ return artifactNames.toArray(new String[0]);
+ } catch (ZipException zipEx) {
+ System.err.println("Error in zip file named : " + zipFilename);
+ zipEx.printStackTrace();
+ } catch (IOException e) {
+ System.err.println("Unhandled exception : ");
+ e.printStackTrace();
+ }
+
+ return null;
+
+ }
// public static Object[] getZipFileNamesFromFolder(String filePath) {
// return filterFileNamesFromFolder(filePath, ".zip");
// }
-
- public static List<String> getZipFileNamesFromFolder(String filepath) {
- return filterFileNamesListFromFolder(filepath, ".zip");
- }
-
- public static int countFilesInZipFile(String[] artifactsArr, String reqExtension){
- int fileCounter = 0;
- for (String artifact : artifactsArr){
- String extensionFile = artifact.substring(artifact.lastIndexOf(".") + 1 , artifact.length());
- if (extensionFile.equals(reqExtension)){
- fileCounter++;
- }
- }
- return fileCounter;
- }
-
-
- /**
- * @return last modified file name from default directory
- * @throws Exception
- */
- public static synchronized File getLastModifiedFileNameFromDir() throws Exception{
- return getLastModifiedFileNameFromDir(SetupCDTest.getWindowTest().getDownloadDirectory());
- }
-
- /**
- * @param dirPath
- * @return last modified file name from dirPath directory
- */
- public static synchronized File getLastModifiedFileNameFromDir(String dirPath){
- File dir = new File(dirPath);
- File[] files = dir.listFiles();
- if (files == null) {
- assertTrue("File not found under directory " + dirPath, false);
- return null;
- }
-
- File lastModifiedFile = files[0];
- for (int i = 1; i < files.length; i++) {
- if(files[i].isDirectory()) {
- continue;
- }
- if (lastModifiedFile.lastModified() < files[i].lastModified()) {
- lastModifiedFile = files[i];
- }
- }
- return lastModifiedFile;
- }
-
- public static void deleteDirectory(String directoryPath) {
- File dir = new File(directoryPath);
- try {
- FileUtils.cleanDirectory(dir);
- } catch (IOException e) {
- System.out.println("Failed to delete " + dir);
- SetupCDTest.getExtendTest().log(Status.INFO, "Failed to delete " + dir);
- }
- }
-
- public static void createDirectory(String directoryPath) {
- File directory = new File(String.valueOf(directoryPath));
- if (! directory.exists()){
- directory.mkdir();
- }
- }
-
-
- /**
- * The method append data to existing file, if file not exists - create it
- * @param pathToFile
- * @param text
- * @param leftSpaceCount
- * @throws IOException
- */
- public static synchronized void writeToFile(File pathToFile, Object text, Integer leftSpaceCount) throws IOException{
-
- BufferedWriter bw = null;
- FileWriter fw = null;
- if(!pathToFile.exists()){
- createEmptyFile(pathToFile);
- }
- try {
- fw = new FileWriter(pathToFile, true);
- bw = new BufferedWriter(fw);
- StringBuilder sb = new StringBuilder();
- if(leftSpaceCount > 0 ){
- for(int i = 0; i < leftSpaceCount; i++){
- sb.append(" ");
- }
- }
- bw.write(sb.toString() + text);
- bw.newLine();
- bw.close();
- fw.close();
- } catch (Exception e) {
- SetupCDTest.getExtendTest().log(Status.INFO, "Unable to write to flie " + pathToFile);
- }
- }
-
- public static synchronized void writeToFile(File pathToFile, Map<String, Pair<String, Object>> dataMap, Integer leftSpaceCount) throws IOException{
-
- BufferedWriter bw = null;
- FileWriter fw = null;
- try {
- if(!pathToFile.exists()){
- createEmptyFile(pathToFile);
- }
- fw = new FileWriter(pathToFile, true);
- bw = new BufferedWriter(fw);
- StringBuilder sb = new StringBuilder();
- if(leftSpaceCount > 0 ){
- for(int i = 0; i < leftSpaceCount; i++){
- sb.append(" ");
- }
- }
- for(Map.Entry<String, Pair<String, Object>> entry : dataMap.entrySet()){
- Object record = ArtifactUIUtils.getFormatedData(entry.getKey(), entry.getValue().right);
- bw.write(sb.toString() + record);
- bw.newLine();
- }
- bw.close();
- fw.close();
- } catch (Exception e) {
- SetupCDTest.getExtendTest().log(Status.INFO, "Unable to write to flie " + pathToFile);
- }
- }
-
- public static void deleteLastDowloadedFiles(List<File> files) throws IOException {
- for (File file : files) {
- File fileToDelete =new File(Config.instance().getWindowsDownloadDirectory()+file.getName());
- fileToDelete.delete();
- }
- }
-
- public static void cleanCurrentDownloadDir() throws IOException {
- try{
- ExtentTestActions.log(Status.INFO, "Cleaning directory " + SetupCDTest.getWindowTest().getDownloadDirectory());
- System.gc();
- FileUtils.cleanDirectory(new File(SetupCDTest.getWindowTest().getDownloadDirectory()));
- }
- catch(Exception e){
-
- }
- }
-
- public static boolean isFileDownloaded(String downloadPath, String fileName) {
- boolean flag = false;
- File dir = new File(downloadPath);
- File[] dir_contents = dir.listFiles();
- for (int i = 0; i < dir_contents.length; i++) {
- if (dir_contents[i].getName().equals(fileName))
- return flag = true;
- }
- return flag;
- }
-
- public static String getMD5OfFile(File file) throws IOException {
- String content = FileUtils.readFileToString(file);
- String md5 = GeneralUtility.calculateMD5Base64EncodedByString(content);
- return md5;
- }
-
- public static File createEmptyFile(String fileToCreate) {
- File file= new File(fileToCreate);
- try {
- if(file.exists()){
- deleteFile(file);
- }
- file.createNewFile();
- SetupCDTest.getExtendTest().log(Status.INFO, "Create file " + fileToCreate);
- } catch (IOException e) {
- SetupCDTest.getExtendTest().log(Status.INFO, "Failed to create file " + fileToCreate);
- e.printStackTrace();
- }
- return file;
- }
-
- public static File createEmptyFile(File fileToCreate) {
- try {
- if(fileToCreate.exists()){
- deleteFile(fileToCreate);
- }
- fileToCreate.createNewFile();
- SetupCDTest.getExtendTest().log(Status.INFO, "Create file " + fileToCreate);
- } catch (IOException e) {
- SetupCDTest.getExtendTest().log(Status.INFO, "Failed to create file " + fileToCreate);
- e.printStackTrace();
- }
- return fileToCreate;
- }
-
- public static void deleteFile(File file){
-
- try{
- if(file.exists()){
- file.deleteOnExit();
- SetupCDTest.getExtendTest().log(Status.INFO, "File " + file.getName() + "has been deleted");
- }else{
- SetupCDTest.getExtendTest().log(Status.INFO, "Failed to delete file " + file.getName());
- }
- }catch(Exception e){
- e.printStackTrace();
- }
-
- }
-
-
- /**
- * get file list from directory by extension array
- * @param directory
- * @param okFileExtensions
- * @return
- */
- public static List<File> getHeatAndHeatEnvArtifactsFromZip(File directory, String[] okFileExtensions){
-
- List<File> fileList = new ArrayList<>();
- File[] files = directory.listFiles();
-
- for (String extension : okFileExtensions){
- for(File file : files){
- if (file.getName().toLowerCase().endsWith(extension)){
- fileList.add(file);
- }
- }
- }
- return fileList;
- }
-
- private static final int BUFFER_SIZE = 4096;
+
+ public static List<String> getZipFileNamesFromFolder(String filepath) {
+ return filterFileNamesListFromFolder(filepath, ".zip");
+ }
+
+ public static int countFilesInZipFile(String[] artifactsArr, String reqExtension) {
+ int fileCounter = 0;
+ for (String artifact : artifactsArr) {
+ String extensionFile = artifact.substring(artifact.lastIndexOf(".") + 1, artifact.length());
+ if (extensionFile.equals(reqExtension)) {
+ fileCounter++;
+ }
+ }
+ return fileCounter;
+ }
+
+
+ /**
+ * @return last modified file name from default directory
+ * @throws Exception
+ */
+ public static synchronized File getLastModifiedFileNameFromDir() throws Exception {
+ return getLastModifiedFileNameFromDir(SetupCDTest.getWindowTest().getDownloadDirectory());
+ }
+
+ /**
+ * @param dirPath
+ * @return last modified file name from dirPath directory
+ */
+ public static synchronized File getLastModifiedFileNameFromDir(String dirPath) {
+ File dir = new File(dirPath);
+ File[] files = dir.listFiles();
+ if (files == null) {
+ assertTrue("File not found under directory " + dirPath, false);
+ return null;
+ }
+
+ File lastModifiedFile = files[0];
+ for (int i = 1; i < files.length; i++) {
+ if (files[i].isDirectory()) {
+ continue;
+ }
+ if (lastModifiedFile.lastModified() < files[i].lastModified()) {
+ lastModifiedFile = files[i];
+ }
+ }
+ return lastModifiedFile;
+ }
+
+ public static void deleteDirectory(String directoryPath) {
+ File dir = new File(directoryPath);
+ if (dir.exists()) {
+ try {
+ FileUtils.cleanDirectory(dir);
+ } catch (IllegalArgumentException e) {
+ System.out.println("Failed to clean " + dir);
+ } catch (IOException e) {
+ System.out.println("Failed to clean " + dir);
+ }
+ }
+ }
+
+ public static void createDirectory(String directoryPath) {
+ File directory = new File(String.valueOf(directoryPath));
+ if (!directory.exists()) {
+ directory.mkdir();
+ }
+ }
+
+
+ /**
+ * The method append data to existing file, if file not exists - create it
+ *
+ * @param pathToFile
+ * @param text
+ * @param leftSpaceCount
+ * @throws IOException
+ */
+ public static synchronized void writeToFile(File pathToFile, Object text, Integer leftSpaceCount) throws IOException {
+
+ BufferedWriter bw = null;
+ FileWriter fw = null;
+ if (!pathToFile.exists()) {
+ createEmptyFile(pathToFile);
+ }
+ try {
+ fw = new FileWriter(pathToFile, true);
+ bw = new BufferedWriter(fw);
+ StringBuilder sb = new StringBuilder();
+ if (leftSpaceCount > 0) {
+ for (int i = 0; i < leftSpaceCount; i++) {
+ sb.append(" ");
+ }
+ }
+ bw.write(sb.toString() + text);
+ bw.newLine();
+ bw.close();
+ fw.close();
+ } catch (Exception e) {
+ SetupCDTest.getExtendTest().log(Status.INFO, "Unable to write to flie " + pathToFile);
+ }
+ }
+
+ public static synchronized void writeToFile(File pathToFile, Map<String, Pair<String, Object>> dataMap, Integer leftSpaceCount) throws IOException {
+
+ BufferedWriter bw = null;
+ FileWriter fw = null;
+ try {
+ if (!pathToFile.exists()) {
+ createEmptyFile(pathToFile);
+ }
+ fw = new FileWriter(pathToFile, true);
+ bw = new BufferedWriter(fw);
+ StringBuilder sb = new StringBuilder();
+ if (leftSpaceCount > 0) {
+ for (int i = 0; i < leftSpaceCount; i++) {
+ sb.append(" ");
+ }
+ }
+ for (Map.Entry<String, Pair<String, Object>> entry : dataMap.entrySet()) {
+ Object record = ArtifactUIUtils.getFormatedData(entry.getKey(), entry.getValue().right);
+ bw.write(sb.toString() + record);
+ bw.newLine();
+ }
+ bw.close();
+ fw.close();
+ } catch (Exception e) {
+ SetupCDTest.getExtendTest().log(Status.INFO, "Unable to write to flie " + pathToFile);
+ }
+ }
+
+ public static void deleteLastDowloadedFiles(List<File> files) throws IOException {
+ for (File file : files) {
+ File fileToDelete = new File(Config.instance().getWindowsDownloadDirectory() + file.getName());
+ fileToDelete.delete();
+ }
+ }
+
+ public static void cleanCurrentDownloadDir() throws IOException {
+ try {
+ ExtentTestActions.log(Status.INFO, "Cleaning directory " + SetupCDTest.getWindowTest().getDownloadDirectory());
+ System.gc();
+ FileUtils.cleanDirectory(new File(SetupCDTest.getWindowTest().getDownloadDirectory()));
+ } catch (Exception e) {
+
+ }
+ }
+
+ public static boolean isFileDownloaded(String downloadPath, String fileName) {
+ boolean flag = false;
+ File dir = new File(downloadPath);
+ File[] dir_contents = dir.listFiles();
+ for (int i = 0; i < dir_contents.length; i++) {
+ if (dir_contents[i].getName().equals(fileName))
+ return flag = true;
+ }
+ return flag;
+ }
+
+ public static String getMD5OfFile(File file) throws IOException {
+ String content = FileUtils.readFileToString(file);
+ String md5 = GeneralUtility.calculateMD5Base64EncodedByString(content);
+ return md5;
+ }
+
+ public static File createEmptyFile(String fileToCreate) {
+ File file = new File(fileToCreate);
+ try {
+ if (file.exists()) {
+ deleteFile(file);
+ }
+ file.createNewFile();
+ SetupCDTest.getExtendTest().log(Status.INFO, "Create file " + fileToCreate);
+ } catch (IOException e) {
+ SetupCDTest.getExtendTest().log(Status.INFO, "Failed to create file " + fileToCreate);
+ e.printStackTrace();
+ }
+ return file;
+ }
+
+ public static File createEmptyFile(File fileToCreate) {
+ try {
+ if (fileToCreate.exists()) {
+ deleteFile(fileToCreate);
+ }
+ fileToCreate.createNewFile();
+ SetupCDTest.getExtendTest().log(Status.INFO, "Create file " + fileToCreate);
+ } catch (IOException e) {
+ SetupCDTest.getExtendTest().log(Status.INFO, "Failed to create file " + fileToCreate);
+ e.printStackTrace();
+ }
+ return fileToCreate;
+ }
+
+ public static void deleteFile(File file) {
+
+ try {
+ if (file.exists()) {
+ file.deleteOnExit();
+ SetupCDTest.getExtendTest().log(Status.INFO, "File " + file.getName() + "has been deleted");
+ } else {
+ SetupCDTest.getExtendTest().log(Status.INFO, "Failed to delete file " + file.getName());
+ }
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+
+ }
+
+
+ /**
+ * get file list from directory by extension array
+ *
+ * @param directory
+ * @param okFileExtensions
+ * @return
+ */
+ public static List<File> getHeatAndHeatEnvArtifactsFromZip(File directory, String[] okFileExtensions) {
+
+ List<File> fileList = new ArrayList<>();
+ File[] files = directory.listFiles();
+
+ for (String extension : okFileExtensions) {
+ for (File file : files) {
+ if (file.getName().toLowerCase().endsWith(extension)) {
+ fileList.add(file);
+ }
+ }
+ }
+ return fileList;
+ }
+
+ private static final int BUFFER_SIZE = 4096;
+
public static void unzip(String zipFilePath, String destDirectory) throws IOException {
File destDir = new File(destDirectory);
if (!destDir.exists()) {
ZipEntry entry = zipIn.getNextEntry();
// iterates over entries in the zip file
while (entry != null) {
- String entryName;
- if(System.getProperty("os.name").contains("Windows")){
- entryName = entry.getName().replaceAll("/", "\\"+File.separator);
- }else{
- entryName = entry.getName();
- }
+ String entryName;
+ if (System.getProperty("os.name").contains("Windows")) {
+ entryName = entry.getName().replaceAll("/", "\\" + File.separator);
+ } else {
+ entryName = entry.getName();
+ }
String filePath = destDirectory + entryName;
String currPath = destDirectory;
- String[] dirs = entryName.split("\\"+File.separator);
+ String[] dirs = entryName.split("\\" + File.separator);
String currToken;
- for(int i = 0; i<dirs.length;++i){
- currToken = dirs[i];
- if(!entry.isDirectory() && i==dirs.length-1){
- extractFile(zipIn, filePath);
- } else {
- if(currPath.endsWith(File.separator)){
- currPath = currPath + currToken;
- }else{
- currPath = currPath + File.separator + currToken;
- }
+ for (int i = 0; i < dirs.length; ++i) {
+ currToken = dirs[i];
+ if (!entry.isDirectory() && i == dirs.length - 1) {
+ extractFile(zipIn, filePath);
+ } else {
+ if (currPath.endsWith(File.separator)) {
+ currPath = currPath + currToken;
+ } else {
+ currPath = currPath + File.separator + currToken;
+ }
// if the entry is a directory, make the directory
File dir = new File(currPath);
dir.mkdir();
}
bos.close();
}
-
- public static int getFileCountFromDefaulDownloadDirectory(){
- return new File(SetupCDTest.getWindowTest().getDownloadDirectory()).listFiles().length;
+
+ public static int getFileCountFromDefaulDownloadDirectory() {
+ return new File(SetupCDTest.getWindowTest().getDownloadDirectory()).listFiles().length;
}
-
-
+
+
public static String getKeyByValueFromPropertyFormatFile(String fullPath, String key) {
- Properties prop = new Properties();
- InputStream input = null;
- String value = null;
- try {
- input = new FileInputStream(fullPath);
- prop.load(input);
- value = (prop.getProperty(key));
-
- } catch (IOException ex) {
- ex.printStackTrace();
- } finally {
- if (input != null) {
- try {
- input.close();
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- }
-
- return value.replaceAll("\"","");
- }
+ Properties prop = new Properties();
+ InputStream input = null;
+ String value = null;
+ try {
+ input = new FileInputStream(fullPath);
+ prop.load(input);
+ value = (prop.getProperty(key));
+
+ } catch (IOException ex) {
+ ex.printStackTrace();
+ } finally {
+ if (input != null) {
+ try {
+ input.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+ }
+
+ return value.replaceAll("\"", "");
+ }
}
Code Review / sdc.git / blobdiff