Function TestOTPClearOnTokenKeyChange

core/otp_model_test.go:304–363

(t *testing.T)

}

func TestOTPClearOnTokenKeyChange(t *testing.T) {
	t.Parallel()

	app, _ := tests.NewTestApp()
	defer app.Cleanup()

	user1, err := app.FindAuthRecordByEmail("users", "test@example.com")
	if err != nil {
		t.Fatal(err)
	}

	user2, err := app.FindAuthRecordByEmail("users", "test2@example.com")
	if err != nil {
		t.Fatal(err)
	}

	otpsToCreate := map[*core.Record]int{
		user1: 3,
		user2: 2,
	}
	for user, total := range otpsToCreate {
		for range total {
			otp := core.NewOTP(app)
			otp.SetCollectionRef(user.Collection().Id)
			otp.SetRecordRef(user.Id)
			otp.SetPassword("123456")
			if err := app.Save(otp); err != nil {
				t.Fatal(err)
			}
		}
	}

	// update both users
	err = app.Save(user1)
	if err != nil {
		t.Fatal(err)
	}

	user2.RefreshTokenKey()
	err = app.Save(user2)
	if err != nil {
		t.Fatal(err)
	}

	expectedOTPs := map[*core.Record]int{
		user1: 3,
		user2: 0,
	}

	for user, expected := range expectedOTPs {
		otps, err := app.FindAllOTPsByRecord(user)
		if err != nil {
			t.Fatal(err)
		}

		if len(otps) != expected {
			t.Fatalf("Expected %d OTPs, got %d", expected, len(otps))
		}