/
fileproxy_test.go
124 lines (88 loc) · 2.63 KB
/
fileproxy_test.go
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main
import (
"testing"
"github.com/roelrymenants/fileproxy/commands"
"github.com/roelrymenants/fileproxy/proxyconfig"
)
var source, destination, firstDestination = "http://www.google.com/some/source", "/some/destination", "/first/destination"
var completeAddExample = []string{"proxytool", "add", "-force", source, destination}
func TestAddCommandFlagParsing(t *testing.T) {
cmd, err := commands.ParseAddCommand(completeAddExample[2:])
if err != nil {
t.Error("Error parsing command")
}
addCommand, ok := cmd.(*commands.AddCommand)
if !ok {
t.Fatal("Parsed command is not an AddCommand")
}
if !addCommand.IsForce {
t.Error("Force option not parsed")
}
if addCommand.Source.String() != source {
t.Errorf("Source not parsed, found %s", addCommand.Source)
}
if addCommand.Destination != destination {
t.Errorf("Destination not parsed, found %s", addCommand.Destination)
}
}
func TestAddCommandBasic(t *testing.T) {
cmd, err := commands.ParseAddCommand(completeAddExample[2:])
if err != nil {
t.Error("Error parsing command")
}
addCommand, ok := cmd.(*commands.AddCommand)
if !ok {
t.Fatal("Parsed command is not an AddCommand")
}
addCommand.IsForce = false
config := proxyconfig.NewConfig()
addCommand.Execute(func() (*proxyconfig.Config, error) {
return config, nil
})
put, ok := config.Rewrites[source]
if !ok || put != destination {
t.Errorf("Source-dest mapping not made")
}
}
func TestAddCommandForceOverwrite(t *testing.T) {
cmd, err := commands.ParseAddCommand(completeAddExample[2:])
if err != nil {
t.Error("Error parsing command")
}
addCommand, ok := cmd.(*commands.AddCommand)
if !ok {
t.Fatal("Parsed command is not an AddCommand")
}
config := proxyconfig.NewConfig()
config.Rewrites[source] = firstDestination
addCommand.Execute(func() (*proxyconfig.Config, error) {
return config, nil
})
put, ok := config.Rewrites[source]
if !ok || put != destination {
t.Errorf("Source-dest mapping not made")
}
}
func TestAddCommandNoForceNoOverwrite(t *testing.T) {
cmd, err := commands.ParseAddCommand(completeAddExample[2:])
if err != nil {
t.Error("Error parsing command")
}
addCommand, ok := cmd.(*commands.AddCommand)
if !ok {
t.Fatal("Parsed command is not an AddCommand")
}
addCommand.IsForce = false
config := proxyconfig.NewConfig()
config.Rewrites[source] = firstDestination
err = addCommand.Execute(func() (*proxyconfig.Config, error) {
return config, nil
})
put, ok := config.Rewrites[source]
if err == nil {
t.Errorf("No error thrown on existing key")
}
if !ok || put == destination {
t.Errorf("Source-dest mapping made, force not defined")
}
}