@@ -117,6 +117,7 @@ func TestContainer_Add(t *testing.T) {
117117 BytesRead : 100 ,
118118 RowsRead : 20 ,
119119 RowsWritten : 5 ,
120+ KVCPUTime : 30 ,
120121 Failed : true ,
121122 Generic : true ,
122123 AppliedStmtHints : true ,
@@ -137,6 +138,7 @@ func TestContainer_Add(t *testing.T) {
137138 RowsRead : 20 ,
138139 RowsWritten : 5 ,
139140 BytesRead : 100 ,
141+ KVCPUTime : 30 * time .Nanosecond ,
140142 }
141143 require .NoError (t , src .RecordTransaction (ctx , txnStats ))
142144
@@ -160,6 +162,7 @@ func TestContainer_Add(t *testing.T) {
160162 BytesRead : 60 ,
161163 RowsRead : 70 ,
162164 RowsWritten : 80 ,
165+ KVCPUTime : 90 ,
163166 Failed : true ,
164167 Generic : true ,
165168 AppliedStmtHints : true ,
@@ -177,6 +180,7 @@ func TestContainer_Add(t *testing.T) {
177180 RowsRead : 20 ,
178181 RowsWritten : 5 ,
179182 BytesRead : 100 ,
183+ KVCPUTime : 90 * time .Nanosecond ,
180184 }
181185 require .NoError (t , dest .RecordStatement (ctx , reducedStmtStats ))
182186 require .NoError (t , dest .RecordTransaction (ctx , reducedTxnStats ))
@@ -221,6 +225,7 @@ func verifyStmtStatsMultiple(
221225 require .InEpsilon (t , float64 (stmtStats .BytesRead ), destStmtStats .mu .data .BytesRead .Mean , epsilon )
222226 require .InEpsilon (t , float64 (stmtStats .RowsRead ), destStmtStats .mu .data .RowsRead .Mean , epsilon )
223227 require .InEpsilon (t , float64 (stmtStats .RowsWritten ), destStmtStats .mu .data .RowsWritten .Mean , epsilon )
228+ require .InEpsilon (t , float64 (stmtStats .KVCPUTime ), destStmtStats .mu .data .KVCPUTime .Mean , epsilon )
224229}
225230
226231// verifyStmtStatsReduced verifies that statement statistics have been properly
@@ -241,6 +246,7 @@ func verifyStmtStatsReduced(
241246 require .InEpsilon (t , float64 (stmtStats .BytesRead )/ cnt , destStmtStats .mu .data .BytesRead .Mean , epsilon )
242247 require .InEpsilon (t , float64 (stmtStats .RowsRead )/ cnt , destStmtStats .mu .data .RowsRead .Mean , epsilon )
243248 require .InEpsilon (t , float64 (stmtStats .RowsWritten )/ cnt , destStmtStats .mu .data .RowsWritten .Mean , epsilon )
249+ require .InEpsilon (t , float64 (stmtStats .KVCPUTime )/ cnt , destStmtStats .mu .data .KVCPUTime .Mean , epsilon )
244250}
245251
246252// verifyTxnStatsMultiple verifies that transaction statistics have been recorded
@@ -259,6 +265,7 @@ func verifyTxnStatsMultiple(
259265 require .InEpsilon (t , float64 (txnStats .RowsRead ), destTxnStats .mu .data .RowsRead .Mean , epsilon )
260266 require .InEpsilon (t , float64 (txnStats .RowsWritten ), destTxnStats .mu .data .RowsWritten .Mean , epsilon )
261267 require .InEpsilon (t , float64 (txnStats .BytesRead ), destTxnStats .mu .data .BytesRead .Mean , epsilon )
268+ require .InEpsilon (t , float64 (txnStats .KVCPUTime .Nanoseconds ()), destTxnStats .mu .data .KVCPUTime .Mean , epsilon )
262269}
263270
264271// verifyTxnStatsReduced verifies that transaction statistics have been properly
@@ -277,6 +284,7 @@ func verifyTxnStatsReduced(
277284 require .InEpsilon (t , float64 (txnStats .RowsRead )/ cnt , destTxnStats .mu .data .RowsRead .Mean , epsilon )
278285 require .InEpsilon (t , float64 (txnStats .RowsWritten )/ cnt , destTxnStats .mu .data .RowsWritten .Mean , epsilon )
279286 require .InEpsilon (t , float64 (txnStats .BytesRead )/ cnt , destTxnStats .mu .data .BytesRead .Mean , epsilon )
287+ require .InEpsilon (t , float64 (txnStats .KVCPUTime .Nanoseconds ())/ cnt , destTxnStats .mu .data .KVCPUTime .Mean , epsilon )
280288}
281289
282290func testMonitor (
0 commit comments