mirror of
https://github.com/sstent/go-garminconnect.git
synced 2025-12-06 08:02:02 +00:00
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package garth
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSessionPersistence(t *testing.T) {
|
|
// Setup temporary file
|
|
tmpDir := os.TempDir()
|
|
sessionFile := filepath.Join(tmpDir, "test_session.json")
|
|
defer os.Remove(sessionFile)
|
|
|
|
// Create test session
|
|
testSession := &Session{
|
|
OAuth1Token: "test_oauth1_token",
|
|
OAuth1Secret: "test_oauth1_secret",
|
|
OAuth2Token: "test_oauth2_token",
|
|
}
|
|
|
|
// Test saving
|
|
err := testSession.Save(sessionFile)
|
|
assert.NoError(t, err, "Saving session should not produce error")
|
|
|
|
// Test loading
|
|
loadedSession, err := LoadSession(sessionFile)
|
|
assert.NoError(t, err, "Loading session should not produce error")
|
|
assert.Equal(t, testSession, loadedSession, "Loaded session should match saved session")
|
|
|
|
// Test loading non-existent file
|
|
_, err = LoadSession("non_existent_file.json")
|
|
assert.Error(t, err, "Loading non-existent file should return error")
|
|
}
|
|
|
|
func TestSessionContextHandling(t *testing.T) {
|
|
// Create authenticator with session path
|
|
tmpDir := os.TempDir()
|
|
sessionFile := filepath.Join(tmpDir, "context_session.json")
|
|
defer os.Remove(sessionFile)
|
|
|
|
auth := NewAuthenticator("https://example.com", sessionFile)
|
|
|
|
// Verify empty session returns error
|
|
_, err := auth.Login("user", "pass")
|
|
assert.Error(t, err, "Should return error when no active session")
|
|
}
|
|
|
|
func TestMFAPrompterInterface(t *testing.T) {
|
|
// Test console prompter implements interface
|
|
var prompter MFAPrompter = DefaultConsolePrompter{}
|
|
_, err := prompter.GetMFACode(context.Background())
|
|
assert.NoError(t, err, "Default prompter should not produce errors")
|
|
|
|
// Test mock prompter
|
|
mock := &MockMFAPrompter{Code: "123456", Err: nil}
|
|
code, err := mock.GetMFACode(context.Background())
|
|
assert.Equal(t, "123456", code, "Mock prompter should return provided code")
|
|
assert.NoError(t, err, "Mock prompter should not return error when Err is nil")
|
|
|
|
// Test error case
|
|
errorMock := &MockMFAPrompter{Err: errors.New("prompt error")}
|
|
_, err = errorMock.GetMFACode(context.Background())
|
|
assert.Error(t, err, "Mock prompter should return error when set")
|
|
}
|