mirror of https://github.com/hashicorp/consul
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
255 lines
6.5 KiB
255 lines
6.5 KiB
package command |
|
|
|
import ( |
|
"fmt" |
|
"io/ioutil" |
|
"path/filepath" |
|
"strings" |
|
"testing" |
|
"time" |
|
|
|
"github.com/hashicorp/consul/api" |
|
"github.com/mitchellh/cli" |
|
) |
|
|
|
func TestLockCommand_implements(t *testing.T) { |
|
var _ cli.Command = &LockCommand{} |
|
} |
|
|
|
func argFail(t *testing.T, args []string, expected string) { |
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
if code := c.Run(args); code != 1 { |
|
t.Fatalf("expected return code 1, got %d", code) |
|
} |
|
if reason := ui.ErrorWriter.String(); !strings.Contains(reason, expected) { |
|
t.Fatalf("bad reason: got='%s', expected='%s'", reason, expected) |
|
} |
|
|
|
} |
|
|
|
func TestLockCommand_BadArgs(t *testing.T) { |
|
argFail(t, []string{"-try=blah", "test/prefix", "date"}, "parsing try timeout") |
|
argFail(t, []string{"-try=0s", "test/prefix", "date"}, "timeout must be positive") |
|
argFail(t, []string{"-try=-10s", "test/prefix", "date"}, "timeout must be positive") |
|
argFail(t, []string{"-monitor-retry=-5", "test/prefix", "date"}, "must be >= 0") |
|
} |
|
|
|
func TestLockCommand_Run(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "test/prefix", touchCmd} |
|
|
|
code := c.Run(args) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
|
|
// Check for the file |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
} |
|
|
|
func TestLockCommand_Try_Lock(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "-try=10s", "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the try options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.LockOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if !opts.LockTryOnce || opts.LockWaitTime != 10*time.Second { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
} |
|
|
|
func TestLockCommand_Try_Semaphore(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "-n=3", "-try=10s", "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the try options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.SemaphoreOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if !opts.SemaphoreTryOnce || opts.SemaphoreWaitTime != 10*time.Second { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
} |
|
|
|
func TestLockCommand_MonitorRetry_Lock_Default(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the monitor options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.LockOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if opts.MonitorRetries != defaultMonitorRetry || |
|
opts.MonitorRetryTime != defaultMonitorRetryTime { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
} |
|
|
|
func TestLockCommand_MonitorRetry_Semaphore_Default(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "-n=3", "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the monitor options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.SemaphoreOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if opts.MonitorRetries != defaultMonitorRetry || |
|
opts.MonitorRetryTime != defaultMonitorRetryTime { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
} |
|
|
|
func TestLockCommand_MonitorRetry_Lock_Arg(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "-monitor-retry=9", "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the monitor options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.LockOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if opts.MonitorRetries != 9 || |
|
opts.MonitorRetryTime != defaultMonitorRetryTime { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
} |
|
|
|
func TestLockCommand_MonitorRetry_Semaphore_Arg(t *testing.T) { |
|
a1 := testAgent(t) |
|
defer a1.Shutdown() |
|
waitForLeader(t, a1.httpAddr) |
|
|
|
ui := new(cli.MockUi) |
|
c := &LockCommand{Ui: ui} |
|
filePath := filepath.Join(a1.dir, "test_touch") |
|
touchCmd := fmt.Sprintf("touch '%s'", filePath) |
|
args := []string{"-http-addr=" + a1.httpAddr, "-n=3", "-monitor-retry=9", "test/prefix", touchCmd} |
|
|
|
// Run the command. |
|
var lu *LockUnlock |
|
code := c.run(args, &lu) |
|
if code != 0 { |
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) |
|
} |
|
_, err := ioutil.ReadFile(filePath) |
|
if err != nil { |
|
t.Fatalf("err: %v", err) |
|
} |
|
|
|
// Make sure the monitor options were set correctly. |
|
opts, ok := lu.rawOpts.(*api.SemaphoreOptions) |
|
if !ok { |
|
t.Fatalf("bad type") |
|
} |
|
if opts.MonitorRetries != 9 || |
|
opts.MonitorRetryTime != defaultMonitorRetryTime { |
|
t.Fatalf("bad: %#v", opts) |
|
} |
|
}
|
|
|