Skip to content

Commit

Permalink
Added new feature
Browse files Browse the repository at this point in the history
  • Loading branch information
@snwinz
snwinz committed Mar 24, 2022
1 parent 9bd9751 commit 2791909
Show file tree
Hide file tree
Showing 4 changed files with 121 additions and 31 deletions.
56 changes: 32 additions & 24 deletions src/main/java/logic/dynamicdatageneration/TestcaseSimulator.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@
import logic.dynamicdatageneration.testrun.TestData;
import logic.model.Testcase;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Optional;
import java.util.logging.Logger;
Expand Down Expand Up @@ -36,40 +38,45 @@ public Optional<TestData> simulateTestcase(Testcase testcase, Commands commands)
TestData testDataCreated = new TestData(testcase.getFunctions(), commands);

for (int actualRun = 1; actualRun <= maximalNumberOfTries; actualRun++) {

resetApplication();
executeFunctions(testcase, testDataCreated, actualRun);
var logList = executor.getAllNewLogs(0L);
LOGGER.info("Logs for Call of run " + actualRun + ":\n" + String.join("\n", logList));
List<String> specificTargetsToBeCovered = getAllTestTargets(testcase);
boolean allGeneralTestTargetsCovered = areAllTestTargetsCovered(testDataCreated, logList, testcase.getLogsOfTarget());
boolean allSpecificTargetsCovered = areAllTestTargetsCovered(testDataCreated, logList, specificTargetsToBeCovered);

if (allSpecificTargetsCovered) {
testcase.addTextToWriteOutput("Successfully covered in run " + actualRun + ":\n" + testDataCreated + "\n");
testcase.setCovered(true);
testcase.setTestData(testDataCreated);
testcase.setSpecificTargetCovered(true);
testcase.setTestTargetData(testDataCreated);
testcase.setNumberOfRuns(actualRun);
return Optional.of(testDataCreated);
} else if (allGeneralTestTargetsCovered) {
testcase.addTextToWriteOutput("Successfully covered in run " + actualRun + ":\n" + testDataCreated + "\n");
if (!testcase.isCovered()) {
try {
resetApplication();
executeFunctions(testcase, testDataCreated, actualRun);
var logList = executor.getAllNewLogs(0L);
LOGGER.info("Logs for Call of run " + actualRun + ":\n" + String.join("\n", logList));
List<String> specificTargetsToBeCovered = getAllTestTargets(testcase);
boolean allGeneralTestTargetsCovered = areAllTestTargetsCovered(testDataCreated, logList, testcase.getLogsOfTarget());
boolean allSpecificTargetsCovered = areAllTestTargetsCovered(testDataCreated, logList, specificTargetsToBeCovered);

if (allSpecificTargetsCovered) {
testcase.addTextToWriteOutput("Successfully covered in run " + actualRun + ":\n" + testDataCreated + "\n");
testcase.setCovered(true);
testcase.setTestData(testDataCreated);
testcase.setSpecificTargetCovered(true);
testcase.setTestTargetData(testDataCreated);
testcase.setNumberOfRuns(actualRun);
return Optional.of(testDataCreated);
} else if (allGeneralTestTargetsCovered) {
testcase.addTextToWriteOutput("Successfully covered in run " + actualRun + ":\n" + testDataCreated + "\n");
if (!testcase.isCovered()) {
testcase.setCovered(true);
testcase.setTestTargetData(testDataCreated);
testcase.setNumberOfRuns(actualRun);
}
}
/* quite generic and has to be more specific */
} catch (Exception e) {
String message = String.format("Error occurred in run%d: %s", actualRun, e.getMessage());
System.err.println(message);
LOGGER.warning(message);
}


}
return Optional.empty();
}

public void resetApplication() {
if (resetFunctionName != null) {
executor.invokeFunction(resetFunctionName, "{}");
executor.invokeFunction(resetFunctionName, "{}", new HashMap<>());
}
executor.deleteOldLogs();
}
Expand Down Expand Up @@ -114,7 +121,7 @@ private void executeFunctions(Testcase testcase, TestData testData, int actualRu
LOGGER.info(invocation);
testcase.addTextToWriteOutput(invocation);

String result = executor.invokeFunction(functionName, jsonData);
String result = executor.invokeFunction(functionName, jsonData, testData.getOutputValues());
testData.addResultToOutput(result);
String resultFormatted = String.format("result: %s", result);

Expand All @@ -132,7 +139,8 @@ private void executeFunctionsWithOldData(Testcase testcase, TestData testData) {
String invocation = String.format("invoke function '%s' with json '%s'", functionName, jsonData);
LOGGER.info(invocation);
testcase.addTextToWriteOutput(invocation);
String result = executor.invokeFunction(functionName, jsonData);
String result = executor.invokeFunction(functionName, jsonData, testData.getOutputValues());
testData.addResultToOutput(result);
String resultFormatted = String.format("result: %s", result);
LOGGER.info(resultFormatted);
testcase.addTextToWriteOutput(resultFormatted);
Expand Down
25 changes: 24 additions & 1 deletion src/main/java/logic/dynamicdatageneration/executionplatforms/AWSInvoker.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class AWSInvoker implements Executor {
protected AWSLambda amazonLambda;
Expand All @@ -23,10 +24,14 @@ public AWSInvoker(String region) {


@Override
public String invokeFunction(String functionName, String json) {
public String invokeFunction(String functionName, String json, Map<String, List<String>> outputValues) {
InvokeRequest invokeRequest = new InvokeRequest()
.withFunctionName(functionName);
json = json.replaceAll("'", "\"");

json = setOutputsOfPreviousFunctions(json, outputValues);


invokeRequest.setPayload(json);

InvokeResult invokeResult = amazonLambda.invoke(invokeRequest);
Expand All @@ -38,6 +43,24 @@ public String invokeFunction(String functionName, String json) {
return new String(invokeResult.getPayload().array(), StandardCharsets.UTF_8);
}

private String setOutputsOfPreviousFunctions(String json, Map<String, List<String>> outputValues) {
while (json.contains("##PREVIOUSOUTPUT__")) {
String key = json.split("##PREVIOUSOUTPUT")[1].split("__")[1];
var occurence = Integer.parseInt(json.split("##PREVIOUSOUTPUT")[1].split("__")[2]);
if (outputValues.containsKey(key) && outputValues.get(key).size()>occurence) {
String value = outputValues.get(key).get(occurence);
var jsonFirstPart= json.split("##PREVIOUSOUTPUT")[0];
var jsonSecondPart= json.split("PREVIOUSOUTPUT##",2)[1];
json = jsonFirstPart+value+jsonSecondPart;
}
else {
break;
}

}
return json;
}


@Override
public List<String> getAllNewLogs(long startTime) {
Expand Down
3 changes: 2 additions & 1 deletion src/main/java/logic/dynamicdatageneration/executionplatforms/Executor.java
View file
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
package logic.dynamicdatageneration.executionplatforms;

import java.util.List;
import java.util.Map;

public interface Executor {
String invokeFunction(String functionName, String jsonData);
String invokeFunction(String functionName, String jsonData, Map<String, List<String>> outputValues);

List<String> getAllNewLogs(long startTime);

Expand Down
68 changes: 63 additions & 5 deletions src/main/java/logic/dynamicdatageneration/testrun/FunctionWithInputData.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -107,18 +107,46 @@ public void changeData(TestData testData) {
var inputValues = testData.getInputValues().get(inputData.getKey());
setRandomEntryOfListAsValue(inputData, inputValues);
} else if (decisionRandomOutputAsValue < testData.getProbRandomOutputAsValue()) {
var inputValues = testData.getOutputValues().values().stream()
.flatMap(Collection::stream).collect(Collectors.toList());
setRandomEntryOfListAsValue(inputData, inputValues);
var inputValues = testData.getOutputValues().entrySet().stream().map(entry -> {
var key = entry.getKey();
var values = entry.getValue();
List<Pair> result = new ArrayList<>();
for (int i = 0; i < values.size(); i++) {
var value = values.get(i);
result.add(new Pair(key, value, i));
}
return result;
}).flatMap(List::stream).collect(Collectors.toList());
setRandomEntryOfOutputAsValue(inputData, inputValues);
} else if (decisionSimilarOutputAsValue < testData.getProbSimilarOutputAsValue()) {
var inputValues = testData.getOutputValues().get(inputData.getKey());
setRandomEntryOfListAsValue(inputData, inputValues);
var inputValues = testData.getOutputValues().entrySet().stream()
.filter(entry -> entry.getKey().equals(inputData.getKey()))
.map(entry -> {
var key = entry.getKey();
var values = entry.getValue();
List<Pair> result = new ArrayList<>();
for (int i = 0; i < values.size(); i++) {
var value = values.get(i);
result.add(new Pair(key, value, i));
}
return result;
}).flatMap(List::stream).collect(Collectors.toList());
setRandomEntryOfOutputAsValue(inputData, inputValues);
}
}
}

}

private void setRandomEntryOfOutputAsValue(GeneralInput inputData, List<Pair> inputValues) {
if (inputValues != null && inputValues.size() > 0) {
var item = getRandomItem2(inputValues);
var inputText = String.format("##PREVIOUSOUTPUT__%s__%s__PREVIOUSOUTPUT##", item.getKey(),item.getOccurence());
inputData.setGeneratedValue(inputText);
}

}

private void setRandomEntryOfListAsValue(GeneralInput inputData, List<String> inputValues) {
if (inputValues != null && inputValues.size() > 0) {
var item = getRandomItem(inputValues);
Expand All @@ -131,4 +159,34 @@ private <T> T getRandomItem(List<T> list) {
int entryNumber = rn.nextInt(list.size());
return list.get(entryNumber);
}

private <T> T getRandomItem2(List<T> list) {
int entryNumber = rn.nextInt(list.size());
return list.get(entryNumber);
}


private final class Pair {
private final String key;
private final String value;
private final int occurence;

Pair(String key, String value, int occurence) {
this.key = key;
this.value = value;
this.occurence = occurence;
}

public String getKey() {
return key;
}

public String getValue() {
return value;
}

public int getOccurence() {
return occurence;
}
}
}

0 comments on commit 2791909

Please sign in to comment.