more tests, fixes

getantibody · Sep 5, 2015 · 895db31 · 895db31
1 parent 5381737
commit 895db31
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/antibody_test.go b/antibody_test.go
@@ -5,16 +5,17 @@ import (
 	"testing"
 
 	"github.com/caarlos0/antibody"
+	"github.com/caarlos0/antibody/bundle"
 	"github.com/caarlos0/antibody/internal"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestBundleAndUpdate(t *testing.T) {
 	home := internal.TempHome()
 	defer os.RemoveAll(home)
-	a := antibody.New([]antibody.Bundle{
-		antibody.NewBundle("caarlos0/zsh-pg", home),
-		antibody.NewBundle("caarlos0/zsh-open-pr", home),
+	a := antibody.New([]bundle.Bundle{
+		bundle.New("caarlos0/zsh-pg", home),
+		bundle.New("caarlos0/zsh-open-pr", home),
 	})
 	a.Download()
 	a.Update()

diff --git a/bundle/bundle_test.go b/bundle/bundle_test.go
@@ -24,6 +24,14 @@ func TestSourceablesWithoutDownload(t *testing.T) {
 	assert.Empty(t, b.Sourceables())
 }
 
+func TestSourceablesDotSh(t *testing.T) {
+	home := internal.TempHome()
+	defer os.RemoveAll(home)
+	b := bundle.New("rupa/z", home)
+	b.Download()
+	assert.Len(t, b.Sourceables(), 1)
+}
+
 func TestListEmptyFolder(t *testing.T) {
 	home := internal.TempHome()
 	defer os.RemoveAll(home)