Skip to content

Commit

Permalink
add test case
Browse files Browse the repository at this point in the history
  • Loading branch information
anjaliagg9791 committed Oct 4, 2023
1 parent 0925646 commit 1a3ce2a
Showing 1 changed file with 44 additions and 0 deletions.
44 changes: 44 additions & 0 deletions internal/store/elasticsearch/es_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,50 @@ func TestElasticsearch(t *testing.T) {
analyzedTokens = append(analyzedTokens, tok.Token)
}

if reflect.DeepEqual(expectTokens, analyzedTokens) == false {
return fmt.Errorf("expected analyzer to tokenize %q as %v, was %v", textToAnalyze, expectTokens, analyzedTokens)
}
return nil
},
},
{
Title: "created index should be able to correctly tokenize CamelCase text",
Service: daggerService,
Validate: func(esClient *store.Client, cli *elasticsearch.Client, indexName string) error {
textToAnalyze := "walking"
analyzerPath := fmt.Sprintf("/%s/_analyze", indexName)
analyzerPayload := fmt.Sprintf(`{"analyzer": "my_analyzer", "text": %q}`, textToAnalyze)

//nolint:noctx
req, err := http.NewRequest("POST", analyzerPath, strings.NewReader(analyzerPayload))

Check failure on line 140 in internal/store/elasticsearch/es_test.go

View workflow job for this annotation

GitHub Actions / golangci

"POST" can be replaced by http.MethodPost (usestdlibvars)
if err != nil {
return fmt.Errorf("error creating analyzer request: %w", err)
}
req.Header.Add("content-type", "application/json")

res, err := cli.Perform(req)
if err != nil {
return fmt.Errorf("invoke analyzer: %w", err)
}
defer res.Body.Close()
if res.StatusCode != http.StatusOK {
return fmt.Errorf("elasticsearch returned non-200 response: %d", res.StatusCode)
}
var response struct {
Tokens []struct {
Token string `json:"token"`
} `json:"tokens"`
}
err = json.NewDecoder(res.Body).Decode(&response)
if err != nil {
return fmt.Errorf("error decoding response: %w", err)
}
expectTokens := []string{"walk"}
analyzedTokens := []string{}
for _, tok := range response.Tokens {
analyzedTokens = append(analyzedTokens, tok.Token)
}

if reflect.DeepEqual(expectTokens, analyzedTokens) == false {
return fmt.Errorf("expected analyzer to tokenize %q as %v, was %v", textToAnalyze, expectTokens, analyzedTokens)
}
Expand Down

0 comments on commit 1a3ce2a

Please sign in to comment.