FilesExpand file tree

 main

/

TiDBTLPHavingOracle.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

62 lines (53 loc) · 2.25 KB

 main

/

TiDBTLPHavingOracle.java

Top

File metadata and controls

• Code
• Blame

62 lines (53 loc) · 2.25 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package sqlancer.tidb.oracle;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

import sqlancer.ComparatorHelper;

import sqlancer.Randomly;

import sqlancer.common.oracle.TestOracle;

import sqlancer.tidb.TiDBErrors;

import sqlancer.tidb.TiDBProvider.TiDBGlobalState;

import sqlancer.tidb.ast.TiDBExpression;

import sqlancer.tidb.visitor.TiDBVisitor;

public class TiDBTLPHavingOracle extends TiDBTLPBase implements TestOracle<TiDBGlobalState> {

private String generatedQueryString;

public TiDBTLPHavingOracle(TiDBGlobalState state) {

super(state);

TiDBErrors.addExpressionHavingErrors(errors);

}

@Override

public void check() throws SQLException {

super.check();

if (Randomly.getBoolean()) {

select.setWhereClause(gen.generateExpression());

}

boolean orderBy = Randomly.getBoolean();

if (orderBy) {

select.setOrderByClauses(gen.generateOrderBys());

}

select.setGroupByExpressions(gen.generateExpressions(Randomly.smallNumber() + 1));

select.setHavingClause(null);

String originalQueryString = TiDBVisitor.asString(select);

generatedQueryString = originalQueryString;

List<String> resultSet = ComparatorHelper.getResultSetFirstColumnAsString(originalQueryString, errors, state);

select.setHavingClause(predicate);

String firstQueryString = TiDBVisitor.asString(select);

select.setHavingClause(negatedPredicate);

String secondQueryString = TiDBVisitor.asString(select);

select.setHavingClause(isNullPredicate);

String thirdQueryString = TiDBVisitor.asString(select);

List<String> combinedString = new ArrayList<>();

List<String> secondResultSet = ComparatorHelper.getCombinedResultSet(firstQueryString, secondQueryString,

thirdQueryString, combinedString, !orderBy, state, errors);

ComparatorHelper.assumeResultSetsAreEqual(resultSet, secondResultSet, originalQueryString, combinedString,

state);

}

@Override

protected TiDBExpression generatePredicate() {

return gen.generateHavingClause();

}

@Override

public String getLastQueryString() {

return generatedQueryString;

}

}