Change establish session to allow for several commands (#20)

Fixes #6

Author: ogenstad

pkg/netrasp/asa.go

@@ -11,5 +11,5 @@ type asa struct {
 
 // Dial opens a connection to a device.
 func (a asa) Dial(ctx context.Context) error {
-	return establishConnection(ctx, a, a.Connection, a.basePrompt(), "no terminal pager")
+	return establishConnection(ctx, a, a.Connection, a.basePrompt(), []string{"no terminal pager"})
 }

pkg/netrasp/helpers.go

@@ -6,7 +6,7 @@ import (
 	"regexp"
 )
 
-func establishConnection(ctx context.Context, p Platform, c connection, prompt *regexp.Regexp, disablePager string) error {
+func establishConnection(ctx context.Context, p Platform, c connection, prompt *regexp.Regexp, preparationCommands []string) error {
 	err := c.Dial(ctx)
 	if err != nil {
 		return fmt.Errorf("unable to open connection: %w", err)
@@ -19,9 +19,11 @@ func establishConnection(ctx context.Context, p Platform, c connection, prompt *
 		return fmt.Errorf("unable to find the initial prompt: %w", err)
 	}
 
-	_, err = p.Run(ctx, disablePager)
-	if err != nil {
-		return fmt.Errorf("unable to disable terminal paging: %w", err)
+	for _, command := range preparationCommands {
+		output, err := p.Run(ctx, command)
+		if err != nil {
+			return fmt.Errorf("unable to prepare session using the command '%s': %w", output, err)
+		}
 	}
 
 	return nil

pkg/netrasp/ios.go

@@ -46,7 +46,9 @@ func (i ios) Configure(ctx context.Context, commands []string) (string, error) {
 
 // Dial opens a connection to a device.
 func (i ios) Dial(ctx context.Context) error {
-	return establishConnection(ctx, i, i.Connection, i.basePrompt(), "terminal length 0")
+	commands := []string{"terminal length 0", "terminal width 511"}
+
+	return establishConnection(ctx, i, i.Connection, i.basePrompt(), commands)
 }
 
 // Enable elevates privileges.

pkg/netrasp/sros.go

@@ -48,7 +48,7 @@ func (s sros) Configure(ctx context.Context, commands []string) (string, error)
 
 // Dial opens a connection to a device.
 func (s sros) Dial(ctx context.Context) error {
-	return establishConnection(ctx, s, s.Connection, s.basePrompt(), "environment more false")
+	return establishConnection(ctx, s, s.Connection, s.basePrompt(), []string{"environment more false"})
 }
 
 // Enable elevates privileges.

pkg/netrasp/testdata/ios/basic/terminal_width_511.txt

@@ -0,0 +1 @@
+router1#

pkg/netrasp/testdata/ios/enable_with_config/terminal_width_511.txt

@@ -0,0 +1 @@
+router1#

pkg/netrasp/testdata/nxos/basic/terminal_width_511.txt

@@ -0,0 +1 @@
+sr#