package convert

import (
	"bytes"
	"compress/gzip"
	"fmt"
	"os"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	"github.com/pyroscope-io/pyroscope/pkg/agent/spy"
)

var _ = Describe("convert", func() {
	Describe("ParsePprof", func() {
		It("parses data correctly", func() {
			result := []string{}

			b, err := os.ReadFile("testdata/cpu.pprof")
			Expect(err).ToNot(HaveOccurred())
			r := bytes.NewReader(b)
			g, err := gzip.NewReader(r)
			Expect(err).ToNot(HaveOccurred())
			p, err := ParsePprof(g)
			Expect(err).ToNot(HaveOccurred())

			p.Get("samples", func(labels *spy.Labels, name []byte, val int) error {
				result = append(result, fmt.Sprintf("%s %d", name, val))
				return nil
			})
			Expect(result).To(ContainElement("runtime.main;main.work 1"))
		})
	})

	Describe("ParseGroups", func() {
		It("parses data correctly", func() {
			r := bytes.NewReader([]byte("foo;bar 10\nfoo;baz 20\n"))
			result := []string{}
			ParseGroups(r, func(name []byte, val int) {
				result = append(result, fmt.Sprintf("%s %d", name, val))
			})
			Expect(result).To(ConsistOf("foo;bar 10", "foo;baz 20"))
		})
	})

	Describe("ParseIndividualLines", func() {
		It("parses data correctly", func() {
			r := bytes.NewReader([]byte("foo;bar\nfoo;baz\n"))
			result := []string{}
			ParseIndividualLines(r, func(name []byte, val int) {
				result = append(result, fmt.Sprintf("%s %d", name, val))
			})
			Expect(result).To(ConsistOf("foo;bar 1", "foo;baz 1"))
		})
	})
})