1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
package clients
import (
"strings"
"testing"
"time"
"github.com/mimecast/dtail/internal/color"
"github.com/mimecast/dtail/internal/config"
)
func TestNewClientRuntimeBoundaryDefaults(t *testing.T) {
runtime := newClientRuntimeBoundary(config.RuntimeConfig{})
if runtime.SSHPort() != 2222 {
t.Fatalf("Expected default SSH port 2222, got %d", runtime.SSHPort())
}
if runtime.SSHConnectTimeout() != 2*time.Second {
t.Fatalf("Expected default timeout 2s, got %v", runtime.SSHConnectTimeout())
}
if runtime.InterruptPause() != 3*time.Second {
t.Fatalf("Expected default interrupt pause 3s, got %v", runtime.InterruptPause())
}
if got := runtime.output.PaintMaprRawQuery("select 1"); got != "select 1" {
t.Fatalf("Expected plain raw query output, got %q", got)
}
}
func TestNewClientRuntimeBoundaryUsesConfiguredSSHSettings(t *testing.T) {
runtime := newClientRuntimeBoundary(config.RuntimeConfig{
Common: &config.CommonConfig{
SSHPort: 4022,
SSHConnectTimeoutMs: 4500,
},
})
if runtime.SSHPort() != 4022 {
t.Fatalf("Expected configured SSH port 4022, got %d", runtime.SSHPort())
}
if runtime.SSHConnectTimeout() != 4500*time.Millisecond {
t.Fatalf("Expected configured timeout 4.5s, got %v", runtime.SSHConnectTimeout())
}
}
func TestClientOutputFormatterColorModes(t *testing.T) {
plain := newClientOutputFormatter(nil)
if got := plain.FormatInterruptMessage(1, "hello"); got != " hello" {
t.Fatalf("Expected plain interrupt message, got %q", got)
}
if got := plain.PaintMaprRawQuery("select 1"); got != "select 1" {
t.Fatalf("Expected plain raw query, got %q", got)
}
cfg := &config.ClientConfig{TermColorsEnable: true}
cfg.TermColors.Client.ClientFg = color.FgBlack
cfg.TermColors.Client.ClientBg = color.BgYellow
cfg.TermColors.Client.ClientAttr = color.AttrBold
cfg.TermColors.MaprTable.RawQueryFg = color.FgCyan
cfg.TermColors.MaprTable.RawQueryBg = color.BgBlack
cfg.TermColors.MaprTable.RawQueryAttr = color.AttrUnderline
cfg.TermColors.MaprTable.HeaderFg = color.FgWhite
cfg.TermColors.MaprTable.HeaderBg = color.BgBlue
cfg.TermColors.MaprTable.HeaderAttr = color.AttrBold
cfg.TermColors.MaprTable.HeaderDelimiterFg = color.FgWhite
cfg.TermColors.MaprTable.HeaderDelimiterBg = color.BgBlue
cfg.TermColors.MaprTable.HeaderDelimiterAttr = color.AttrDim
cfg.TermColors.MaprTable.HeaderSortKeyAttr = color.AttrUnderline
cfg.TermColors.MaprTable.HeaderGroupKeyAttr = color.AttrReverse
cfg.TermColors.MaprTable.DataFg = color.FgWhite
cfg.TermColors.MaprTable.DataBg = color.BgBlue
cfg.TermColors.MaprTable.DataAttr = color.AttrNone
cfg.TermColors.MaprTable.DelimiterFg = color.FgWhite
cfg.TermColors.MaprTable.DelimiterBg = color.BgBlue
cfg.TermColors.MaprTable.DelimiterAttr = color.AttrDim
colored := newClientOutputFormatter(cfg)
if got := colored.FormatInterruptMessage(0, "hello"); got != " hello" {
t.Fatalf("Expected first interrupt line to stay plain, got %q", got)
}
if got := colored.FormatInterruptMessage(1, "hello"); !strings.Contains(got, "\x1b[") {
t.Fatalf("Expected colored interrupt output, got %q", got)
}
if got := colored.PaintMaprRawQuery("select 1"); !strings.Contains(got, "\x1b[") {
t.Fatalf("Expected colored raw query output, got %q", got)
}
if colored.MaprResultRenderer() == nil {
t.Fatal("Expected non-nil mapreduce result renderer")
}
}
|
