From e949b616ce4511801ff70a4644c29ef920727419 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Sat, 21 Feb 2026 14:31:29 +0200 Subject: Add byte count tracking to event pairs Amp-Thread-ID: https://ampcode.com/threads/T-019c8012-eaeb-768d-a264-5a704f3939ef Co-authored-by: Amp --- internal/eventloop_bytes_test.go | 73 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 73 insertions(+) create mode 100644 internal/eventloop_bytes_test.go (limited to 'internal/eventloop_bytes_test.go') diff --git a/internal/eventloop_bytes_test.go b/internal/eventloop_bytes_test.go new file mode 100644 index 0000000..ed7f7af --- /dev/null +++ b/internal/eventloop_bytes_test.go @@ -0,0 +1,73 @@ +package internal + +import ( + "testing" + + "ior/internal/types" +) + +func TestBytesFromRet(t *testing.T) { + tests := []struct { + name string + retEvent *types.RetEvent + expected uint64 + }{ + {name: "nil", retEvent: nil, expected: 0}, + { + name: "negative", + retEvent: &types.RetEvent{ + Ret: -1, + RetType: types.READ_CLASSIFIED, + }, + expected: 0, + }, + { + name: "zero", + retEvent: &types.RetEvent{ + Ret: 0, + RetType: types.READ_CLASSIFIED, + }, + expected: 0, + }, + { + name: "unclassified", + retEvent: &types.RetEvent{ + Ret: 512, + RetType: types.UNCLASSIFIED, + }, + expected: 0, + }, + { + name: "read", + retEvent: &types.RetEvent{ + Ret: 128, + RetType: types.READ_CLASSIFIED, + }, + expected: 128, + }, + { + name: "write", + retEvent: &types.RetEvent{ + Ret: 256, + RetType: types.WRITE_CLASSIFIED, + }, + expected: 256, + }, + { + name: "transfer", + retEvent: &types.RetEvent{ + Ret: 1024, + RetType: types.TRANSFER_CLASSIFIED, + }, + expected: 1024, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := bytesFromRet(tt.retEvent); got != tt.expected { + t.Errorf("bytesFromRet() = %d, want %d", got, tt.expected) + } + }) + } +} -- cgit v1.2.3