Skip to content

Commit

Permalink
#2048[WIP] - add tests to increase coverage
Browse files Browse the repository at this point in the history
Signed-off-by: santosh <bsantosh@thoughtworks.com>
  • Loading branch information
bhiravabhatla committed Jan 22, 2021
1 parent 5ad1b1e commit 3bffcc5
Showing 1 changed file with 38 additions and 2 deletions.
40 changes: 38 additions & 2 deletions plugin/storage/es/factory_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -354,7 +354,7 @@ func TestGetSpanServiceMappings(t *testing.T) {
err: "",
},
{
name: "ES Version 7 Error",
name: "ES Version 7 Service Error",
args: args{
shards: 3,
replicas: 3,
Expand Down Expand Up @@ -392,7 +392,7 @@ func TestGetSpanServiceMappings(t *testing.T) {
err: "",
},
{
name: "ES Version < 7 Error",
name: "ES Version < 7 Service Error",
args: args{
shards: 3,
replicas: 3,
Expand All @@ -410,6 +410,42 @@ func TestGetSpanServiceMappings(t *testing.T) {
},
err: "template load error",
},
{
name: "ES Version < 7 Span Error",
args: args{
shards: 3,
replicas: 3,
esVersion: 6,
esPrefix: "test",
useILM: true,
},
mockNewTextTemplateBuilder: func() es.TemplateBuilder {
tb := mocks.TemplateBuilder{}
ta := mocks.TemplateApplier{}
ta.On("Execute", mock.Anything, mock.Anything).Return(errors.New("template load error"))
tb.On("Parse", mock.Anything).Return(&ta, nil)
return &tb
},
err: "template load error",
},
{
name: "ES Version 7 Span Error",
args: args{
shards: 3,
replicas: 3,
esVersion: 7,
esPrefix: "test",
useILM: true,
},
mockNewTextTemplateBuilder: func() es.TemplateBuilder {
tb := mocks.TemplateBuilder{}
ta := mocks.TemplateApplier{}
ta.On("Execute", mock.Anything, mock.Anything).Return(errors.New("template load error")).Once()
tb.On("Parse", mock.Anything).Return(&ta, nil)
return &tb
},
err: "template load error",
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
Expand Down

0 comments on commit 3bffcc5

Please sign in to comment.