Add integration tests for BoldDB & fix implementation

data.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -104,7 +105,7 @@ func NewSource(alias string, URL *url.URL) (s *Source) {
 		Ext:   ext,
 	}
 
-	if ext != "" {
+	if ext != "" && URL.Scheme != "boltdb" {
 		mediatype := mime.TypeByExtension(ext)
 		t, params, err := mime.ParseMediaType(mediatype)
 		if err != nil {
@@ -346,7 +347,13 @@ func readLibKV(source *Source, args ...string) ([]byte, error) {
 	}
 
 	p := source.URL.Path
-	if len(args) == 1 {
+
+	if source.URL.Scheme == "boltdb" {
+		if len(args) != 1 {
+			return nil, errors.New("missing key")
+                }
+		p = args[0]
+	} else if len(args) == 1 {
 		p = p + "/" + args[0]
 	}
 

test/integration/Dockerfile

@@ -25,5 +25,6 @@ COPY mirror /bin/mirror
 COPY *.sh /tests/
 COPY *.bash /tests/
 COPY *.bats /tests/
+COPY *.db /test/integration/
 
 CMD ["/tests/test.sh"]

test/integration/datasources_boltdb.bats

@@ -0,0 +1,27 @@
+#!/usr/bin/env bats
+
+load helper
+
+tmpdir=$(mktemp -u)
+
+function setup () {
+  mkdir -p $tmpdir
+}
+
+function teardown () {
+  rm -rf $tmpdir || true
+}
+
+@test "supports BoltDB datasource file" {
+  cp test/integration/config.db $tmpdir/config.db
+  gomplate -d config=boltdb://$tmpdir/config.db#Bucket1 -i '{{(datasource "config" "foo")}}'
+  [ "$status" -eq 0 ]
+  [[ "${output}" == "bar" ]]
+}
+
+@test "supports multi-bucket BoltDB datasource file" {
+  cp test/integration/config.db $tmpdir/config.db
+  gomplate -d config=boltdb://$tmpdir/config.db#Bucket1 -d config2=boltdb://$tmpdir/config.db#Bucket2 -i '{{(datasource "config" "foo")}}-{{(datasource "config2" "foobar")}}'
+  [ "$status" -eq 0 ]
+  [[ "${output}" == "bar-baz" ]]
+}