Function newTestJupyterClient

kernel_test.go:336–373

newTestJupyterClient creates and connects a fresh client to the kernel. Upon error, newTestJupyterClient will Fail the test.

(t *testing.T)

// newTestJupyterClient creates and connects a fresh client to the kernel. Upon error, newTestJupyterClient
// will Fail the test.
func newTestJupyterClient(t *testing.T) (testJupyterClient, func()) {
	t.Helper()

	var (
		err       error
		ctx       = context.Background()
		addrShell = fmt.Sprintf("%s://%s:%d", transport, ip, shellPort)
		addrIO    = fmt.Sprintf("%s://%s:%d", transport, ip, iopubPort)
	)

	// Prepare the shell socket.
	shell := zmq4.NewReq(ctx)
	if err = shell.Dial(addrShell); err != nil {
		t.Fatalf("\t%s shell.Connect: %s", failure, err)
	}

	// Prepare the IOPub socket.
	iopub := zmq4.NewSub(ctx)
	if err = iopub.Dial(addrIO); err != nil {
		t.Fatalf("\t%s iopub.Connect: %s", failure, err)
	}

	if err = iopub.SetOption(zmq4.OptionSubscribe, ""); err != nil {
		t.Fatalf("\t%s iopub.SetSubscribe: %s", failure, err)
	}

	// Wait for a second to give the tcp connection time to complete to avoid missing the early pub messages.
	time.Sleep(1 * time.Second)

	return testJupyterClient{shell, iopub}, func() {
		if err := shell.Close(); err != nil {
			t.Errorf("\t%s shell.Close: %s", failure, err)
		}
		if err = iopub.Close(); err != nil {
			t.Errorf("\t%s iopub.Close: %s", failure, err)
		}
	}
}

// sendShellRequest sends a message to the kernel over the shell channel. Upon error, sendShellRequest
// will Fail the test.