99 lines
1.9 KiB
Go
99 lines
1.9 KiB
Go
package imap_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/emersion/go-imap"
|
|
)
|
|
|
|
func TestCommand_Command(t *testing.T) {
|
|
cmd := &imap.Command{
|
|
Tag: "A001",
|
|
Name: "NOOP",
|
|
}
|
|
|
|
if cmd.Command() != cmd {
|
|
t.Error("Command should return itself")
|
|
}
|
|
}
|
|
|
|
func TestCommand_WriteTo_NoArgs(t *testing.T) {
|
|
var b bytes.Buffer
|
|
w := imap.NewWriter(&b)
|
|
|
|
cmd := &imap.Command{
|
|
Tag: "A001",
|
|
Name: "NOOP",
|
|
}
|
|
|
|
if err := cmd.WriteTo(w); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if b.String() != "A001 NOOP\r\n" {
|
|
t.Fatal("Not the expected command")
|
|
}
|
|
}
|
|
|
|
func TestCommand_WriteTo_WithArgs(t *testing.T) {
|
|
var b bytes.Buffer
|
|
w := imap.NewWriter(&b)
|
|
|
|
cmd := &imap.Command{
|
|
Tag: "A002",
|
|
Name: "LOGIN",
|
|
Arguments: []interface{}{"username", "password"},
|
|
}
|
|
|
|
if err := cmd.WriteTo(w); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if b.String() != "A002 LOGIN username password\r\n" {
|
|
t.Fatal("Not the expected command")
|
|
}
|
|
}
|
|
|
|
func TestCommand_Parse_NoArgs(t *testing.T) {
|
|
fields := []interface{}{"a", "NOOP"}
|
|
|
|
cmd := &imap.Command{}
|
|
|
|
if err := cmd.Parse(fields); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cmd.Tag != "a" {
|
|
t.Error("Invalid tag:", cmd.Tag)
|
|
}
|
|
if cmd.Name != "NOOP" {
|
|
t.Error("Invalid name:", cmd.Name)
|
|
}
|
|
if len(cmd.Arguments) != 0 {
|
|
t.Error("Invalid arguments:", cmd.Arguments)
|
|
}
|
|
}
|
|
|
|
func TestCommand_Parse_WithArgs(t *testing.T) {
|
|
fields := []interface{}{"a", "LOGIN", "username", "password"}
|
|
|
|
cmd := &imap.Command{}
|
|
|
|
if err := cmd.Parse(fields); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cmd.Tag != "a" {
|
|
t.Error("Invalid tag:", cmd.Tag)
|
|
}
|
|
if cmd.Name != "LOGIN" {
|
|
t.Error("Invalid name:", cmd.Name)
|
|
}
|
|
if len(cmd.Arguments) != 2 {
|
|
t.Error("Invalid arguments:", cmd.Arguments)
|
|
}
|
|
if username, ok := cmd.Arguments[0].(string); !ok || username != "username" {
|
|
t.Error("Invalid first argument:", cmd.Arguments[0])
|
|
}
|
|
if password, ok := cmd.Arguments[1].(string); !ok || password != "password" {
|
|
t.Error("Invalid second argument:", cmd.Arguments[1])
|
|
}
|
|
}
|