Write tests for new method

storage/ud/facts_test.go

@@ -16,6 +16,7 @@ import (
 	"testing"
 )
 
+// Smoke test.
 func TestNewStore(t *testing.T) {
 
 	kv := versioned.NewKV(make(ekv.Memstore))
@@ -27,6 +28,67 @@ func TestNewStore(t *testing.T) {
 
 }
 
+// Unit test
+func TestStore_RestoreFromBackUp(t *testing.T) {
+
+	kv := versioned.NewKV(make(ekv.Memstore))
+
+	s, err := NewStore(kv)
+	if err != nil {
+		t.Errorf("NewStore() produced an error: %v", err)
+	}
+
+	expected := fact.Fact{
+		Fact: "josh",
+		T:    fact.Username,
+	}
+
+	fl := fact.FactList{expected}
+
+	err = s.RestoreFromBackUp(fl)
+	if err != nil {
+		t.Fatalf("RestoreFromBackup err: %v", err)
+	}
+
+	_, exists := s.confirmedFacts[expected]
+	if !exists {
+		t.Fatalf("Fact %s does not exist in map", expected)
+	}
+
+}
+
+// Error case.
+func TestStore_RestoreFromBackUp_StatefulStore(t *testing.T) {
+
+	kv := versioned.NewKV(make(ekv.Memstore))
+
+	s, err := NewStore(kv)
+	if err != nil {
+		t.Errorf("NewStore() produced an error: %v", err)
+	}
+
+	confirmId := "confirm"
+	expected := fact.Fact{
+		Fact: "josh",
+		T:    fact.Username,
+	}
+
+	err = s.StoreUnconfirmedFact(confirmId, expected)
+	if err != nil {
+		t.Fatalf("StoreUnconfirmedFact error: %v", err)
+	}
+
+	// Expected error: should error when restoring on
+	// a stateful store.
+	fl := fact.FactList{expected}
+	err = s.RestoreFromBackUp(fl)
+	if err == nil {
+		t.Fatalf("RestoreFromBackup err: %v", err)
+	}
+
+}
+
+// Unit test.
 func TestStore_ConfirmFact(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))
 
@@ -93,6 +155,7 @@ func TestStore_StoreUnconfirmedFact(t *testing.T) {
 	}
 }
 
+// Unit test.
 func TestStore_DeleteFact(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))
 
@@ -125,6 +188,7 @@ func TestStore_DeleteFact(t *testing.T) {
 
 }
 
+// Unit test.
 func TestStore_BackUpMissingFacts(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))
 
@@ -160,6 +224,7 @@ func TestStore_BackUpMissingFacts(t *testing.T) {
 
 }
 
+// Error case.
 func TestStore_BackUpMissingFacts_DuplicateFactType(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))
 
@@ -197,6 +262,7 @@ func TestStore_BackUpMissingFacts_DuplicateFactType(t *testing.T) {
 
 }
 
+// Unit test.
 func TestStore_GetFacts(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))
 
@@ -246,6 +312,7 @@ func TestStore_GetFacts(t *testing.T) {
 	}
 }
 
+// Unit test.
 func TestStore_GetFactStrings(t *testing.T) {
 	kv := versioned.NewKV(make(ekv.Memstore))