indexes_test.go

package pkgs_test

import (
	"context"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/arduino/arduino-create-agent/gen/indexes"
	"github.com/arduino/arduino-create-agent/v2/pkgs"
)

// TestIndexes performs a series of operations about indexes, ensuring it behaves as expected.
func TestIndexes(t *testing.T) {
	// Use local file as index
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "testdata/package_index.json")
	}))
	defer ts.Close()

	// Initialize indexes with a temp folder
	tmp, err := ioutil.TempDir("", "")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(tmp)

	// Create extraneous folder in temp folder
	os.MkdirAll(filepath.Join(tmp, "arduino"), 0755)

	service := pkgs.Indexes{
		Folder: tmp,
	}

	ctx := context.Background()

	// List indexes, they should be 0
	list, err := service.List(ctx)
	if err != nil {
		t.Fatal(err)
	}
	if len(list) != 0 {
		t.Fatalf("expected %d == %d (%s)", len(list), 0, "len(list)")
	}

	// Add a faulty index
	_, err = service.Add(ctx, &indexes.IndexPayload{URL: ":"})
	if err == nil || !strings.Contains(err.Error(), "parse :: missing protocol scheme") {
		t.Fatalf("expected '%v' == '%v' (%s)", err, "parse :: missing protocol scheme", "err")
	}

	// Add a new index
	_, err = service.Add(ctx, &indexes.IndexPayload{URL: ts.URL})
	if err != nil {
		t.Fatal(err)
	}

	// List indexes, they should be 1
	list, err = service.List(ctx)
	if err != nil {
		t.Fatal(err)
	}
	if len(list) != 1 {
		t.Fatalf("expected %d == %d (%s)", len(list), 1, "len(list)")
	}
	if list[0] != ts.URL {
		t.Fatalf("expected %s == %s (%s)", list[0], "downloads.arduino.cc/packages/package_index.json", "list[0]")
	}

	// Remove the index
	_, err = service.Remove(ctx, &indexes.IndexPayload{URL: ts.URL})
	if err != nil {
		t.Fatal(err)
	}

	// List indexes, they should be 0
	list, err = service.List(ctx)
	if err != nil {
		t.Fatal(err)
	}
	if len(list) != 0 {
		t.Fatalf("expected %d == %d (%s)", len(list), 0, "len(list)")
	}
}