144 lines
3.1 KiB
Go
144 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestParseConfigFindsShares(t *testing.T) {
|
|
doc, err := ParseConfig(`
|
|
# comment
|
|
[global]
|
|
workgroup = WORKGROUP
|
|
|
|
[media]
|
|
path = /srv/media
|
|
valid users = alice, bob
|
|
guest ok = no
|
|
`)
|
|
if err != nil {
|
|
t.Fatalf("ParseConfig() error = %v", err)
|
|
}
|
|
|
|
shares := doc.ShareSections()
|
|
if len(shares) != 1 {
|
|
t.Fatalf("expected 1 share, got %d", len(shares))
|
|
}
|
|
|
|
cfg := ShareFromSection(shares[0])
|
|
if cfg.Name != "media" {
|
|
t.Fatalf("expected share name media, got %q", cfg.Name)
|
|
}
|
|
if cfg.Path != "/srv/media" {
|
|
t.Fatalf("expected path /srv/media, got %q", cfg.Path)
|
|
}
|
|
if strings.Join(cfg.ValidUsers, ",") != "alice,bob" {
|
|
t.Fatalf("expected users alice,bob, got %v", cfg.ValidUsers)
|
|
}
|
|
}
|
|
|
|
func TestBuildShareSectionUpdatesValues(t *testing.T) {
|
|
section := &Section{
|
|
Name: "public",
|
|
Lines: []Line{
|
|
{Key: "path", Value: "/old", IsKV: true},
|
|
{Key: "comment", Value: "old", IsKV: true},
|
|
},
|
|
}
|
|
|
|
updated := BuildShareSection(section, ShareConfig{
|
|
Name: "public",
|
|
Path: "/srv/public",
|
|
Comment: "",
|
|
Browseable: "yes",
|
|
ReadOnly: "no",
|
|
GuestOK: "yes",
|
|
ValidUsers: []string{"alice"},
|
|
})
|
|
|
|
cfg := ShareFromSection(updated)
|
|
if cfg.Path != "/srv/public" {
|
|
t.Fatalf("expected updated path, got %q", cfg.Path)
|
|
}
|
|
if cfg.Comment != "" {
|
|
t.Fatalf("expected comment removed, got %q", cfg.Comment)
|
|
}
|
|
if strings.Join(cfg.ValidUsers, ",") != "alice" {
|
|
t.Fatalf("expected alice, got %v", cfg.ValidUsers)
|
|
}
|
|
}
|
|
|
|
func TestSerializeProducesSectionHeaders(t *testing.T) {
|
|
doc := &Document{
|
|
Preamble: []string{"# smb.conf"},
|
|
Sections: []*Section{
|
|
BuildShareSection(nil, ShareConfig{
|
|
Name: "docs",
|
|
Path: "/srv/docs",
|
|
Browseable: "yes",
|
|
ReadOnly: "no",
|
|
GuestOK: "no",
|
|
}),
|
|
},
|
|
}
|
|
|
|
out := doc.Serialize()
|
|
for _, want := range []string{
|
|
"# smb.conf",
|
|
"[docs]",
|
|
"path = /srv/docs",
|
|
"browseable = yes",
|
|
} {
|
|
if !strings.Contains(out, want) {
|
|
t.Fatalf("serialized output missing %q:\n%s", want, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDetectSambaInstallPlanPrefersApt(t *testing.T) {
|
|
lookPath := func(file string) (string, error) {
|
|
switch file {
|
|
case "apt-get", "sudo":
|
|
return "/usr/bin/" + file, nil
|
|
default:
|
|
return "", execErr(file)
|
|
}
|
|
}
|
|
|
|
plan, ok := DetectSambaInstallPlan(lookPath, false)
|
|
if !ok {
|
|
t.Fatalf("expected install plan")
|
|
}
|
|
if plan.ManagerName != "apt" {
|
|
t.Fatalf("expected apt manager, got %q", plan.ManagerName)
|
|
}
|
|
if plan.DisplayCommand() != "sudo sh -c apt-get update && apt-get install -y samba" {
|
|
t.Fatalf("unexpected command: %q", plan.DisplayCommand())
|
|
}
|
|
}
|
|
|
|
func TestDetectSambaInstallPlanWithoutSudo(t *testing.T) {
|
|
lookPath := func(file string) (string, error) {
|
|
switch file {
|
|
case "pacman":
|
|
return "/usr/bin/pacman", nil
|
|
default:
|
|
return "", execErr(file)
|
|
}
|
|
}
|
|
|
|
plan, ok := DetectSambaInstallPlan(lookPath, false)
|
|
if !ok {
|
|
t.Fatalf("expected install plan")
|
|
}
|
|
if plan.DisplayCommand() != "pacman -Sy --noconfirm samba" {
|
|
t.Fatalf("unexpected command: %q", plan.DisplayCommand())
|
|
}
|
|
}
|
|
|
|
type execErr string
|
|
|
|
func (e execErr) Error() string {
|
|
return "not found: " + string(e)
|
|
}
|