func (s *systemtestSuite) CheckBgpNoConnectionForaNode(c *C, node vagrantssh.TestbedNode) error {
for i := 0; i < 100; i++ {
time.Sleep(3 * time.Second)
out, _ := node.RunCommandWithOutput("/opt/gopath/bin/gobgp neighbor")
fmt.Println(out)
if !strings.Contains(out, "Establ") {
return nil
}
}
return errors.New("BGP connection persists")
}
func (s *SystemTestSuite) waitForStatToFail(c *C, nut vagrantssh.TestbedNode, file string) {
out, err := tutils.WaitForDone(func() (string, bool) {
cmdStr := fmt.Sprintf("stat -t %s", file)
out, err := nut.RunCommandWithOutput(cmdStr)
if err == nil {
return out, false
}
return out, true
}, 1*time.Second, 10*time.Second, fmt.Sprintf("file %q still seems to exist", file))
s.Assert(c, err, IsNil, Commentf("output: %s", out))
}
func (s *SystemTestSuite) waitForSerfMembership(c *C, nut vagrantssh.TestbedNode, nodeName, state string) {
out, err := tutils.WaitForDone(func() (string, bool) {
out, err := nut.RunCommandWithOutput(`serf members`)
if err != nil {
return out, false
}
stateStr := fmt.Sprintf(`%s.*%s.*`, nodeName, state)
if match, err := regexp.MatchString(stateStr, out); err != nil || !match {
return out, false
}
return out, true
}, 1*time.Second, time.Duration(10)*time.Second,
fmt.Sprintf("%s's serf membership is not in %s state", nodeName, state))
s.Assert(c, err, IsNil, Commentf("output: %s", out))
}
func (s *SystemTestSuite) checkProvisionStatus(c *C, tbn1 vagrantssh.TestbedNode, nodeName, exptdStatus string) {
exptdStr := fmt.Sprintf(`.*"status".*"%s".*`, exptdStatus)
out, err := tutils.WaitForDone(func() (string, bool) {
cmdStr := fmt.Sprintf("clusterctl node get %s --json", nodeName)
out, err := tbn1.RunCommandWithOutput(cmdStr)
if err != nil {
return out, false
//replace newline with empty string for regex to match properly
} else if match, err := regexp.MatchString(exptdStr,
strings.Replace(out, "\n", "", -1)); err == nil && match {
return out, true
}
return out, false
}, 1*time.Second, 30*time.Second, fmt.Sprintf("node is still not in %q status", exptdStatus))
s.Assert(c, err, IsNil, Commentf("output: %s", out))
}
//ServiceStatus queries and returns status result of systemd service unit
func ServiceStatus(n vagrantssh.TestbedNode, srv string) (string, error) {
return n.RunCommandWithOutput(fmt.Sprintf("systemctl status %s", srv))
}
func ClearEtcd(node vagrantssh.TestbedNode) {
log.Infof("Clearing etcd data")
node.RunCommand(`for i in $(etcdctl ls /); do etcdctl rm --recursive "$i"; done`)
}
//ServiceLogs queries and returns upto maxLogLines lines from systemd service unit logs
func ServiceLogs(n vagrantssh.TestbedNode, srv string, maxLogLines int) (string, error) {
return n.RunCommandWithOutput(fmt.Sprintf("sudo systemctl status -ln%d %s", maxLogLines, srv))
}
//ServiceRestart restarts a systemd service unit
func ServiceRestart(n vagrantssh.TestbedNode, srv string) (string, error) {
return n.RunCommandWithOutput(fmt.Sprintf("sudo systemctl restart %s", srv))
}
func (s *SystemTestSuite) touchFileAndWaitForStatToSucceed(c *C, nut vagrantssh.TestbedNode, file string) {
cmdStr := fmt.Sprintf("touch %s", file)
out, err := nut.RunCommandWithOutput(cmdStr)
s.Assert(c, err, IsNil, Commentf("output: %s", out))
s.waitForStatToSucceed(c, nut, file)
}
