mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
Add error handler for Mongo GetCDrs remove
This commit is contained in:
committed by
Dan Christian Bogos
parent
c83d097e9e
commit
43daf3efe5
@@ -1102,6 +1102,9 @@ func (ms *MongoStorage) GetCDRs(qryFltr *utils.CDRsFilter, remove bool) ([]*CDR,
|
||||
var chgd int64
|
||||
err := ms.query(func(sctx mongo.SessionContext) (err error) {
|
||||
dr, err := ms.getCol(ColCDRs).DeleteMany(sctx, filters)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
chgd = dr.DeletedCount
|
||||
return err
|
||||
})
|
||||
|
||||
@@ -44,6 +44,7 @@ var sTestsCdrIT = []func(t *testing.T){
|
||||
testCdrITConnect,
|
||||
testCdrITFlush,
|
||||
testCdrITMigrateAndMove,
|
||||
testMongoGetCdrsRemoveErr,
|
||||
}
|
||||
|
||||
func TestCdrITMongo(t *testing.T) {
|
||||
@@ -187,3 +188,16 @@ func testCdrITMigrateAndMove(t *testing.T) {
|
||||
t.Errorf("Unexpected version returned: %d", vrs[utils.CDRs])
|
||||
}
|
||||
}
|
||||
|
||||
func testMongoGetCdrsRemoveErr(t *testing.T) {
|
||||
cdrMigrator.Close()
|
||||
|
||||
expErr := "client is disconnected"
|
||||
if cdrPathIn == "/usr/share/cgrates/conf/samples/tutmongo" {
|
||||
_, _, err := cdrMigrator.storDBOut.StorDB().GetCDRs(new(utils.CDRsFilter), true)
|
||||
if err == nil || err.Error() != expErr {
|
||||
t.Errorf("Expected error <%v>, Received <%v>", expErr, err)
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user