add prompt.StubConfirm

This commit is contained in:
vilmibm 2020-07-27 10:00:19 -05:00
parent 4c57566cdc
commit dadcf4ba96
2 changed files with 15 additions and 24 deletions

View file

@ -294,12 +294,7 @@ func TestRepoFork_outside_survey_yes(t *testing.T) {
cs.Stub("") // git clone
cs.Stub("") // git remote add
oldConfirm := prompt.Confirm
prompt.Confirm = func(_ string, result *bool) error {
*result = true
return nil
}
defer func() { prompt.Confirm = oldConfirm }()
defer prompt.StubConfirm(true)()
output, err := RunCommand("repo fork OWNER/REPO")
if err != nil {
@ -329,12 +324,7 @@ func TestRepoFork_outside_survey_no(t *testing.T) {
return &test.OutputStub{}
})()
oldConfirm := prompt.Confirm
prompt.Confirm = func(_ string, result *bool) error {
*result = false
return nil
}
defer func() { prompt.Confirm = oldConfirm }()
defer prompt.StubConfirm(false)()
output, err := RunCommand("repo fork OWNER/REPO")
if err != nil {
@ -367,12 +357,7 @@ func TestRepoFork_in_parent_survey_yes(t *testing.T) {
return &test.OutputStub{}
})()
oldConfirm := prompt.Confirm
prompt.Confirm = func(_ string, result *bool) error {
*result = true
return nil
}
defer func() { prompt.Confirm = oldConfirm }()
defer prompt.StubConfirm(true)()
output, err := RunCommand("repo fork")
if err != nil {
@ -411,12 +396,7 @@ func TestRepoFork_in_parent_survey_no(t *testing.T) {
return &test.OutputStub{}
})()
oldConfirm := prompt.Confirm
prompt.Confirm = func(_ string, result *bool) error {
*result = false
return nil
}
defer func() { prompt.Confirm = oldConfirm }()
defer prompt.StubConfirm(false)()
output, err := RunCommand("repo fork")
if err != nil {