Esempio n. 1
0
// Run the restore through restore.Runner
func (c *RestoreCommand) Run(args []string) int {
	if len(args) != 1 {
		c.UI.Error("You need to specify a restore file path from base of bucket")
		return 1
	}

	c.UI.Info(fmt.Sprintf("v%v: Starting Consul Snapshot", c.Version))
	response := restore.Runner(args[0])
	return response
}
func TestAcceptance(t *testing.T) {
	var err error
	conf := config.ParseConfig(true)
	if conf.Acceptance == false {
		t.Skip("Skipping acceptance test, Set ACCEPTANCE_TEST=1 to run")
	}
	command := "consul"
	commandargs := []string{"agent", "-dev", "-bind=127.0.0.1"}
	cmd := &LocalDevConsul{Command: command, CommandArgs: commandargs}
	go cmd.Run()

	// 5-10 seconds is about the time it takes for consul to wake up
	time.Sleep(5 * time.Second)

	c := consul.Client()

	seedData := &Seeder{}

	t.Log("Adding random seed data to consul kv")
	for i := 1; i < 500; i++ {
		// more entropy is needed here.
		randname := fmt.Sprintf("%s.%s", randomdata.SillyName(), randomdata.SillyName())
		err = putKey(c, randname, randomdata.Address(), seedData)
		if err != nil {
			t.Errorf("Failed putting test data to kv: %v", err)
		}
		err = putKey(c, randomdata.IpV4Address(), randomdata.Paragraph(), seedData)
		if err != nil {
			t.Errorf("Failed putting test data to kv: %v", err)
		}
	}

	t.Log("Starting Backup")

	backup.Runner("test")

	_, err = c.KV().DeleteTree("", nil)
	if err != nil {
		for _, i := range cmd.CommandOut {
			log.Printf("CONSUL: %v", i)
		}
		t.Errorf("Unable to clear consul kv store after backup; %v", err)
	}

	restore.Runner("/tmp/acceptancetest.tar.gz")

	for _, kv := range seedData.Data {
		//log.Printf("SEED: %v | %v", kv.Key, string(kv.Value))
		err := checkKey(c, kv)
		if err != nil {
			for _, i := range cmd.CommandOut {
				log.Printf("CONSUL: %v", i)
			}
			t.Errorf("Key Failure: %v", err)
		}
	}

	// Remove the staging path
	err = os.RemoveAll("/tmp/acceptancetest")
	if err != nil {
		log.Printf("Unable to remove temporary backup file: %v", err)
	}

	err = os.RemoveAll("/tmp/acceptancetest.tar.gz")
	if err != nil {
		log.Printf("Unable to remove temporary backup file: %v", err)
	}

}