From b978e3857dcfc8b8cf01aa0fb708349795816f9a Mon Sep 17 00:00:00 2001
From: Philip O'Toole <philip.otoole@yahoo.com>
Date: Sat, 22 Oct 2022 13:28:33 -0400
Subject: [PATCH] More HTTP-level testing of Load

---
 http/service_test.go  |  83 ++++++++++++++++++++++++++++++++++++++++--
 http/testdata/load.db | Bin 0 -> 8192 bytes
 2 files changed, 79 insertions(+), 4 deletions(-)
 create mode 100644 http/testdata/load.db

diff --git a/http/service_test.go b/http/service_test.go
index 9f7a5869..0cade6d2 100644
--- a/http/service_test.go
+++ b/http/service_test.go
@@ -1,6 +1,7 @@
 package http
 
 import (
+	"bytes"
 	"crypto/tls"
 	"encoding/json"
 	"fmt"
@@ -8,6 +9,7 @@ import (
 	"io/ioutil"
 	"net/http"
 	"net/url"
+	"os"
 	"strings"
 	"testing"
 	"time"
@@ -687,6 +689,75 @@ func Test_BackupFlagsNoLeaderOK(t *testing.T) {
 	}
 }
 
+func Test_LoadOK(t *testing.T) {
+	m := &MockStore{}
+	c := &mockClusterService{}
+	s := New("127.0.0.1:0", m, c, nil)
+	if err := s.Start(); err != nil {
+		t.Fatalf("failed to start service")
+	}
+	defer s.Close()
+
+	client := &http.Client{}
+	host := fmt.Sprintf("http://%s", s.Addr().String())
+	resp, err := client.Post(host+"/db/load", "application/octet-stream", strings.NewReader("SELECT"))
+	if err != nil {
+		t.Fatalf("failed to make load request")
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		t.Fatalf("failed to get expected StatusOK for load, got %d", resp.StatusCode)
+	}
+}
+
+func Test_LoadFlagsNoLeader(t *testing.T) {
+	m := &MockStore{
+		leaderAddr: "foo:1234",
+	}
+	c := &mockClusterService{
+		apiAddr: "http://1.2.3.4:999",
+	}
+
+	s := New("127.0.0.1:0", m, c, nil)
+
+	if err := s.Start(); err != nil {
+		t.Fatalf("failed to start service")
+	}
+	defer s.Close()
+
+	testData, err := os.ReadFile("testdata/load.db")
+	if err != nil {
+		t.Fatalf("failed to load test SQLite data")
+	}
+
+	m.loadFn = func(br *command.LoadRequest) error {
+		return store.ErrNotLeader
+	}
+
+	clusterLoadCalled := false
+	c.loadFn = func(lr *command.LoadRequest, nodeAddr string, timeout time.Duration) error {
+		clusterLoadCalled = true
+		if bytes.Compare(lr.Data, testData) != 0 {
+			t.Fatalf("wrong data passed to cluster load")
+		}
+		return nil
+	}
+
+	client := &http.Client{}
+	host := fmt.Sprintf("http://%s", s.Addr().String())
+	resp, err := client.Post(host+"/db/load", "application/octet-stream", bytes.NewReader(testData))
+	if err != nil {
+		t.Fatalf("failed to make load request")
+	}
+	if resp.StatusCode != http.StatusOK {
+		t.Fatalf("failed to get expected StatusOK for load, got %d", resp.StatusCode)
+	}
+
+	if !clusterLoadCalled {
+		t.Fatalf("cluster load was not called")
+	}
+}
+
 func Test_RegisterStatus(t *testing.T) {
 	var stats *mockStatusReporter
 	m := &MockStore{}
@@ -1046,6 +1117,7 @@ type MockStore struct {
 	executeFn  func(er *command.ExecuteRequest) ([]*command.ExecuteResult, error)
 	queryFn    func(qr *command.QueryRequest) ([]*command.QueryRows, error)
 	backupFn   func(br *command.BackupRequest, dst io.Writer) error
+	loadFn     func(lr *command.LoadRequest) error
 	leaderAddr string
 }
 
@@ -1063,10 +1135,6 @@ func (m *MockStore) Query(qr *command.QueryRequest) ([]*command.QueryRows, error
 	return nil, nil
 }
 
-func (m *MockStore) Load(lr *command.LoadRequest) error {
-	return nil
-}
-
 func (m *MockStore) Join(id, addr string, voter bool) error {
 	return nil
 }
@@ -1098,6 +1166,13 @@ func (m *MockStore) Backup(br *command.BackupRequest, w io.Writer) error {
 	return m.backupFn(br, w)
 }
 
+func (m *MockStore) Load(lr *command.LoadRequest) error {
+	if m.loadFn != nil {
+		return m.loadFn(lr)
+	}
+	return nil
+}
+
 type mockClusterService struct {
 	apiAddr   string
 	executeFn func(er *command.ExecuteRequest, addr string, t time.Duration) ([]*command.ExecuteResult, error)
diff --git a/http/testdata/load.db b/http/testdata/load.db
new file mode 100644
index 0000000000000000000000000000000000000000..416ba9961ec3b181db8f50cd291ad67c48a149ff
GIT binary patch
literal 8192
zcmeI#!3u&v5C-5`3+xgG9=y)6AiDGbWwAtLX0|~djTD4*A$);esQ2kbI=5DL&%r;;
z?(D+Az;~O&J}BypbA2n*nx`}%A!yD_MAq+^@(E$??(f}{=HC_;O=i!&=@QO^Od=ov
z0SG_<0uX=z1Rwwb2tWV=5NH*MgzdSmxZ8DlzGk`BCaq;GeI;4>i$I!1Uh}ZH@H$j-
zC1VbEio;_Na1^h%ew^?|CS$JBGGirAYV_UBU*x=6hob=k5P$##AOHafKmY;|fB*y_
Q0D->}=!p(_d7-QH4Yy1qssI20

literal 0
HcmV?d00001