/*
Copyright The Helm Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package main

import (
	"bytes"
	"fmt"
	"io"
	"os"
	"strings"
	"testing"

	"k8s.io/helm/pkg/getter"
	"k8s.io/helm/pkg/helm/helmpath"
	"k8s.io/helm/pkg/repo"
	"k8s.io/helm/pkg/repo/repotest"
)

func TestUpdateCmd(t *testing.T) {
	thome, err := tempHelmHome(t)
	if err != nil {
		t.Fatal(err)
	}

	cleanup := resetEnv()
	defer func() {
		os.RemoveAll(thome.String())
		cleanup()
	}()

	settings.Home = thome

	out := bytes.NewBuffer(nil)
	// Instead of using the HTTP updater, we provide our own for this test.
	// The TestUpdateCharts test verifies the HTTP behavior independently.
	updater := func(repos []*repo.ChartRepository, out io.Writer, hh helmpath.Home, strict bool) error {
		for _, re := range repos {
			fmt.Fprintln(out, re.Config.Name)
		}
		return nil
	}
	uc := &repoUpdateCmd{
		update: updater,
		home:   helmpath.Home(thome),
		out:    out,
	}
	if err := uc.run(); err != nil {
		t.Fatal(err)
	}

	if got := out.String(); !strings.Contains(got, "charts") || !strings.Contains(got, "local") {
		t.Errorf("Expected 'charts' and 'local' (in any order) got %q", got)
	}
}

func TestUpdateCharts(t *testing.T) {
	ts, thome, err := repotest.NewTempServer("testdata/testserver/*.*")
	if err != nil {
		t.Fatal(err)
	}

	hh := helmpath.Home(thome)
	cleanup := resetEnv()
	defer func() {
		ts.Stop()
		os.RemoveAll(thome.String())
		cleanup()
	}()
	if err := ensureTestHome(hh, t); err != nil {
		t.Fatal(err)
	}

	settings.Home = thome

	r, err := repo.NewChartRepository(&repo.Entry{
		Name:  "charts",
		URL:   ts.URL(),
		Cache: hh.CacheIndex("charts"),
	}, getter.All(settings))
	if err != nil {
		t.Error(err)
	}

	b := bytes.NewBuffer(nil)
	updateCharts([]*repo.ChartRepository{r}, b, hh, false)

	got := b.String()
	if strings.Contains(got, "Unable to get an update") {
		t.Errorf("Failed to get a repo: %q", got)
	}
	if !strings.Contains(got, "Update Complete.") {
		t.Error("Update was not successful")
	}
}

func TestUpdateCmdStrictFlag(t *testing.T) {
	thome, err := tempHelmHome(t)
	if err != nil {
		t.Fatal(err)
	}

	cleanup := resetEnv()
	defer func() {
		os.RemoveAll(thome.String())
		cleanup()
	}()

	settings.Home = thome

	out := bytes.NewBuffer(nil)
	cmd := newRepoUpdateCmd(out)
	cmd.ParseFlags([]string{"--strict"})

	if err := cmd.RunE(cmd, []string{}); err == nil {
		t.Fatal("expected error due to strict flag")
	}

	if got := out.String(); !strings.Contains(got, "Unable to get an update") {
		t.Errorf("Expected 'Unable to get an update', got %q", got)
	}
}

func TestUpdateCmdWithSingleRepoNameWhichDoesntExist(t *testing.T) {
	thome, err := tempHelmHome(t)
	if err != nil {
		t.Fatal(err)
	}

	cleanup := resetEnv()
	defer func() {
		os.RemoveAll(thome.String())
		cleanup()
	}()

	settings.Home = thome

	out := bytes.NewBuffer(nil)
	cmd := newRepoUpdateCmd(out)

	if err = cmd.RunE(cmd, []string{"randomRepo"}); err == nil {
		t.Fatal("expected error due to wrong repo name")
	}

	if got := fmt.Sprintf("%v", err); !strings.Contains(got, "no repositories found matching the provided name. Verify if the repo exists") {
		t.Errorf("Expected 'no repositories found matching the provided name. Verify if the repo exists', got %q", got)
	}
}

func TestUpdateRepo(t *testing.T) {
	ts, thome, err := repotest.NewTempServer("testdata/testserver/*.*")
	if err != nil {
		t.Fatal(err)
	}

	hh := helmpath.Home(thome)
	cleanup := resetEnv()
	defer func() {
		ts.Stop()
		os.RemoveAll(thome.String())
		cleanup()
	}()
	if err := ensureTestHome(hh, t); err != nil {
		t.Fatal(err)
	}

	settings.Home = thome

	if err := addRepository("repo1", ts.URL(), "", "", hh, "", "", "", true); err != nil {
		t.Error(err)
	}

	if err := addRepository("repo2", ts.URL(), "", "", hh, "", "", "", true); err != nil {
		t.Error(err)
	}

	out := bytes.NewBuffer(nil)
	cmd := newRepoUpdateCmd(out)

	if err = cmd.RunE(cmd, []string{"repo1"}); err != nil {
		t.Fatal("expected to update repo1 correctly")
	}

	got := out.String()

	if !strings.Contains(got, "Successfully got an update from the \"repo1\"") {
		t.Errorf("Expected to successfully update \"repo1\" repository, got %q", got)
	}

	if strings.Contains(got, "Successfully got an update from the \"repo2\"") {
		t.Errorf("Shouldn't have updated \"repo2\" repository, got %q", got)
	}

	if !strings.Contains(got, "Update Complete.") {
		t.Error("Update was not successful")
	}
}