Add byte count tracking to event pairs

Amp-Thread-ID: https://ampcode.com/threads/T-019c8012-eaeb-768d-a264-5a704f3939ef
Co-authored-by: Amp <amp@ampcode.com>

1 files changed, 73 insertions, 0 deletions

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)
+			}
+		})
+	}
+}