forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTestOracle.java
More file actions
91 lines (81 loc) · 4.59 KB
/
TestOracle.java
File metadata and controls
91 lines (81 loc) · 4.59 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package sqlancer;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
public interface TestOracle {
void check() throws SQLException;
default boolean onlyWorksForNonEmptyTables() {
return false;
}
static void assumeResultSetsAreEqual(List<String> resultSet, List<String> secondResultSet,
String originalQueryString, List<String> combinedString, GlobalState<?> state) {
if (resultSet.size() != secondResultSet.size()) {
String queryFormatString = "%s; -- cardinality: %d";
String firstQueryString = String.format(queryFormatString, originalQueryString, resultSet.size());
String secondQueryString = String.format(queryFormatString,
combinedString.stream().collect(Collectors.joining(";")), secondResultSet.size());
state.getState().statements.add(new QueryAdapter(firstQueryString));
state.getState().statements.add(new QueryAdapter(secondQueryString));
String assertionMessage = String.format("the size of the result sets mismatch (%d and %d)!\n%s\n%s",
resultSet.size(), secondResultSet.size(), firstQueryString, secondQueryString);
throw new AssertionError(assertionMessage);
}
Set<String> firstHashSet = new HashSet<>(resultSet);
Set<String> secondHashSet = new HashSet<>(secondResultSet);
if (!firstHashSet.equals(secondHashSet)) {
Set<String> firstResultSetMisses = new HashSet<>(firstHashSet);
firstResultSetMisses.removeAll(secondHashSet);
Set<String> secondResultSetMisses = new HashSet<>(secondHashSet);
secondResultSetMisses.removeAll(firstHashSet);
String queryFormatString = "%s; -- misses: %s";
String firstQueryString = String.format(queryFormatString, originalQueryString, firstResultSetMisses);
String secondQueryString = String.format(queryFormatString,
combinedString.stream().collect(Collectors.joining(";")), secondResultSetMisses);
state.getState().statements.add(new QueryAdapter(firstQueryString));
state.getState().statements.add(new QueryAdapter(secondQueryString));
String assertionMessage = String.format("the content of the result sets mismatch!\n%s\n%s",
firstQueryString, secondQueryString);
throw new AssertionError(assertionMessage);
}
}
static List<String> getCombinedResultSet(String firstQueryString, String secondQueryString, String thirdQueryString,
List<String> combinedString, boolean asUnion, GlobalState<?> state, Set<String> errors)
throws SQLException {
List<String> secondResultSet;
if (asUnion) {
String unionString = firstQueryString + " UNION ALL " + secondQueryString + " UNION ALL "
+ thirdQueryString;
combinedString.add(unionString);
secondResultSet = DatabaseProvider.getResultSetFirstColumnAsString(unionString, errors,
state.getConnection(), state);
} else {
secondResultSet = new ArrayList<>();
secondResultSet.addAll(DatabaseProvider.getResultSetFirstColumnAsString(firstQueryString, errors,
state.getConnection(), state));
secondResultSet.addAll(DatabaseProvider.getResultSetFirstColumnAsString(secondQueryString, errors,
state.getConnection(), state));
secondResultSet.addAll(DatabaseProvider.getResultSetFirstColumnAsString(thirdQueryString, errors,
state.getConnection(), state));
combinedString.add(firstQueryString);
combinedString.add(secondQueryString);
combinedString.add(thirdQueryString);
}
return secondResultSet;
}
static List<String> getCombinedResultSetNoDuplicates(String firstQueryString, String secondQueryString,
String thirdQueryString, List<String> combinedString, boolean asUnion, GlobalState<?> state,
Set<String> errors) throws SQLException {
if (!asUnion) {
throw new AssertionError();
}
List<String> secondResultSet;
String unionString = firstQueryString + " UNION " + secondQueryString + " UNION " + thirdQueryString;
combinedString.add(unionString);
secondResultSet = DatabaseProvider.getResultSetFirstColumnAsString(unionString, errors, state.getConnection(),
state);
return secondResultSet;
}
}