Skip to content

Commit

Permalink
feat(testing): add new test case for afterStreamingOnly functionality
Browse files Browse the repository at this point in the history
  • Loading branch information
phodal committed Jul 22, 2024
1 parent 3655a59 commit a91be0d
Show file tree
Hide file tree
Showing 3 changed files with 120 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -23,5 +23,9 @@ class RealWorldShireParsingTest : ParsingTestCase("realworld", "shire", ShirePar
fun testWhenAfterStreaming() {
doTest(true)
}

fun testAfterStreamingOnly() {
doTest(true)
}
}

Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
---
name: "Search"
variables:
"testTemplate": /.*.kt/ { caching("disk") | splitting | embedding }
afterStreaming: { searching($output) | execute("search.shire") }
---

You are a coding assistant who helps the user answer questions about code in their workspace by providing a list of
relevant keywords they can search for to answer the question.
107 changes: 107 additions & 0 deletions shirelang/src/test/testData/realworld/AfterStreamingOnly.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,107 @@
ShireFile
ShireFrontMatterHeaderImpl(FRONT_MATTER_HEADER)
PsiElement(ShireTokenType.FRONTMATTER_START)('---')
PsiElement(ShireTokenType.NEWLINE)('\n')
ShireFrontMatterEntriesImpl(FRONT_MATTER_ENTRIES)
ShireFrontMatterEntryImpl(FRONT_MATTER_ENTRY)
ShireFrontMatterKeyImpl(FRONT_MATTER_KEY)
ShireFrontMatterIdImpl(FRONT_MATTER_ID)
PsiElement(ShireTokenType.IDENTIFIER)('name')
PsiElement(ShireTokenType.COLON)(':')
PsiWhiteSpace(' ')
ShireFrontMatterValueImpl(FRONT_MATTER_VALUE)
PsiElement(ShireTokenType.QUOTE_STRING)('"Search"')
PsiElement(ShireTokenType.NEWLINE)('\n')
ShireFrontMatterEntryImpl(FRONT_MATTER_ENTRY)
ShireFrontMatterKeyImpl(FRONT_MATTER_KEY)
ShireFrontMatterIdImpl(FRONT_MATTER_ID)
PsiElement(ShireTokenType.IDENTIFIER)('variables')
PsiElement(ShireTokenType.COLON)(':')
ShireFrontMatterValueImpl(FRONT_MATTER_VALUE)
PsiElement(ShireTokenType.NEWLINE)('\n')
ShireObjectKeyValueImpl(OBJECT_KEY_VALUE)
PsiElement(ShireTokenType.INDENT)(' ')
ShireKeyValueImpl(KEY_VALUE)
ShireFrontMatterEntryImpl(FRONT_MATTER_ENTRY)
ShireFrontMatterKeyImpl(FRONT_MATTER_KEY)
PsiElement(ShireTokenType.QUOTE_STRING)('"testTemplate"')
PsiElement(ShireTokenType.COLON)(':')
PsiWhiteSpace(' ')
ShirePatternActionImpl(PATTERN_ACTION)
PatternElement(PATTERN)
PsiElement(ShireTokenType.PATTERN_EXPR)('/.*.kt/')
PsiWhiteSpace(' ')
ShireActionBlockImpl(ACTION_BLOCK)
PsiElement(ShireTokenType.{)('{')
PsiWhiteSpace(' ')
ShireActionBodyImpl(ACTION_BODY)
ShireActionExprImpl(ACTION_EXPR)
ShireFuncCallImpl(FUNC_CALL)
ShireFuncNameImpl(FUNC_NAME)
PsiElement(ShireTokenType.IDENTIFIER)('caching')
PsiElement(ShireTokenType.()('(')
ShirePipelineArgsImpl(PIPELINE_ARGS)
ShirePipelineArgImpl(PIPELINE_ARG)
PsiElement(ShireTokenType.QUOTE_STRING)('"disk"')
PsiElement(ShireTokenType.))(')')
PsiWhiteSpace(' ')
PsiElement(ShireTokenType.|)('|')
PsiWhiteSpace(' ')
ShireActionExprImpl(ACTION_EXPR)
ShireFuncCallImpl(FUNC_CALL)
ShireFuncNameImpl(FUNC_NAME)
PsiElement(ShireTokenType.IDENTIFIER)('splitting')
PsiWhiteSpace(' ')
PsiElement(ShireTokenType.|)('|')
PsiWhiteSpace(' ')
ShireActionExprImpl(ACTION_EXPR)
ShireFuncCallImpl(FUNC_CALL)
ShireFuncNameImpl(FUNC_NAME)
PsiElement(ShireTokenType.IDENTIFIER)('embedding')
PsiWhiteSpace(' ')
PsiElement(ShireTokenType.})('}')
PsiElement(ShireTokenType.NEWLINE)('\n')
ShireFrontMatterEntryImpl(FRONT_MATTER_ENTRY)
ShireFrontMatterKeyImpl(FRONT_MATTER_KEY)
ShireFrontMatterIdImpl(FRONT_MATTER_ID)
PsiElement(ShireTokenType.IDENTIFIER)('afterStreaming')
PsiElement(ShireTokenType.COLON)(':')
PsiWhiteSpace(' ')
ShireFunctionStatementImpl(FUNCTION_STATEMENT)
PsiElement(ShireTokenType.{)('{')
PsiWhiteSpace(' ')
ShireFunctionBodyImpl(FUNCTION_BODY)
ShireActionBodyImpl(ACTION_BODY)
ShireActionExprImpl(ACTION_EXPR)
ShireFuncCallImpl(FUNC_CALL)
ShireFuncNameImpl(FUNC_NAME)
PsiElement(ShireTokenType.IDENTIFIER)('searching')
PsiElement(ShireTokenType.()('(')
ShirePipelineArgsImpl(PIPELINE_ARGS)
ShirePipelineArgImpl(PIPELINE_ARG)
ShireVariableStartImpl(VARIABLE_START)
PsiElement(VARIABLE_START)('$')
ShireVariableIdImpl(VARIABLE_ID)
PsiElement(ShireTokenType.IDENTIFIER)('output')
PsiElement(ShireTokenType.))(')')
PsiWhiteSpace(' ')
PsiElement(ShireTokenType.|)('|')
PsiWhiteSpace(' ')
ShireActionExprImpl(ACTION_EXPR)
ShireFuncCallImpl(FUNC_CALL)
ShireFuncNameImpl(FUNC_NAME)
PsiElement(ShireTokenType.IDENTIFIER)('execute')
PsiElement(ShireTokenType.()('(')
ShirePipelineArgsImpl(PIPELINE_ARGS)
ShirePipelineArgImpl(PIPELINE_ARG)
PsiElement(ShireTokenType.QUOTE_STRING)('"search.shire"')
PsiElement(ShireTokenType.))(')')
PsiWhiteSpace(' ')
PsiElement(ShireTokenType.})('}')
PsiElement(ShireTokenType.NEWLINE)('\n')
PsiElement(ShireTokenType.FRONTMATTER_END)('---')
PsiElement(ShireTokenType.NEWLINE)('\n')
PsiElement(ShireTokenType.NEWLINE)('\n')
PsiElement(ShireTokenType.TEXT_SEGMENT)('You are a coding assistant who helps the user answer questions about code in their workspace by providing a list of')
PsiElement(ShireTokenType.NEWLINE)('\n')
PsiElement(ShireTokenType.TEXT_SEGMENT)('relevant keywords they can search for to answer the question.')

0 comments on commit a91be0d

Please sign in to comment.