Function TestExtractExperimentData

pkg/cli/audit_report_experiments_test.go:47–123

(t *testing.T)

}

func TestExtractExperimentData(t *testing.T) {
	t.Run("returns nil for empty logsPath", func(t *testing.T) {
		assert.Nil(t, extractExperimentData(""), "should return nil for empty logsPath")
	})

	t.Run("returns nil when no state.json present", func(t *testing.T) {
		dir := t.TempDir()
		assert.Nil(t, extractExperimentData(dir), "should return nil when state.json missing")
	})

	t.Run("returns nil for invalid JSON", func(t *testing.T) {
		dir := t.TempDir()
		require.NoError(t, os.WriteFile(filepath.Join(dir, "state.json"), []byte("not-json"), 0o600))
		assert.Nil(t, extractExperimentData(dir), "should return nil for invalid JSON")
	})

	t.Run("returns nil for empty counts", func(t *testing.T) {
		dir := t.TempDir()
		require.NoError(t, os.WriteFile(filepath.Join(dir, "state.json"), []byte(`{"counts":{}}`), 0o600))
		assert.Nil(t, extractExperimentData(dir), "should return nil when counts map is empty")
	})

	t.Run("extracts single experiment with two variants", func(t *testing.T) {
		dir := t.TempDir()
		state := map[string]any{
			"counts": map[string]any{
				"caveman": map[string]int{"yes": 3, "no": 2},
			},
		}
		raw, err := json.Marshal(state)
		require.NoError(t, err)
		require.NoError(t, os.WriteFile(filepath.Join(dir, "state.json"), raw, 0o600))

		got := extractExperimentData(dir)
		require.NotNil(t, got, "should return non-nil ExperimentData")
		assert.Equal(t, "yes", got.Assignments["caveman"], "variant with highest count should be selected")
		assert.Equal(t, 3, got.CumulativeCounts["caveman"]["yes"], "cumulative count for yes should be 3")
		assert.Equal(t, 2, got.CumulativeCounts["caveman"]["no"], "cumulative count for no should be 2")
	})

	t.Run("reads state.json from experiment subdirectory", func(t *testing.T) {
		dir := t.TempDir()
		subDir := filepath.Join(dir, "experiment")
		require.NoError(t, os.MkdirAll(subDir, 0o755))
		state := map[string]any{
			"counts": map[string]any{
				"style": map[string]int{"concise": 1, "detailed": 2},
			},
		}
		raw, err := json.Marshal(state)
		require.NoError(t, err)
		require.NoError(t, os.WriteFile(filepath.Join(subDir, "state.json"), raw, 0o600))

		got := extractExperimentData(dir)
		require.NotNil(t, got, "should return non-nil ExperimentData from subdir")
		assert.Equal(t, "detailed", got.Assignments["style"], "detailed has higher count so should be selected")
	})