gudae
2 months ago
14 changed files with 132 additions and 23 deletions
@ -0,0 +1,28 @@ |
|||||
|
package mock |
||||
|
|
||||
|
import ( |
||||
|
"testing" |
||||
|
|
||||
|
"tera-clone/services/tera-api/internal/domain" |
||||
|
) |
||||
|
|
||||
|
func TestPortListenerConflictsAreScopedToOperationInterface(t *testing.T) { |
||||
|
ssh := domain.PortOperation{Mode: "reverse_ssh", Port: 10001} |
||||
|
telnet := domain.PortOperation{Mode: "reverse_telnet", Port: 10001} |
||||
|
for _, test := range []struct { |
||||
|
name string |
||||
|
configs map[string]portConfig |
||||
|
conflict bool |
||||
|
}{ |
||||
|
{"same channel different interfaces", map[string]portConfig{"CH01": {Operation1: ssh, Operation2: telnet}}, false}, |
||||
|
{"different channels same eth0", map[string]portConfig{"CH01": {Operation1: ssh}, "CH02": {Operation1: telnet}}, true}, |
||||
|
{"different channels same eth1", map[string]portConfig{"CH01": {Operation2: ssh}, "CH02": {Operation2: telnet}}, true}, |
||||
|
{"different channels different interfaces", map[string]portConfig{"CH01": {Operation1: ssh}, "CH02": {Operation2: telnet}}, false}, |
||||
|
} { |
||||
|
t.Run(test.name, func(t *testing.T) { |
||||
|
if err := validatePortOperationConflicts(test.configs); (err != nil) != test.conflict { |
||||
|
t.Fatalf("conflict=%v, error=%v", test.conflict, err) |
||||
|
} |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,40 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"net/http" |
||||
|
"path/filepath" |
||||
|
"testing" |
||||
|
|
||||
|
"tera-clone/services/tera-api/internal/adapters/mock" |
||||
|
"tera-clone/services/tera-api/internal/domain" |
||||
|
) |
||||
|
|
||||
|
func TestOperationTwoTelnetCanBeStagedAppliedSavedAndReloadedBesideSSH(t *testing.T) { |
||||
|
path := filepath.Join(t.TempDir(), "config.json") |
||||
|
store, err := mock.NewStoreWithConfigFile(fixedClock, path) |
||||
|
if err != nil { |
||||
|
t.Fatal(err) |
||||
|
} |
||||
|
handler := NewRouter(store) |
||||
|
// Use the same separate PATCH + apply requests sent by the two UI tabs.
|
||||
|
getJSONAs(t, handler, http.MethodPatch, "/api/v1/ports/CH01", `{"operation_1":{"mode":"reverse_ssh","port":10001,"max_clients":2,"authentication":"local","newline":"cr"}}`, "mock-admin") |
||||
|
getJSONAs(t, handler, http.MethodPost, "/api/v1/config/apply", "", "mock-admin") |
||||
|
getJSONAs(t, handler, http.MethodPatch, "/api/v1/ports/CH01", `{"operation_2":{"mode":"reverse_telnet","port":10001,"max_clients":4,"authentication":"local","newline":"cr"}}`, "mock-admin") |
||||
|
getJSONAs(t, handler, http.MethodPost, "/api/v1/config/apply", "", "mock-admin") |
||||
|
assertDualOperation(t, getPort(t, handler, "CH01")) |
||||
|
getJSONAs(t, handler, http.MethodPost, "/api/v1/config/save", "", "mock-admin") |
||||
|
reloaded, err := mock.NewStoreWithConfigFile(fixedClock, path) |
||||
|
if err != nil { |
||||
|
t.Fatal(err) |
||||
|
} |
||||
|
assertDualOperation(t, getPort(t, NewRouter(reloaded), "CH01")) |
||||
|
} |
||||
|
|
||||
|
func assertDualOperation(t *testing.T, port domain.Port) { |
||||
|
t.Helper() |
||||
|
if port.Operation1.Mode != "reverse_ssh" || port.Operation2.Mode != "reverse_telnet" || |
||||
|
port.Operation1.Port != 10001 || port.Operation2.Port != 10001 || |
||||
|
port.Operation1.MaxClients != 2 || port.Operation2.MaxClients != 2 { |
||||
|
t.Fatalf("CH01 operations not preserved: op1=%+v op2=%+v", port.Operation1, port.Operation2) |
||||
|
} |
||||
|
} |
||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue