How to use enableReferentialIntegrity method of org.evomaster.client.java.controller.db.DbCleaner class

Best EvoMaster code snippet using org.evomaster.client.java.controller.db.DbCleaner.enableReferentialIntegrity

Source:DbCleaner.java Github

copy

Full Screen

...38 */39 disableReferentialIntegrity(statement, type);40 cleanDataInTables(tablesToSkip, statement, type, schemaName, isSingleCleanCommand(type), doDropTable);41 resetSequences(statement, type, schemaName);42 enableReferentialIntegrity(statement, type);43 statement.close();44 } catch (Exception e) {45 /*46 this could happen if there is a current transaction with a lock on any table.47 We could check the content of INFORMATION_SCHEMA.LOCKS, or simply look at error message48 */49 String msg = e.getMessage();50 if(msg != null && msg.toLowerCase().contains("timeout")){51 if(retries > 0) {52 SimpleLogger.warn("Timeout issue with cleaning DB. Trying again.");53 //let's just wait a bit, and retry54 try {55 Thread.sleep(2000);56 } catch (InterruptedException interruptedException) {57 }58 retries--;59 clearDatabase(retries, connection, schemaName, tablesToSkip, type, doDropTable);60 } else {61 SimpleLogger.error("Giving up cleaning the DB. There are still timeouts.");62 }63 }64 throw new RuntimeException(e);65 }66 }67 public static void clearDatabase(Connection connection, List<String> tablesToSkip, DatabaseType type){68 clearDatabase(connection, getDefaultSchema(type), tablesToSkip, type);69 }70 public static void clearDatabase(Connection connection, String schemaName, List<String> tablesToSkip, DatabaseType type){71 clearDatabase(getDefaultReties(type), connection, schemaName, tablesToSkip, type, false);72 }73 public static void dropDatabaseTables(Connection connection, String schemaName, List<String> tablesToSkip, DatabaseType type){74 if (type != DatabaseType.MYSQL && type != DatabaseType.MARIADB)75 throw new IllegalArgumentException("Dropping tables are not supported by "+type);76 clearDatabase(getDefaultReties(type), connection, schemaName, tablesToSkip, type, true);77 }78 public static void clearDatabase_Postgres(Connection connection, String schemaName, List<String> tablesToSkip ) {79 clearDatabase(getDefaultReties(DatabaseType.POSTGRES), connection, schemaName, tablesToSkip, DatabaseType.POSTGRES, false);80 }81 /**82 *83 * @param tablesToSkip are tables to be skipped84 * @param statement is to execute the SQL command85 * @param schema specify the schema of data to clean. if [schema] is empty, all data will be cleaned.86 * @param singleCommand specify whether to execute the SQL commands (e.g., truncate table/tables) by single command87 * @param doDropTable specify whether to drop tables which is only for MySQL and MariaDB now.88 * @throws SQLException are exceptions during sql execution89 */90 private static void cleanDataInTables(List<String> tablesToSkip, Statement statement, DatabaseType type, String schema, boolean singleCommand, boolean doDropTable) throws SQLException {91 // Find all tables and truncate them92 Set<String> tables = new HashSet<>();93 ResultSet rs = statement.executeQuery(getAllTableCommand(type, schema));94 while (rs.next()) {95 tables.add(rs.getString(1));96 }97 rs.close();98 if (tables.isEmpty()) {99 throw new IllegalStateException("Could not find any table");100 }101 if (tablesToSkip != null) {102 for (String skip : tablesToSkip) {103 if (!tables.stream().anyMatch(t -> t.equalsIgnoreCase(skip))) {104 String msg = "Asked to skip table '" + skip + "', but it does not exist.";105 msg += " Existing tables in schema '"+schema+"': [" +106 tables.stream().collect(Collectors.joining(", ")) + "]";107 throw new IllegalStateException(msg);108 }109 }110 }111 Set<String> tablesHaveIdentifies = new HashSet<>();112 if (type == DatabaseType.MS_SQL_SERVER){113 ResultSet rst = statement.executeQuery(getAllTableHasIdentify(type, schema));114 while (rst.next()) {115 tablesHaveIdentifies.add(rst.getString(1));116 }117 rst.close();118 }119 List<String> tablesToClear = tables.stream()120 .filter(n -> tablesToSkip == null || tablesToSkip.isEmpty() ||121 !tablesToSkip.stream().anyMatch(skip -> skip.equalsIgnoreCase(n)))122 .collect(Collectors.toList());123 if (singleCommand) {124 String ts = tablesToClear.stream()125 .sorted()126 .collect(Collectors.joining(","));127 if (type != DatabaseType.POSTGRES)128 throw new IllegalArgumentException("do not support for cleaning all data by one single command for " +type);129 if (doDropTable)130 dropTables(statement, ts);131 else{132 statement.executeUpdate("TRUNCATE TABLE " + ts);133 }134 } else {135 //note: if one at a time, need to make sure to first disable FK checks136 for(String t : tablesToClear){137 if (doDropTable)138 dropTables(statement, t);139 else{140 /*141 for MS_SQL_SERVER, we cannot use truncate tables if there exist fk142 see143 https://docs.microsoft.com/en-us/sql/t-sql/statements/truncate-table-transact-sql?view=sql-server-ver15#restrictions144 https://stackoverflow.com/questions/155246/how-do-you-truncate-all-tables-in-a-database-using-tsql#156813145 then it will cause a problem to reset identify146 */147 if (type == DatabaseType.MS_SQL_SERVER)148 deleteTables(statement, t, tablesHaveIdentifies);149 else150 truncateTables(statement, t);151 }152 }153 }154 }155 private static void dropTables(Statement statement, String table) throws SQLException {156 statement.executeUpdate("DROP TABLE IF EXISTS " +table);157 }158 private static void deleteTables(Statement statement, String table, Set<String> tableHasIdentify) throws SQLException {159 statement.executeUpdate("DELETE FROM "+table);160// NOTE TAHT ideally we should reseed identify here, but there would case an issue, i.e., does not contain an identity column161 if (tableHasIdentify.contains(table))162 statement.executeUpdate("DBCC CHECKIDENT ('"+table+"', RESEED, 0)");163 }164 private static void truncateTables(Statement statement, String table) throws SQLException {165 statement.executeUpdate("TRUNCATE TABLE " + table);166 }167 private static void resetSequences(Statement s, DatabaseType type, String schemaName) throws SQLException {168 ResultSet rs;// Idem for sequences169 Set<String> sequences = new HashSet<>();170 rs = s.executeQuery(getAllSequenceCommand(type, schemaName));171 while (rs.next()) {172 sequences.add(rs.getString(1));173 }174 rs.close();175 for (String seq : sequences) {176 s.executeUpdate(resetSequenceCommand(seq, type));177 }178 /*179 Note: we reset all sequences from 1. But the original database might180 have used a different value.181 In most cases (99.99%), this should not be a problem.182 We could allow using different values in this API... but, maybe just easier183 for the user to reset it manually if really needed?184 */185 }186 private static void disableReferentialIntegrity(Statement s, DatabaseType type) throws SQLException {187 switch (type)188 {189 case POSTGRES: break;190 case MS_SQL_SERVER:191 //https://stackoverflow.com/questions/159038/how-can-foreign-key-constraints-be-temporarily-disabled-using-t-sql192 //https://stackoverflow.com/questions/155246/how-do-you-truncate-all-tables-in-a-database-using-tsql#156813193 //https://docs.microsoft.com/en-us/sql/relational-databases/tables/disable-foreign-key-constraints-with-insert-and-update-statements?view=sql-server-ver15194 s.execute("EXEC sp_MSForEachTable \"ALTER TABLE ? NOCHECK CONSTRAINT all\"");195 break;196 case H2:197 s.execute("SET REFERENTIAL_INTEGRITY FALSE");198 break;199 case MARIADB:200 case MYSQL:201 s.execute("SET @@foreign_key_checks = 0;");202 break;203 case OTHER:204 throw new DbUnsupportedException(type);205 }206 }207 private static void enableReferentialIntegrity(Statement s, DatabaseType type) throws SQLException {208 switch (type)209 {210 case POSTGRES: break;211 case MS_SQL_SERVER:212 s.execute("exec sp_MSForEachTable \"ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all\"");213 break;214 case H2:215 /*216 For H2, we have to delete tables one at a time... but, to avoid issues217 with FKs, we must temporarily disable the integrity checks218 */219 s.execute( "SET REFERENTIAL_INTEGRITY TRUE");220 break;221 case MARIADB:...

Full Screen

Full Screen

enableReferentialIntegrity

Using AI Code Generation

copy

Full Screen

1import org.evomaster.client.java.controller.db.DbCleaner2import org.evomaster.client.java.controller.db.SqlScriptRunner3import org.evomaster.client.java.controller.db.Table4import org.evomaster.client.java.controller.db.TableColumn5import org.evomaster.client.java.controller.db.TableRow6import org.evomaster.client.java.controller.db.TableSchema7import org.evomaster.client.java.controller.problem.ProblemInfo8import org.evomaster.client.java.controller.problem.RestProblem9import org.evomaster.client.java.controller.problem.RestResourceCalls10import org.evomaster.client.java.controller.problem.RestResourceFitness11import org.evomaster.client.java.controller.problem.RestResourceIndividual12import org.evomaster.client.java.controller.problem.RestResourceStructure13import org.evomaster.client.java.controller.problem.RestResourceTemplate14import org.evomaster.client.java.controller.problem.RestVerb15import org.evomaster.client.java.controller.problem.Res

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful