org.assertj.core.api.AssertProvider Java Examples

static AssertProvider<ExpressionAssert> newExpressionAssert(String expression, QueryRunner runner, Session session)
{
    MaterializedResult result = runner.execute(session, "VALUES " + expression);
    Type type = result.getTypes().get(0);
    Object value = result.getOnlyColumnAsSet().iterator().next();
    return () -> new ExpressionAssert(runner, session, value, type)
            .withRepresentation(TYPE_RENDERER);
}
 

private static <T extends Node> AssertProvider<ParserAssert> createAssertion(Function<String, T> parser, String sql)
{
    return () -> new ParserAssert(parser.apply(sql), newRecursiveComparisonConfig())
            .withRepresentation(NODE_REPRESENTATION)
            .satisfies(parsed -> new ParserAssert(parser.apply(formatSql((Node) parsed)), newRecursiveComparisonConfig())
                    .describedAs("Validate SQL->AST->SQL roundtrip")
                    .withRepresentation(NODE_REPRESENTATION)
                    .ignoringLocation()
                    .isEqualTo(parser.apply(sql)));
}
 

private AssertProvider<GroovyDslGradleSettingsAssert> forSampleGradleSettings() {
	String path = "project/build/gradle/sample-settings.gradle";
	try (InputStream in = new ClassPathResource(path).getInputStream()) {
		String content = StreamUtils.copyToString(in, StandardCharsets.UTF_8);
		return () -> new GroovyDslGradleSettingsAssert(content);
	}
	catch (IOException ex) {
		throw new IllegalStateException("No content found at " + path, ex);
	}
}
 

private AssertProvider<GroovyDslGradleBuildAssert> forSampleGradleBuild() {
	String path = "project/build/gradle/sample-build.gradle";
	try (InputStream in = new ClassPathResource(path).getInputStream()) {
		String content = StreamUtils.copyToString(in, StandardCharsets.UTF_8);
		return () -> new GroovyDslGradleBuildAssert(content);
	}
	catch (IOException ex) {
		throw new IllegalStateException("No content found at " + path, ex);
	}
}
 

private AssertProvider<MavenBuildAssert> forMavenBuild(String name) {
	try (InputStream in = new ClassPathResource("project/build/maven/" + name).getInputStream()) {
		String content = StreamUtils.copyToString(in, StandardCharsets.UTF_8);
		return () -> new MavenBuildAssert(content);
	}
	catch (IOException ex) {
		throw new IllegalStateException("No content found at " + name, ex);
	}
}
 

static AssertProvider<QueryAssert> newQueryAssert(String query, QueryRunner runner, Session session)
{
    MaterializedResult result = runner.execute(session, query);
    return () -> new QueryAssert(runner, session, result);
}
 

public static AssertProvider<ParserAssert> type(String sql)
{
    return createAssertion(new SqlParser()::createType, sql);
}
 

public static AssertProvider<ParserAssert> expression(String sql)
{
    return createAssertion(expression -> new SqlParser().createExpression(expression, new ParsingOptions(AS_DECIMAL)), sql);
}
 

public static AssertProvider<ParserAssert> statement(String sql)
{
    return createAssertion(statement -> new SqlParser().createStatement(statement, new ParsingOptions(AS_DECIMAL)), sql);
}
 

protected AssertProvider<MavenBuildAssert> mavenPom(ProjectRequest request) {
	request.setType("maven-build");
	String content = new String(getInvoker().invokeBuildGeneration(request));
	return () -> new MavenBuildAssert(content);
}
 

protected AssertProvider<GroovyDslGradleBuildAssert> gradleBuild(ProjectRequest request) {
	request.setType("gradle-build");
	String content = new String(getInvoker().invokeBuildGeneration(request));
	return () -> new GroovyDslGradleBuildAssert(content);
}
 

private AssertProvider<ModuleAssert> forDirectory(Path projectDirectory) {
	return () -> new ModuleAssert(projectDirectory);
}
 

private AssertProvider<AbstractJvmModuleAssert<?>> forJavaProject(Path root) {
	return () -> new JvmModuleAssert(root, JAVA_LANGUAGE);
}
 

private AssertProvider<MavenBuildAssert> forSampleMavenBuild() {
	return forMavenBuild("sample-pom.xml");
}
 

private AssertProvider<TextAssert> forContent(String content) {
	return () -> new TextAssert(content);
}
 

private AssertProvider<TextAssert> forContent(Path file) {
	return () -> new TextAssert(file);
}