added test case

This commit is contained in:
MohitKatare-protean
2025-05-06 10:29:23 +05:30
parent ae98cc992f
commit 72f8be52fa
2 changed files with 151 additions and 0 deletions

View File

@@ -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)

View 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")
}
})
}
}