with garth

internal/auth/garth/session_test.go

@@ -0,0 +1,69 @@
+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")
+}