FilesExpand file tree

 reduction

/

H2QueryPartitioningBase.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

60 lines (51 loc) · 2.14 KB

 reduction

/

H2QueryPartitioningBase.java

Top

File metadata and controls

• Code
• Blame

60 lines (51 loc) · 2.14 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

package sqlancer.h2;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

import java.util.stream.Collectors;

import sqlancer.common.ast.newast.ColumnReferenceNode;

import sqlancer.common.ast.newast.Node;

import sqlancer.common.ast.newast.TableReferenceNode;

import sqlancer.common.gen.ExpressionGenerator;

import sqlancer.common.oracle.TernaryLogicPartitioningOracleBase;

import sqlancer.common.oracle.TestOracle;

import sqlancer.h2.H2Provider.H2GlobalState;

import sqlancer.h2.H2Schema.H2Column;

import sqlancer.h2.H2Schema.H2Table;

import sqlancer.h2.H2Schema.H2Tables;

public class H2QueryPartitioningBase extends TernaryLogicPartitioningOracleBase<Node<H2Expression>, H2GlobalState>

implements TestOracle {

H2Schema s;

H2Tables targetTables;

H2ExpressionGenerator gen;

H2Select select;

public H2QueryPartitioningBase(H2GlobalState state) {

super(state);

H2Errors.addExpressionErrors(errors);

}

@Override

public void check() throws SQLException {

s = state.getSchema();

targetTables = s.getRandomTableNonEmptyTables();

gen = new H2ExpressionGenerator(state).setColumns(targetTables.getColumns());

initializeTernaryPredicateVariants();

select = new H2Select();

select.setFetchColumns(generateFetchColumns());

List<H2Table> tables = targetTables.getTables();

List<TableReferenceNode<H2Expression, H2Table>> tableList = tables.stream()

.map(t -> new TableReferenceNode<H2Expression, H2Table>(t)).collect(Collectors.toList());

List<Node<H2Expression>> joins = H2Join.getJoins(tableList, state);

select.setJoinList(joins.stream().collect(Collectors.toList()));

select.setFromList(tableList.stream().collect(Collectors.toList()));

select.setWhereClause(null);

}

List<Node<H2Expression>> generateFetchColumns() {

List<Node<H2Expression>> columns = new ArrayList<>();

columns.add(new ColumnReferenceNode<>(new H2Column("*", null)));

return columns;

}

@Override

protected ExpressionGenerator<Node<H2Expression>> getGen() {

return gen;

}

}