itest: export HashmailHarness struct

Author: ViktorTigerstrom

itest/connection_test.go

@@ -45,14 +45,14 @@ func testHashmailServerReconnect(t *harnessTest) {
 	require.Equal(t.t, len(defaultMessage)*10, len(resp.Resp))
 
 	// Shut down hashmail server
-	require.NoError(t.t, t.hmserver.stop())
+	require.NoError(t.t, t.hmserver.Stop())
 
 	// Check that the client and server status are updated appropriately.
 	assertServerStatus(t, mailbox.ServerStatusNotConnected)
 	assertClientStatus(t, mailbox.ClientStatusNotConnected)
 
 	// Restart hashmail server
-	require.NoError(t.t, t.hmserver.start())
+	require.NoError(t.t, t.hmserver.Start())
 
 	// Check that the client and server successfully reconnect.
 	assertServerStatus(t, mailbox.ServerStatusInUse)

itest/hashmailserver_harness.go

@@ -12,14 +12,17 @@ import (
 	"github.com/lightningnetwork/lnd/lntest/wait"
 )
 
-type hashmailHarness struct {
-	aperture    *aperture.Aperture
-	apertureCfg *aperture.Config
+type HashmailHarness struct {
+	aperture *aperture.Aperture
+
+	// ApertureCfg is the configuration aperture uses when being initialized.
+	ApertureCfg *aperture.Config
 }
 
-func newHashmailHarness() *hashmailHarness {
-	return &hashmailHarness{
-		apertureCfg: &aperture.Config{
+// NewHashmailHarness creates a new instance of the HashmailHarness.
+func NewHashmailHarness() *HashmailHarness {
+	return &HashmailHarness{
+		ApertureCfg: &aperture.Config{
 			ListenAddr: fmt.Sprintf("127.0.0.1:%d",
 				node.NextAvailablePort()),
 			Authenticator: &aperture.AuthConfig{
@@ -39,8 +42,8 @@ func newHashmailHarness() *hashmailHarness {
 }
 
 // initAperture starts the aperture proxy.
-func (hm *hashmailHarness) initAperture() error {
-	hm.aperture = aperture.NewAperture(hm.apertureCfg)
+func (hm *HashmailHarness) initAperture() error {
+	hm.aperture = aperture.NewAperture(hm.ApertureCfg)
 	errChan := make(chan error)
 
 	if err := hm.aperture.Start(errChan); err != nil {
@@ -60,7 +63,7 @@ func (hm *hashmailHarness) initAperture() error {
 	}
 	return wait.NoError(func() error {
 		apertureAddr := fmt.Sprintf("https://%s/dummy",
-			hm.apertureCfg.ListenAddr)
+			hm.ApertureCfg.ListenAddr)
 
 		resp, err := http.Get(apertureAddr)
 		if err != nil {
@@ -75,14 +78,16 @@ func (hm *hashmailHarness) initAperture() error {
 	}, 3*time.Second)
 }
 
-func (hm *hashmailHarness) start() error {
+// Start attempts to start the aperture proxy.
+func (hm *HashmailHarness) Start() error {
 	if err := hm.initAperture(); err != nil {
 		return fmt.Errorf("could not start aperture: %v", err)
 	}
 
 	return nil
 }
 
-func (hm *hashmailHarness) stop() error {
+// Stop attempts to stop the aperture proxy.
+func (hm *HashmailHarness) Stop() error {
 	return hm.aperture.Stop()
 }

itest/test_harness.go

@@ -44,7 +44,7 @@ type harnessTest struct {
 
 	server *serverHarness
 
-	hmserver *hashmailHarness
+	hmserver *HashmailHarness
 }
 
 // testConfig determines the way in which the test will be set up.
@@ -61,11 +61,11 @@ func newHarnessTest(t *testing.T, cfg *testConfig) *harnessTest {
 	mailboxAddr := testnetMailbox
 	var insecure bool
 	if !cfg.stagingMailbox {
-		ht.hmserver = newHashmailHarness()
-		if err := ht.hmserver.start(); err != nil {
+		ht.hmserver = NewHashmailHarness()
+		if err := ht.hmserver.Start(); err != nil {
 			t.Fatalf("could not start hashmail server: %v", err)
 		}
-		mailboxAddr = ht.hmserver.apertureCfg.ListenAddr
+		mailboxAddr = ht.hmserver.ApertureCfg.ListenAddr
 		insecure = true
 	}
 
@@ -186,7 +186,7 @@ func (h *harnessTest) shutdown() error {
 	h.server.stop()
 
 	if h.hmserver != nil {
-		err := h.hmserver.stop()
+		err := h.hmserver.Stop()
 		if err != nil {
 			returnErr = err
 		}