MyGoNavi/internal/app/methods_jvm_monitoring_test.go

package app

import (
	"context"
	"errors"
	"strings"
	"testing"

	"GoNavi-Wails/internal/connection"
	"GoNavi-Wails/internal/jvm"
)

type fakeJVMMonitoringManager struct {
	startSnapshot     jvm.MonitoringSessionSnapshot
	startErr          error
	historySnapshot   jvm.MonitoringSessionSnapshot
	historyErr        error
	stopErr           error
	startCfg          connection.ConnectionConfig
	startMode         string
	historyConnection string
	historyMode       string
	stopConnection    string
	stopMode          string
}

func (f *fakeJVMMonitoringManager) Start(_ context.Context, cfg connection.ConnectionConfig, mode string) (jvm.MonitoringSessionSnapshot, error) {
	f.startCfg = cfg
	f.startMode = mode
	return f.startSnapshot, f.startErr
}

func (f *fakeJVMMonitoringManager) GetHistory(connectionID string, providerMode string) (jvm.MonitoringSessionSnapshot, error) {
	f.historyConnection = connectionID
	f.historyMode = providerMode
	return f.historySnapshot, f.historyErr
}

func (f *fakeJVMMonitoringManager) Stop(connectionID string, providerMode string) error {
	f.stopConnection = connectionID
	f.stopMode = providerMode
	return f.stopErr
}

func swapJVMMonitoringManager(manager jvmMonitoringService) func() {
	prev := currentJVMMonitoringManager
	currentJVMMonitoringManager = manager
	return func() { currentJVMMonitoringManager = prev }
}

func TestJVMStartMonitoringReturnsManagerSnapshot(t *testing.T) {
	app := NewAppWithSecretStore(nil)
	manager := &fakeJVMMonitoringManager{
		startSnapshot: jvm.MonitoringSessionSnapshot{
			ConnectionID: "conn-monitor",
			ProviderMode: jvm.ModeEndpoint,
			Running:      true,
			Points: []jvm.JVMMonitoringPoint{
				{Timestamp: 1713945600000, ThreadCount: 21},
			},
		},
	}
	restore := swapJVMMonitoringManager(manager)
	defer restore()

	res := app.JVMStartMonitoring(connection.ConnectionConfig{
		ID:   "conn-monitor",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeEndpoint,
			AllowedModes:  []string{jvm.ModeEndpoint},
		},
	})

	if !res.Success {
		t.Fatalf("expected success, got %+v", res)
	}
	snapshot, ok := res.Data.(jvm.MonitoringSessionSnapshot)
	if !ok {
		t.Fatalf("expected monitoring snapshot, got %#v", res.Data)
	}
	if !snapshot.Running || len(snapshot.Points) != 1 {
		t.Fatalf("unexpected snapshot: %#v", snapshot)
	}
	if manager.startCfg.ID != "conn-monitor" {
		t.Fatalf("expected manager to receive config ID, got %#v", manager.startCfg)
	}
}

func TestJVMGetMonitoringHistoryResolvesPreferredMode(t *testing.T) {
	app := NewAppWithSecretStore(nil)
	manager := &fakeJVMMonitoringManager{
		historySnapshot: jvm.MonitoringSessionSnapshot{
			ConnectionID: "conn-history",
			ProviderMode: jvm.ModeJMX,
			Running:      true,
		},
	}
	restore := swapJVMMonitoringManager(manager)
	defer restore()

	res := app.JVMGetMonitoringHistory(connection.ConnectionConfig{
		ID:   "conn-history",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeJMX,
			AllowedModes:  []string{jvm.ModeJMX},
		},
	}, "")

	if !res.Success {
		t.Fatalf("expected success, got %+v", res)
	}
	if manager.historyConnection != "conn-history" || manager.historyMode != jvm.ModeJMX {
		t.Fatalf("unexpected manager history args: connection=%q mode=%q", manager.historyConnection, manager.historyMode)
	}
}

func TestJVMStopMonitoringReturnsManagerError(t *testing.T) {
	app := NewAppWithSecretStore(nil)
	manager := &fakeJVMMonitoringManager{
		stopErr: errors.New("session not found"),
	}
	restore := swapJVMMonitoringManager(manager)
	defer restore()

	res := app.JVMStopMonitoring(connection.ConnectionConfig{
		ID:   "conn-stop",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeAgent,
			AllowedModes:  []string{jvm.ModeAgent},
		},
	}, "")

	if res.Success {
		t.Fatalf("expected failure, got %+v", res)
	}
	if res.Message != "session not found" {
		t.Fatalf("expected message %q, got %#v", "session not found", res)
	}
	if manager.stopConnection != "conn-stop" || manager.stopMode != jvm.ModeAgent {
		t.Fatalf("unexpected manager stop args: connection=%q mode=%q", manager.stopConnection, manager.stopMode)
	}
}

