Files
proton-bridge/internal/kb/suggester_test.go
2023-12-07 09:35:05 +01:00

58 lines
2.1 KiB
Go

// Copyright (c) 2023 Proton AG
//
// This file is part of Proton Mail Bridge.
//
// Proton Mail Bridge is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Proton Mail Bridge is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Proton Mail Bridge. If not, see <https://www.gnu.org/licenses/>.
package kb
import (
"fmt"
"testing"
"github.com/stretchr/testify/require"
)
func Test_ArticleList(t *testing.T) {
articles, err := GetArticleList()
require.NoError(t, err)
require.NotEmpty(t, articles)
var bits uint64
for _, article := range articles {
require.Truef(t, article.Index < 64, "Invalid KB article index %d, (must be < 64)", article.Index)
require.Zerof(t, bits&(1<<article.Index), "Duplicate index %d in knowledge base", article.Index)
bits |= bits | (1 << article.Index)
require.NotEmpty(t, article.URL, "KB article with index %d has no URL", article.Index)
require.NotEmpty(t, article.Title, "KB article with index %d has no title", article.Index)
require.NotEmpty(t, article.Keywords, "KB article with index %d has no keyword", article.Index)
}
}
func Test_GetSuggestions(t *testing.T) {
suggestions, err := GetSuggestions("Thunderbird is not working, error during password")
require.NoError(t, err)
require.True(t, len(suggestions) <= 3)
for _, article := range suggestions {
fmt.Printf("Score: %v - %#v\n", article.Score, article.Title)
}
suggestions, err = GetSuggestions("Supercalifragilisticexpialidocious Sesquipedalian Worcestershire")
require.NoError(t, err)
require.Empty(t, suggestions)
}
func Test_simplifyUserInput(t *testing.T) {
require.Equal(t, "word1 ñóÄ don't déjà 33 pizza", simplifyUserInput(" \nword1 \n\tñóÄ don't\n\n\ndéjà, 33 pizza=🍕\n,\n"))
}