Implement PacketEngine struct and test

agwpe_test.go

@@ -1 +1,105 @@
-agwpe_test.go
+package agwpe
+
+import (
+	"testing"
+	"time"
+)
+
+func TestNewPacketEngine(t *testing.T) {
+	tests := []struct {
+		name            string
+		address         string
+		options         []Option
+		expectedError   bool
+		expectedTimeout time.Duration
+	}{
+		{
+			name:            "Default configuration",
+			address:         "localhost:1234",
+			options:         []Option{},
+			expectedError:   false,
+			expectedTimeout: 5 * time.Second,
+		},
+		{
+			name:            "Custom timeout",
+			address:         "localhost:1234",
+			options:         []Option{WithTimeout(30 * time.Second)},
+			expectedError:   false,
+			expectedTimeout: 30 * time.Second,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			engine, err := NewPacketEngine(tt.address, tt.options...)
+
+			if (err != nil) != tt.expectedError {
+				t.Errorf("Expected error: %v, got: %v", tt.expectedError, err)
+			}
+
+			if !tt.expectedError {
+				if engine == nil {
+					t.Fatal("Expected non-nil engine")
+				}
+				if engine.Ready {
+					t.Fatal("Expected engine that is not ready")
+				}
+				if engine.cfg.Timeout != tt.expectedTimeout {
+					t.Errorf("Expected timeout %v, got %v", tt.expectedTimeout, engine.cfg.Timeout)
+				}
+			}
+
+		})
+	}
+}
+
+func TestConnect(t *testing.T) {
+	tests := []struct {
+		name          string
+		address       string
+		options       []Option
+		expectedError bool
+	}{
+		{
+			name:          "Successful connection",
+			address:       "localhost:1234",
+			expectedError: false,
+		},
+		{
+			name:          "Invalid address",
+			address:       "invalid-address",
+			expectedError: true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			engine, _ := NewPacketEngine(tt.address, tt.options...)
+			err := engine.Connect()
+
+			if tt.expectedError && (err == nil) {
+				t.Errorf("Expected error: %v, got: %v", tt.expectedError, err)
+			}
+
+			if !tt.expectedError && (err != nil) {
+				t.Errorf("Did not expect error: %v", err)
+			}
+		})
+	}
+}
+
+func TestDisconnect(t *testing.T) {
+	engine, _ := NewPacketEngine("localhost:1234")
+	engine.Connect()
+
+	err := engine.Disconnect()
+	if err != nil {
+		t.Errorf("Did not expect error: %v", err)
+	}
+
+	// Call to Disconnect() on disconnected packet engine should return an error
+	err = engine.Disconnect()
+	if err == nil {
+		t.Error("Expected an error, but got", nil)
+	}
+}