42 lines
984 B
Go
42 lines
984 B
Go
|
package factory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetSMSNotifier(t *testing.T) {
|
||
|
|
notification := Notification{"5552233", "Hello world!"}
|
||
|
|
notifier, err := GetNotifier(SMS)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Notifier type 'SMS' must be returned")
|
||
|
|
}
|
||
|
|
|
||
|
|
resultStr := notifier.SendNotification(notification)
|
||
|
|
if !strings.Contains(resultStr, "was sent to number") {
|
||
|
|
t.Error("SMS notifier result message was not correct.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetEmailNotifier(t *testing.T) {
|
||
|
|
notification := Notification{"5552233", "Hello world!"}
|
||
|
|
notifier, err := GetNotifier(Email)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Notifier type 'Email' must be returned")
|
||
|
|
}
|
||
|
|
|
||
|
|
resultStr := notifier.SendNotification(notification)
|
||
|
|
if !strings.Contains(resultStr, "was sent to email") {
|
||
|
|
t.Error("Email notifier result message was not correct.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetNonExistanceNotifier(t *testing.T) {
|
||
|
|
_, err := GetNotifier(3)
|
||
|
|
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("Error must be returned for unrecognized notifier type.")
|
||
|
|
}
|
||
|
|
}
|