|
6 | 6 |
|
7 | 7 | import com.contentgrid.opa.client.api.CompileApi; |
8 | 8 | import com.contentgrid.opa.client.api.CompileApi.PartialEvaluationRequest; |
9 | | -import com.contentgrid.opa.rego.ast.Query; |
| 9 | +import com.contentgrid.opa.client.api.DataApi; |
| 10 | +import com.contentgrid.opa.client.api.DataApi.GetDataResponse; |
| 11 | +import com.contentgrid.opa.client.api.PolicyApi.ListPoliciesResponse; |
10 | 12 | import com.contentgrid.opa.client.rest.http.HttpStatusException; |
11 | 13 | import com.contentgrid.opa.rego.ast.Expression; |
| 14 | +import com.contentgrid.opa.rego.ast.Query; |
12 | 15 | import com.contentgrid.opa.rego.ast.Term; |
13 | 16 | import com.contentgrid.opa.rego.ast.Term.Ref; |
14 | | -import com.contentgrid.opa.client.api.DataApi; |
15 | | -import com.contentgrid.opa.client.api.DataApi.GetDataResponse; |
16 | | -import com.contentgrid.opa.client.api.PolicyApi.ListPoliciesResponse; |
17 | 17 | import java.io.IOException; |
18 | 18 | import java.io.UncheckedIOException; |
19 | 19 | import java.nio.charset.StandardCharsets; |
@@ -322,11 +322,11 @@ void abacExample() { |
322 | 322 | assertThat(expr.getTerms()).hasSize(3); |
323 | 323 | assertThat(expr.getTerms()).anySatisfy( |
324 | 324 | term -> { |
325 | | - assertThat(term.getClass()).isEqualTo(Ref.class); |
326 | | - var ref = (Ref) term; |
327 | | - assertThat(ref.getValue()).singleElement().satisfies( |
328 | | - t -> assertThat(t.toString()).isEqualTo("eq") |
329 | | - ); |
| 325 | + assertThat(term).isInstanceOfSatisfying(Ref.class, ref -> { |
| 326 | + assertThat(ref.getValue()).singleElement().satisfies( |
| 327 | + t -> assertThat(t).hasToString("eq") |
| 328 | + ); |
| 329 | + }); |
330 | 330 | } |
331 | 331 | ); |
332 | 332 | })); |
|
0 commit comments