go/engine/paperkey_submit_test.go - master - githubmirror/client

go/engine/paperkey_submit_test.go
87 строк2 KB
Joshua Blum
add revive to golangci-lint (#28303)
22 янв 2025, 20:51
22 янв 2025, 20:5122d0242
// Copyright 2016 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.

package engine

import (
	"testing"

	"github.com/keybase/client/go/libkb"
	"github.com/keybase/client/go/protocol/keybase1"
	"github.com/stretchr/testify/require"
)

func TestPaperKeySubmit(t *testing.T) {
	tc := SetupEngineTest(t, "login")
	defer tc.Cleanup()

	tc.G.SetService()
	listener := &nlistener{}
	tc.G.NotifyRouter.AddListener(listener)

	// signup and get the paper key
	fu := NewFakeUserOrBust(t, "paper")
	arg := MakeTestSignupEngineRunArg(fu)
	arg.SkipPaper = false
	loginUI := &paperLoginUI{Username: fu.Username}
	uis := libkb.UIs{
		LogUI:    tc.G.UI.GetLogUI(),
		GPGUI:    &gpgtestui{},
		SecretUI: fu.NewSecretUI(),
		LoginUI:  loginUI,
	}
	s := NewSignupEngine(tc.G, &arg)
	m := NewMetaContextForTest(tc).WithUIs(uis)
	if err := RunEngine2(m, s); err != nil {
		t.Fatal(err)
	}

	assertNumDevicesAndKeys(tc, fu, 2, 4)

	Logout(tc)

	paperkey := loginUI.PaperPhrase
	if len(paperkey) == 0 {
		t.Fatal("login ui has no paper key phrase")
	}

	fu.LoginOrBust(tc)

	assertPaperKeyCached(m, t, false)

	// submit the paper key
	m = NewMetaContextForTestWithLogUI(tc)
	eng := NewPaperKeySubmit(tc.G, paperkey)
	if err := RunEngine2(m, eng); err != nil {
		t.Fatal(err)
	}

	assertPaperKeyCached(m, t, true)

	if len(listener.paperEncKIDs) != 1 {
		t.Fatalf("num paperkey notifications: %d, expected 1", len(listener.paperEncKIDs))
	}
	if listener.paperEncKIDs[0].NotEqual(eng.deviceWithKeys.EncryptionKey().GetKID()) {
		t.Errorf("enc kid from notify: %s, expected %s", listener.paperEncKIDs[0], eng.deviceWithKeys.EncryptionKey().GetKID())
	}
}

func assertPaperKeyCached(m libkb.MetaContext, t *testing.T, wantCached bool) {
	device := m.ActiveDevice().ProvisioningKey(m)
	var isCached bool
	if device != nil {
		isCached = device.HasBothKeys()
	}
	require.Equal(t, isCached, wantCached)
}

type nlistener struct {
	libkb.NoopNotifyListener
	paperEncKIDs []keybase1.KID
}

var _ libkb.NotifyListener = (*nlistener)(nil)

func (n *nlistener) PaperKeyCached(_ keybase1.UID, encKID, _ keybase1.KID) {
	n.paperEncKIDs = append(n.paperEncKIDs, encKID)
}