Skip to content

Commit 087a08d

Browse files
Add tokens to AsPattern interface
1 parent 651a9f4 commit 087a08d

File tree

3 files changed

+10
-5
lines changed

3 files changed

+10
-5
lines changed

python-frontend/src/main/java/org/sonar/plugins/python/api/tree/AsPattern.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,5 +31,7 @@ public interface AsPattern extends Pattern {
3131

3232
Pattern pattern();
3333

34+
Token asKeyword();
35+
3436
Name alias();
3537
}

python-frontend/src/main/java/org/sonar/python/tree/AsPatternImpl.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,11 @@ public Pattern pattern() {
4444
return pattern;
4545
}
4646

47+
@Override
48+
public Token asKeyword() {
49+
return asKeyword;
50+
}
51+
4752
@Override
4853
public Name alias() {
4954
return alias;

python-frontend/src/test/java/org/sonar/python/tree/PythonTreeMakerMatchStatementTest.java

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -150,15 +150,13 @@ public void or_pattern() {
150150

151151
@Test
152152
public void as_pattern() {
153-
setRootRule(PythonGrammar.CASE_BLOCK);
154-
CaseBlock caseBlock = parse("case \"foo\" as x: ...", treeMaker::caseBlock);
155-
AsPattern asPattern = (AsPattern) caseBlock.pattern();
153+
AsPattern asPattern = pattern("case \"foo\" as x: ...");
156154
assertThat(asPattern.pattern()).isInstanceOf(LiteralPattern.class);
155+
assertThat(asPattern.asKeyword().value()).isEqualTo("as");
157156
assertThat(asPattern.alias().name()).isEqualTo("x");
158157
assertThat(asPattern.children()).extracting(Tree::getKind).containsExactly(Kind.STRING_LITERAL_PATTERN, Tree.Kind.TOKEN, Tree.Kind.NAME);
159158

160-
caseBlock = parse("case value as x: ...", treeMaker::caseBlock);
161-
asPattern = (AsPattern) caseBlock.pattern();
159+
asPattern = pattern("case value as x: ...");
162160
assertThat(asPattern.pattern().getKind()).isEqualTo(Tree.Kind.CAPTURE_PATTERN);
163161
}
164162

0 commit comments

Comments
 (0)