@@ -103,7 +103,7 @@ var userContext = entities.UserContext{
103
103
104
104
func BuildTestImpressionEvent () UserEvent {
105
105
tc := TestConfig {}
106
- impressionUserEvent , _ := CreateImpressionUserEvent (tc , testExperiment , & testVariation , userContext , "" , testExperiment .Key , "experiment" )
106
+ impressionUserEvent , _ := CreateImpressionUserEvent (tc , testExperiment , & testVariation , userContext , "" , testExperiment .Key , "experiment" , true )
107
107
return impressionUserEvent
108
108
}
109
109
@@ -195,20 +195,42 @@ func TestCreateImpressionUserEvent(t *testing.T) {
195
195
}
196
196
197
197
for _ , scenario := range scenarios {
198
- _ , ok := CreateImpressionUserEvent (tc , testExperiment , & testVariation , userContext , "" , testExperiment .Key , scenario .flagType )
198
+ userEvent , ok := CreateImpressionUserEvent (tc , testExperiment , & testVariation , userContext , "" , testExperiment .Key , scenario .flagType , true )
199
199
assert .Equal (t , ok , scenario .expected )
200
+
201
+ if ok {
202
+ metaData := userEvent .Impression .Metadata
203
+ assert .Equal (t , "" , metaData .FlagKey )
204
+ assert .Equal (t , testExperiment .Key , metaData .RuleKey )
205
+ assert .Equal (t , scenario .flagType , metaData .RuleType )
206
+ assert .Equal (t , true , metaData .Enabled )
207
+ }
200
208
}
201
209
202
210
// nil variation should _always_ return false
203
211
for _ , scenario := range scenarios {
204
- _ , ok := CreateImpressionUserEvent (tc , testExperiment , nil , userContext , "" , testExperiment .Key , scenario .flagType )
212
+ userEvent , ok := CreateImpressionUserEvent (tc , testExperiment , nil , userContext , "" , testExperiment .Key , scenario .flagType , false )
205
213
assert .False (t , ok )
214
+ if ok {
215
+ metaData := userEvent .Impression .Metadata
216
+ assert .Equal (t , "" , metaData .FlagKey )
217
+ assert .Equal (t , testExperiment .Key , metaData .RuleKey )
218
+ assert .Equal (t , scenario .flagType , metaData .RuleType )
219
+ assert .Equal (t , false , metaData .Enabled )
220
+ }
206
221
}
207
222
208
223
// should _always_ return true if sendFlagDecisions is set
209
224
tc .sendFlagDecisions = true
210
225
for _ , scenario := range scenarios {
211
- _ , ok := CreateImpressionUserEvent (tc , testExperiment , nil , userContext , "" , testExperiment .Key , scenario .flagType )
226
+ userEvent , ok := CreateImpressionUserEvent (tc , testExperiment , nil , userContext , "" , testExperiment .Key , scenario .flagType , true )
212
227
assert .True (t , ok )
228
+ if ok {
229
+ metaData := userEvent .Impression .Metadata
230
+ assert .Equal (t , "" , metaData .FlagKey )
231
+ assert .Equal (t , testExperiment .Key , metaData .RuleKey )
232
+ assert .Equal (t , scenario .flagType , metaData .RuleType )
233
+ assert .Equal (t , true , metaData .Enabled )
234
+ }
213
235
}
214
236
}
0 commit comments