func TestLoadUserPlusKeysRevoked(t *testing.T) {
tc := SetupEngineTest(t, "login")
defer tc.Cleanup()
fu := CreateAndSignupFakeUserPaper(tc, "login")
me, err := libkb.LoadMe(libkb.NewLoadUserArg(tc.G))
if err != nil {
t.Fatal(err)
}
up, err := libkb.LoadUserPlusKeys(tc.G, me.GetUID())
if err != nil {
t.Fatal(err)
}
if len(up.DeviceKeys) != 4 {
t.Errorf("device keys: %d, expected 4", len(up.DeviceKeys))
}
if len(up.RevokedDeviceKeys) != 0 {
t.Errorf("revoked keys: %d, expected 0", len(up.RevokedDeviceKeys))
}
devices, _ := getActiveDevicesAndKeys(tc, fu)
var paper *libkb.Device
for _, device := range devices {
if device.Type == libkb.DeviceTypePaper {
paper = device
break
}
}
if err := doRevokeDevice(tc, fu, paper.ID, false); err != nil {
t.Fatal(err)
}
up2, err := libkb.LoadUserPlusKeys(tc.G, me.GetUID())
if err != nil {
t.Fatal(err)
}
if len(up2.DeviceKeys) != 2 {
t.Errorf("device keys: %d, expected 2", len(up2.DeviceKeys))
}
if len(up2.RevokedDeviceKeys) != 2 {
t.Errorf("revoked keys: %d, expected 2", len(up2.RevokedDeviceKeys))
}
}
func TestLoadUserPlusKeysHasKeys(t *testing.T) {
tc := SetupEngineTest(t, "user")
defer tc.Cleanup()
CreateAndSignupFakeUser(tc, "login")
me, err := libkb.LoadMe(libkb.NewLoadUserArg(tc.G))
if err != nil {
t.Fatal(err)
}
up, err := libkb.LoadUserPlusKeys(tc.G, me.GetUID(), true)
if err != nil {
t.Fatal(err)
}
if len(up.DeviceKeys) != 4 {
t.Errorf("num device keys: %d, expected 4", len(up.DeviceKeys))
}
}
func (h *UserHandler) LoadUserPlusKeys(_ context.Context, arg keybase1.LoadUserPlusKeysArg) (keybase1.UserPlusKeys, error) {
return libkb.LoadUserPlusKeys(h.G(), arg.Uid, arg.CacheOK)
}