95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package arduinocli
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gitlab.hive-brain.de/MDietermann/Archduino/internal/config"
|
|
"gitlab.hive-brain.de/MDietermann/Archduino/internal/module"
|
|
)
|
|
|
|
type fakeRunner struct {
|
|
runCalls [][]string
|
|
outCalls [][]string
|
|
coreListOut []byte
|
|
}
|
|
|
|
func (f *fakeRunner) Run(_ context.Context, name string, args ...string) error {
|
|
f.runCalls = append(f.runCalls, append([]string{name}, args...))
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeRunner) Output(_ context.Context, name string, args ...string) ([]byte, error) {
|
|
f.outCalls = append(f.outCalls, append([]string{name}, args...))
|
|
if filepath.Base(name) == "arduino-cli" && len(args) >= 2 && args[0] == "core" && args[1] == "list" {
|
|
return f.coreListOut, nil
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
type nop struct{}
|
|
|
|
func (nop) Progress(string, int, int, string) {}
|
|
func (nop) Log(string, string) {}
|
|
|
|
func makeTarGz(t *testing.T, entries map[string][]byte) string {
|
|
t.Helper()
|
|
p := filepath.Join(t.TempDir(), "a.tgz")
|
|
f, _ := os.Create(p)
|
|
gz := gzip.NewWriter(f)
|
|
tw := tar.NewWriter(gz)
|
|
for name, data := range entries {
|
|
_ = tw.WriteHeader(&tar.Header{Name: name, Mode: 0o755, Size: int64(len(data))})
|
|
_, _ = tw.Write(data)
|
|
}
|
|
tw.Close()
|
|
gz.Close()
|
|
f.Close()
|
|
return p
|
|
}
|
|
|
|
func TestArduinoCLIDownloadsAndInstallsCore(t *testing.T) {
|
|
home := t.TempDir()
|
|
t.Setenv("HOME", home)
|
|
archive := makeTarGz(t, map[string][]byte{"arduino-cli": []byte("#!/bin/sh\n")})
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
data, _ := os.ReadFile(archive)
|
|
w.Write(data)
|
|
}))
|
|
defer srv.Close()
|
|
|
|
r := &fakeRunner{coreListOut: []byte("ID Installed\n")}
|
|
m := New(config.ArduinoCLIConfig{
|
|
Version: "1.0.0",
|
|
URL: srv.URL + "/arduino-cli.tgz",
|
|
Cores: []string{"arduino:avr"},
|
|
}, r)
|
|
|
|
st, _ := m.Detect(context.Background())
|
|
p, _ := m.Plan(context.Background(), st)
|
|
if err := m.Apply(context.Background(), p, nop{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(home, ".local", "bin", "arduino-cli")); err != nil {
|
|
t.Fatalf("binary not installed: %v", err)
|
|
}
|
|
sawCore := false
|
|
for _, c := range r.runCalls {
|
|
if filepath.Base(c[0]) == "arduino-cli" && len(c) >= 4 && c[1] == "core" && c[2] == "install" && c[3] == "arduino:avr" {
|
|
sawCore = true
|
|
}
|
|
}
|
|
if !sawCore {
|
|
t.Fatalf("core install not run; calls=%v", r.runCalls)
|
|
}
|
|
}
|
|
|
|
var _ module.Module = Module{}
|