package command

import (
	"bytes"
	"io/ioutil"
	"os"
	"testing"

	"github.com/hashicorp/terraform/remote"
	"github.com/hashicorp/terraform/terraform"
	"github.com/mitchellh/cli"
)

func TestPush_noRemote(t *testing.T) {
	tmp, cwd := testCwd(t)
	defer fixDir(tmp, cwd)

	ui := new(cli.MockUi)
	c := &PushCommand{
		Meta: Meta{
			ContextOpts: testCtxConfig(testProvider()),
			Ui:          ui,
		},
	}

	args := []string{}
	if code := c.Run(args); code != 1 {
		t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
	}
}

func TestPush_cliRemote_noState(t *testing.T) {
	tmp, cwd := testCwd(t)
	defer fixDir(tmp, cwd)

	s := terraform.NewState()
	conf, srv := testRemoteState(t, s, 200)
	defer srv.Close()

	ui := new(cli.MockUi)
	c := &PushCommand{
		Meta: Meta{
			ContextOpts: testCtxConfig(testProvider()),
			Ui:          ui,
		},
	}

	// Remote with no local state!
	args := []string{"-remote", conf.Name, "-remote-server", conf.Server}
	if code := c.Run(args); code != 1 {
		t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
	}
}

func TestPush_cliRemote_withState(t *testing.T) {
	tmp, cwd := testCwd(t)
	defer fixDir(tmp, cwd)

	s := terraform.NewState()
	conf, srv := testRemoteState(t, s, 200)
	defer srv.Close()

	s = terraform.NewState()
	s.Serial = 10

	// Store the local state
	buf := bytes.NewBuffer(nil)
	terraform.WriteState(s, buf)
	err := ioutil.WriteFile(DefaultStateFilename, buf.Bytes(), 0777)
	if err != nil {
		t.Fatalf("Err: %v", err)
	}

	ui := new(cli.MockUi)
	c := &PushCommand{
		Meta: Meta{
			ContextOpts: testCtxConfig(testProvider()),
			Ui:          ui,
		},
	}

	// Remote with default state file
	args := []string{"-remote", conf.Name, "-remote-server", conf.Server}
	if code := c.Run(args); code != 0 {
		t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
	}

	// Should backup state
	if _, err := os.Stat(DefaultStateFilename + DefaultBackupExtention); err != nil {
		t.Fatalf("err: %v", err)
	}

	// Should enable remote state
	if _, err := os.Stat(remote.LocalDirectory + "/" + remote.HiddenStateFile); err != nil {
		t.Fatalf("err: %v", err)
	}
}

func TestPush_local(t *testing.T) {
	tmp, cwd := testCwd(t)
	defer fixDir(tmp, cwd)

	s := terraform.NewState()
	s.Serial = 5
	conf, srv := testRemoteState(t, s, 200)
	defer srv.Close()

	s = terraform.NewState()
	s.Serial = 10
	s.Remote = conf

	// Store the local state
	buf := bytes.NewBuffer(nil)
	terraform.WriteState(s, buf)
	remote.EnsureDirectory()
	remote.Persist(buf)

	ui := new(cli.MockUi)
	c := &PushCommand{
		Meta: Meta{
			ContextOpts: testCtxConfig(testProvider()),
			Ui:          ui,
		},
	}
	args := []string{}
	if code := c.Run(args); code != 0 {
		t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
	}
}