@@ -77,13 +77,17 @@ export const mockDatabaseOverview: DatabaseOverview = {
77
77
describe ( 'OverviewService' , ( ) => {
78
78
let service : DatabaseOverviewProvider ;
79
79
let spyGetNodeInfo ;
80
+ let spyCalculateTotalKeys ;
81
+ let spyCalculateNodesTotalKeys ;
80
82
beforeEach ( async ( ) => {
81
83
const module : TestingModule = await Test . createTestingModule ( {
82
84
providers : [ DatabaseOverviewProvider ] ,
83
85
} ) . compile ( ) ;
84
86
85
87
service = await module . get ( DatabaseOverviewProvider ) ;
86
88
spyGetNodeInfo = jest . spyOn < any , any > ( service , 'getNodeInfo' ) ;
89
+ spyCalculateTotalKeys = jest . spyOn < any , any > ( service , 'calculateTotalKeys' ) ;
90
+ spyCalculateNodesTotalKeys = jest . spyOn < any , any > ( service , 'calculateNodesTotalKeys' ) ;
87
91
mockClient . call = jest . fn ( ) ;
88
92
mockClient . info = jest . fn ( ) ;
89
93
} ) ;
@@ -194,11 +198,10 @@ describe('OverviewService', () => {
194
198
} ) ;
195
199
describe ( 'Cluster' , ( ) => {
196
200
it ( 'Should calculate overview and ignore replica where needed' , async ( ) => {
201
+ const getTotal = jest . spyOn ( Utils , 'getTotal' ) . mockResolvedValue ( mockGetTotalResponse_1 ) ;
197
202
mockCluster . nodes = jest . fn ( )
198
203
. mockReturnValue ( new Array ( 6 ) . fill ( Promise . resolve ( ) ) ) ;
199
204
200
- jest . spyOn ( Utils , 'getTotal' ) . mockResolvedValue ( mockGetTotalResponse_1 ) ;
201
-
202
205
spyGetNodeInfo . mockResolvedValueOnce ( {
203
206
...mockNodeInfo ,
204
207
port : 12001 ,
@@ -238,6 +241,9 @@ describe('OverviewService', () => {
238
241
opsPerSecond : 6 ,
239
242
cpuUsagePercentage : null ,
240
243
} ) ;
244
+ expect ( spyCalculateTotalKeys ) . toHaveBeenCalledTimes ( 0 ) ;
245
+ expect ( spyCalculateNodesTotalKeys ) . toHaveBeenCalledTimes ( 1 ) ;
246
+ expect ( getTotal ) . toHaveBeenCalledTimes ( 6 ) ; // 6 nodes
241
247
242
248
spyGetNodeInfo . mockResolvedValueOnce ( {
243
249
...mockNodeInfo ,
0 commit comments