From 7e389d551cca47012d91e72025fcb66bad86eb4c Mon Sep 17 00:00:00 2001
From: Hector Sanjuan <code@hector.link>
Date: Tue, 1 Feb 2022 22:29:54 +0100
Subject: [PATCH] Add test about local allocation

---
 add_test.go | 46 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/add_test.go b/add_test.go
index eb3bcd92..133328a3 100644
--- a/add_test.go
+++ b/add_test.go
@@ -4,6 +4,7 @@ package ipfscluster
 
 import (
 	"context"
+	"fmt"
 	"mime/multipart"
 	"sync"
 	"testing"
@@ -25,7 +26,7 @@ func TestAdd(t *testing.T) {
 
 	waitForLeaderAndMetrics(t, clusters)
 
-	t.Run("local", func(t *testing.T) {
+	t.Run("default", func(t *testing.T) {
 		params := api.DefaultAddParams()
 		params.Shard = false
 		params.Name = "testlocal"
@@ -56,6 +57,49 @@ func TestAdd(t *testing.T) {
 
 		runF(t, clusters, f)
 	})
+
+	t.Run("local_one_allocation", func(t *testing.T) {
+		params := api.DefaultAddParams()
+		params.Shard = false
+		params.Name = "testlocal"
+		params.ReplicationFactorMin = 1
+		params.ReplicationFactorMax = 1
+		params.Local = true
+		mfr, closer := sth.GetTreeMultiReader(t)
+		defer closer.Close()
+		r := multipart.NewReader(mfr, mfr.Boundary())
+		ci, err := clusters[2].AddFile(r, params)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if ci.String() != test.ShardingDirBalancedRootCID {
+			t.Fatal("unexpected root CID for local add")
+		}
+
+		// We need to sleep a lot because it takes time to
+		// catch up on a first/single pin on crdts
+		time.Sleep(10 * time.Second)
+
+		f := func(t *testing.T, c *Cluster) {
+			pin := c.StatusLocal(ctx, ci)
+			if pin.Error != "" {
+				t.Error(pin.Error)
+			}
+			fmt.Println(pin)
+			switch c.id {
+			case clusters[2].id:
+				if pin.Status != api.TrackerStatusPinned {
+					t.Error("item should be pinned and is", pin.Status)
+				}
+			default:
+				if pin.Status != api.TrackerStatusRemote {
+					t.Errorf("item should only be allocated to cluster2")
+				}
+			}
+		}
+
+		runF(t, clusters, f)
+	})
 }
 
 func TestAddWithUserAllocations(t *testing.T) {