func TestJVMMonitoringMethodsLocalizeManagerLocalizedErrors(t *testing.T) {
	app := NewAppWithSecretStore(nil)
	app.SetLanguage("en-US")
	manager := &fakeJVMMonitoringManager{
		startErr: &jvm.LocalizedError{
			Key: "jvm.backend.monitoring.error.snapshot_unsupported",
			Params: map[string]any{
				"provider": "JMX",
			},
		},
		historyErr: &jvm.LocalizedError{
			Key: "jvm.backend.monitoring.error.session_not_found",
			Params: map[string]any{
				"connectionId": "conn-history",
				"providerMode": jvm.ModeJMX,
			},
		},
		stopErr: &jvm.LocalizedError{
			Key: "jvm.backend.monitoring.error.session_not_found",
			Params: map[string]any{
				"connectionId": "conn-stop",
				"providerMode": jvm.ModeAgent,
			},
		},
	}
	restore := swapJVMMonitoringManager(manager)
	defer restore()

	startRes := app.JVMStartMonitoring(connection.ConnectionConfig{
		ID:   "conn-monitor",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeJMX,
			AllowedModes:  []string{jvm.ModeJMX},
		},
	})
	assertMonitoringEnglishMessage(t, startRes, "JMX monitoring snapshot is not supported yet")

	historyRes := app.JVMGetMonitoringHistory(connection.ConnectionConfig{
		ID:   "conn-history",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeJMX,
			AllowedModes:  []string{jvm.ModeJMX},
		},
	}, "")
	assertMonitoringEnglishMessage(t, historyRes, "Monitoring session not found for conn-history jmx")

	stopRes := app.JVMStopMonitoring(connection.ConnectionConfig{
		ID:   "conn-stop",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeAgent,
			AllowedModes:  []string{jvm.ModeAgent},
		},
	}, "")
	assertMonitoringEnglishMessage(t, stopRes, "Monitoring session not found for conn-stop agent")
}

func TestJVMMonitoringMethodsLocalizeStructuredProviderWarnings(t *testing.T) {
	app := NewAppWithSecretStore(nil)
	app.SetLanguage("en-US")
	manager := &fakeJVMMonitoringManager{
		startSnapshot: jvm.MonitoringSessionSnapshot{
			ConnectionID: "conn-monitor",
			ProviderMode: jvm.ModeJMX,
			Running:      false,
			ProviderWarnings: []string{
				"endpoint cpu metric unavailable",
				"__gonavi_i18n__:jvm.backend.monitoring.warning.sample_auto_stopped:count=3",
			},
		},
		historySnapshot: jvm.MonitoringSessionSnapshot{
			ConnectionID: "conn-monitor",
			ProviderMode: jvm.ModeJMX,
			Running:      false,
			ProviderWarnings: []string{
				"__gonavi_i18n__:jvm.backend.monitoring.warning.sample_auto_stopped:count=3",
				"collector returned HTTP 503",
			},
		},
	}
	restore := swapJVMMonitoringManager(manager)
	defer restore()

	startRes := app.JVMStartMonitoring(connection.ConnectionConfig{
		ID:   "conn-monitor",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeJMX,
			AllowedModes:  []string{jvm.ModeJMX},
		},
	})
	startSnapshot := assertMonitoringSnapshot(t, startRes)
	assertMonitoringWarnings(t, startSnapshot.ProviderWarnings, []string{
		"endpoint cpu metric unavailable",
		"Monitoring sampling failed 3 consecutive times and this session was stopped automatically",
	})

	historyRes := app.JVMGetMonitoringHistory(connection.ConnectionConfig{
		ID:   "conn-monitor",
		Type: "jvm",
		Host: "orders.internal",
		JVM: connection.JVMConfig{
			PreferredMode: jvm.ModeJMX,
			AllowedModes:  []string{jvm.ModeJMX},
		},
	}, "")
	historySnapshot := assertMonitoringSnapshot(t, historyRes)
	assertMonitoringWarnings(t, historySnapshot.ProviderWarnings, []string{
		"Monitoring sampling failed 3 consecutive times and this session was stopped automatically",
		"collector returned HTTP 503",
	})
}

func assertMonitoringEnglishMessage(t *testing.T, res connection.QueryResult, want string) {
	t.Helper()

	if res.Success {
		t.Fatalf("expected monitoring method to fail, got %+v", res)
	}
	if res.Message != want {
		t.Fatalf("expected monitoring message %q, got %+v", want, res)
	}
	if strings.Contains(res.Message, "jvm.backend.") {
		t.Fatalf("expected localized message instead of raw key, got %q", res.Message)
	}
}

func assertMonitoringSnapshot(t *testing.T, res connection.QueryResult) jvm.MonitoringSessionSnapshot {
	t.Helper()

	if !res.Success {
		t.Fatalf("expected monitoring method to succeed, got %+v", res)
	}
	snapshot, ok := res.Data.(jvm.MonitoringSessionSnapshot)
	if !ok {
		t.Fatalf("expected monitoring snapshot, got %#v", res.Data)
	}
	return snapshot
}

func assertMonitoringWarnings(t *testing.T, got []string, want []string) {
	t.Helper()

	if len(got) != len(want) {
		t.Fatalf("expected warnings %#v, got %#v", want, got)
	}
	for index, expected := range want {
		if got[index] != expected {
			t.Fatalf("expected warning %d to be %q, got %#v", index, expected, got)
		}
		if strings.Contains(got[index], "jvm.backend.") {
			t.Fatalf("expected localized warning instead of raw key, got %#v", got)
		}
	}
}