added test case
This commit is contained in:
@@ -8,6 +8,7 @@ type Publisher interface {
|
|||||||
Publish(context.Context, string, []byte) error
|
Publish(context.Context, string, []byte) error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// PublisherProvider is the interface for creating new Publisher instances.
|
||||||
type PublisherProvider interface {
|
type PublisherProvider interface {
|
||||||
// New initializes a new publisher instance with the given configuration.
|
// New initializes a new publisher instance with the given configuration.
|
||||||
New(ctx context.Context, config map[string]string) (Publisher, func() error, error)
|
New(ctx context.Context, config map[string]string) (Publisher, func() error, error)
|
||||||
|
|||||||
150
pkg/plugin/implementation/publisher/publisher_test.go
Normal file
150
pkg/plugin/implementation/publisher/publisher_test.go
Normal file
@@ -0,0 +1,150 @@
|
|||||||
|
package publisher
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestGetConnURLSuccess(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
config *Config
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Valid config with credentials",
|
||||||
|
config: &Config{
|
||||||
|
Addr: "localhost:5672",
|
||||||
|
UseTLS: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Valid config with vhost",
|
||||||
|
config: &Config{
|
||||||
|
Addr: "localhost:5672/myvhost",
|
||||||
|
UseTLS: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set valid credentials
|
||||||
|
os.Setenv("RABBITMQ_USERNAME", "guest")
|
||||||
|
os.Setenv("RABBITMQ_PASSWORD", "guest")
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
url, err := GetConnURL(tt.config)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if url == "" {
|
||||||
|
t.Error("expected non-empty URL, got empty string")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetConnURLFailure(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
username string
|
||||||
|
password string
|
||||||
|
config *Config
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Missing credentials",
|
||||||
|
username: "",
|
||||||
|
password: "",
|
||||||
|
config: &Config{Addr: "localhost:5672"},
|
||||||
|
wantErr: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Missing config address",
|
||||||
|
username: "guest",
|
||||||
|
password: "guest",
|
||||||
|
config: &Config{}, // this won't error unless Validate() is called separately
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if tt.username == "" {
|
||||||
|
os.Unsetenv("RABBITMQ_USERNAME")
|
||||||
|
} else {
|
||||||
|
os.Setenv("RABBITMQ_USERNAME", tt.username)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.password == "" {
|
||||||
|
os.Unsetenv("RABBITMQ_PASSWORD")
|
||||||
|
} else {
|
||||||
|
os.Setenv("RABBITMQ_PASSWORD", tt.password)
|
||||||
|
}
|
||||||
|
|
||||||
|
url, err := GetConnURL(tt.config)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("unexpected error. gotErr = %v, wantErr = %v", err != nil, tt.wantErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err == nil && url == "" {
|
||||||
|
t.Errorf("expected non-empty URL, got empty string")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestValidateSuccess(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
config *Config
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Valid config with Addr and Exchange",
|
||||||
|
config: &Config{
|
||||||
|
Addr: "localhost:5672",
|
||||||
|
Exchange: "ex",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if err := Validate(tt.config); err != nil {
|
||||||
|
t.Errorf("expected no error, got: %v", err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestValidateFailure(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
config *Config
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Nil config",
|
||||||
|
config: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Missing Addr",
|
||||||
|
config: &Config{Exchange: "ex"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Missing Exchange",
|
||||||
|
config: &Config{Addr: "localhost:5672"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Empty Addr and Exchange",
|
||||||
|
config: &Config{Addr: " ", Exchange: " "},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
err := Validate(tt.config)
|
||||||
|
if err == nil {
|
||||||
|
t.Errorf("expected error for invalid config, got nil")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user