| 34 | const useEtcdEnv = "USE_ETCD_DURING_MIGRATIONS" |
| 35 | |
| 36 | func getContextWithConfig(configFile string, useEtcd bool) *cli.Context { |
| 37 | if useEtcd { |
| 38 | os.Setenv(useEtcdEnv, "true") |
| 39 | } |
| 40 | |
| 41 | configFlag := cli.StringFlag{Name: "config-file", Value: configFile} |
| 42 | useEtcdFlag := cli.BoolFlag{Name: FlagLockWithETCD, EnvVar: useEtcdEnv} |
| 43 | |
| 44 | set := flag.NewFlagSet("", flag.ContinueOnError) |
| 45 | configFlag.Apply(set) |
| 46 | useEtcdFlag.Apply(set) |
| 47 | |
| 48 | return cli.NewContext(nil, set, &cli.Context{}) |
| 49 | } |
| 50 | |
| 51 | var _ = Describe("CLI", func() { |
| 52 | const ( |