add tests for new datastore configuration

License: MIT
Signed-off-by: Kevin Atkinson <k@kevina.org>

repo/fsrepo/config_test.go

@@ -0,0 +1,181 @@
+package fsrepo
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
+	"reflect"
+	"testing"
+
+	//"fmt"
+
+	syncmount "gx/ipfs/QmVSase1JP7cq9QkPT46oNwdp9pT6kBkG3oqS14y3QcZjG/go-datastore/syncmount"
+	//levelds "gx/ipfs/QmaHHmfEozrrotyhyN44omJouyuEtx6ahddqV6W5yRaUSQ/go-ds-leveldb"
+	config "github.com/ipfs/go-ipfs/repo/config"
+)
+
+var defaultConfig = []byte(`{
+    "StorageMax": "10GB",
+    "StorageGCWatermark": 90,
+    "GCPeriod": "1h",
+    "Spec": {
+      "mounts": [
+        {
+          "child": {
+            "path": "blocks",
+            "shardFunc": "/repo/flatfs/shard/v1/next-to-last/2",
+            "sync": true,
+            "type": "flatfs"
+          },
+          "mountpoint": "/blocks",
+          "prefix": "flatfs.datastore",
+          "type": "measure"
+        },
+        {
+          "child": {
+            "compression": "none",
+            "path": "datastore",
+            "type": "levelds"
+          },
+          "mountpoint": "/",
+          "prefix": "leveldb.datastore",
+          "type": "measure"
+        }
+      ],
+      "type": "mount"
+    },
+    "HashOnRead": false,
+    "BloomFilterSize": 0
+}`)
+
+var leveldbConfig = []byte(`{
+            "compression": "none",
+            "path": "datastore",
+            "type": "levelds"
+}`)
+
+var flatfsConfig = []byte(`{
+            "path": "blocks",
+            "shardFunc": "/repo/flatfs/shard/v1/next-to-last/2",
+            "sync": true,
+            "type": "flatfs"
+}`)
+
+var measureConfig = []byte(`{
+          "child": {
+            "path": "blocks",
+            "shardFunc": "/repo/flatfs/shard/v1/next-to-last/2",
+            "sync": true,
+            "type": "flatfs"
+          },
+          "mountpoint": "/blocks",
+          "prefix": "flatfs.datastore",
+          "type": "measure"
+}`)
+
+func TestDefaultDatastoreConfig(t *testing.T) {
+	dir, err := ioutil.TempDir("", "ipfs-datastore-config-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir) // clean up
+	repo := FSRepo{path: dir}
+
+	config := new(config.Datastore)
+	err = json.Unmarshal(defaultConfig, config)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ds, err := repo.constructDatastore(config.Spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	_, ok := ds.(*syncmount.Datastore)
+	if !ok {
+		t.Fatal("expected mount datastore at top level")
+	}
+	if typ := reflect.TypeOf(ds).String(); typ != "*syncmount.Datastore" {
+		t.Errorf("expected '*syncmount.Datastore' got '%s'", typ)
+	}
+}
+
+func TestLevelDbConfig(t *testing.T) {
+	config := new(config.Datastore)
+	err := json.Unmarshal(defaultConfig, config)
+	if err != nil {
+		t.Fatal(err)
+	}
+	dir, err := ioutil.TempDir("", "ipfs-datastore-config-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir) // clean up
+	repo := FSRepo{path: dir}
+
+	spec := make(map[string]interface{})
+	err = json.Unmarshal(leveldbConfig, &spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ds, err := repo.constructDatastore(spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if typ := reflect.TypeOf(ds).String(); typ != "*leveldb.datastore" {
+		t.Errorf("expected '*syncmount.Datastore' got '%s'", typ)
+	}
+}
+
+func TestFlatfsConfig(t *testing.T) {
+	config := new(config.Datastore)
+	err := json.Unmarshal(defaultConfig, config)
+	if err != nil {
+		t.Fatal(err)
+	}
+	dir, err := ioutil.TempDir("", "ipfs-datastore-config-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir) // clean up
+	repo := FSRepo{path: dir}
+
+	spec := make(map[string]interface{})
+	err = json.Unmarshal(flatfsConfig, &spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ds, err := repo.constructDatastore(spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if typ := reflect.TypeOf(ds).String(); typ != "*flatfs.Datastore" {
+		t.Errorf("expected '*syncmount.Datastore' got '%s'", typ)
+	}
+}
+
+func TestMeasureConfig(t *testing.T) {
+	config := new(config.Datastore)
+	err := json.Unmarshal(defaultConfig, config)
+	if err != nil {
+		t.Fatal(err)
+	}
+	dir, err := ioutil.TempDir("", "ipfs-datastore-config-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir) // clean up
+	repo := FSRepo{path: dir}
+
+	spec := make(map[string]interface{})
+	err = json.Unmarshal(measureConfig, &spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ds, err := repo.constructDatastore(spec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if typ := reflect.TypeOf(ds).String(); typ != "*measure.measure" {
+		t.Errorf("expected '*syncmount.Datastore' got '%s'", typ)
+	}
+}