package parser

import (
	"bytes"
	"encoding/json"
	"fmt"
	"testing"
	"testing/iotest"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.skia.org/infra/perf/go/ingest/format"
)

const txLogName = "gs://some-example-bucket-name/foo/bar/file.json"

func TestParse_Incoming_Success(t *testing.T) {
	r := bytes.NewBufferString(incoming)
	in, err := Parse(r)
	assert.NoError(t, err)
	assert.Equal(t, "google-marlin-marlin-O", in.Branch)
	assert.Len(t, in.Metrics, 7)
	f, err := in.Metrics["android.platform.systemui.tests.jank.LauncherJankTests#testAppSwitchGMailtoHome"]["frame-avg-jank"].Float64()
	assert.NoError(t, err)
	assert.Equal(t, 8.4, f)
	assert.Equal(t, "coral", in.DeviceName)
	assert.Equal(t, "API_29_R", in.SDKReleaseName)
	assert.Equal(t, "disabled", in.JIT)
}

func TestParse_Incoming2_Success(t *testing.T) {
	r := bytes.NewBufferString(incoming2)
	in, err := Parse(r)
	assert.NoError(t, err)
	assert.Equal(t, "google-angler-angler-O", in.Branch)
	assert.Equal(t, "", in.JIT)
	assert.Len(t, in.Metrics, 1)
	f, err := in.Metrics["coremark"]["score"].Float64()
	assert.NoError(t, err)
	assert.Equal(t, 5439.620216, f)
}

func TestParse_ErrReader_ReturnsError(t *testing.T) {
	_, err := Parse(iotest.ErrReader(fmt.Errorf("Failed")))
	assert.Contains(t, err.Error(), "Failed to decode")
}

type lookupMockGood struct {
}

func (l lookupMockGood) Lookup(buildid int64) (string, error) {
	return "8dcc84f7dc8523dd90501a4feb1f632808337c34", nil
}

type lookupMockBad struct {
}

func (l lookupMockBad) Lookup(buildid int64) (string, error) {
	return "", fmt.Errorf("Failed to find buildid.")
}

func benchDataFromJSON(t *testing.T, b []byte) format.BenchData {
	var ret format.BenchData
	err := json.Unmarshal(b, &ret)
	require.NoError(t, err)
	return ret
}

func v1FormatFromJSON(t *testing.T, b []byte) format.Format {
	var ret format.Format
	err := json.Unmarshal(b, &ret)
	require.NoError(t, err)
	return ret
}

func TestConvert_ParseIncoming_Success(t *testing.T) {
	c := New(lookupMockGood{})
	r := bytes.NewBufferString(incoming)
	key, gitHash, encodedAsJSON, err := c.Convert(r, txLogName)
	assert.NoError(t, err)
	assert.Equal(t, "8dcc84f7dc8523dd90501a4feb1f632808337c34", gitHash)

	benchData := benchDataFromJSON(t, encodedAsJSON)

	assert.Len(t, benchData.Results, 7)
	assert.Equal(t, 8.4, benchData.Results["android.platform.systemui.tests.jank.LauncherJankTests#testAppSwitchGMailtoHome"]["default"]["frame-avg-jank"])
	assert.Equal(t, "marlin-userdebug", key["build_flavor"])
	assert.Equal(t, "google-marlin-marlin-O", key["branch"])
	assert.Equal(t, "coral", key["device_name"])
	assert.Equal(t, "API_29_R", key["sdk_release_name"])
	assert.Equal(t, "disabled", key["jit"])
	require.Equal(t, txLogName, benchData.Source)
}

func TestConvert_ParseIncoming2_Success(t *testing.T) {
	c := New(lookupMockGood{})
	r := bytes.NewBufferString(incoming2)
	key, gitHash, encodedAsJSON, err := c.Convert(r, "")
	require.NoError(t, err)

	benchData := benchDataFromJSON(t, encodedAsJSON)

	assert.Equal(t, "8dcc84f7dc8523dd90501a4feb1f632808337c34", gitHash)
	assert.Len(t, benchData.Results, 1)
	assert.Equal(t, 5439.620216, benchData.Results["coremark"]["default"]["score"])
	assert.Equal(t, "google-angler-angler-O", key["branch"])
}

func TestConvert_NoMetrics_ReturnsErrIgnorable(t *testing.T) {
	r := bytes.NewBufferString(nometrics)
	c := New(lookupMockGood{})
	_, _, _, err := c.Convert(r, "")
	assert.Contains(t, err.Error(), ErrIgnorable.Error())
}

func TestConvert_HashLookupFails_ReturnsError(t *testing.T) {
	c := New(lookupMockBad{})
	r := bytes.NewBufferString(incoming)
	_, _, _, err := c.Convert(r, "")
	assert.Error(t, err)
}

func TestConvert_IgnorePresubmitResults_ReturnsErrIgnorable(t *testing.T) {

	c := New(lookupMockGood{})
	r := bytes.NewBufferString(incoming_presubmit)
	_, _, _, err := c.Convert(r, "")
	assert.Equal(t, ErrIgnorable, err)
}

type lookupMockGoodCheckBuildID struct {
	t *testing.T
}

func (l lookupMockGoodCheckBuildID) Lookup(buildid int64) (string, error) {
	require.Equal(l.t, int64(123456), buildid)
	return "8dcc84f7dc8523dd90501a4feb1f632808337c34", nil
}

func TestConvertAcceptsV1Format_Success(t *testing.T) {
	c := New(lookupMockGoodCheckBuildID{t})
	r := bytes.NewBufferString(v1Format)
	key, gitHash, encodedAsJSON, err := c.Convert(r, txLogName)
	require.NoError(t, err)

	data := v1FormatFromJSON(t, encodedAsJSON)

	assert.Equal(t, "8dcc84f7dc8523dd90501a4feb1f632808337c34", gitHash)

	expectedKey := map[string]string{
		"config": "8888",
		"arch":   "x86",
	}
	require.Equal(t, expectedKey, key)
	require.Equal(t, txLogName, data.Links[rawLogLocationKey])
	assert.Len(t, data.Results, 1)

}

func TestConvertAcceptsV1Format_LookupFails_ReturnsError(t *testing.T) {
	c := New(lookupMockBad{})
	r := bytes.NewBufferString(v1Format)
	_, _, _, err := c.Convert(r, "")
	require.Error(t, err)
}

const v1Format = `{
		"version": 1,
		"git_hash": "123456",
		"key": {
			"config": "8888",
			"arch": "x86"
		},
		"results": [
			{
				"key": {
					"test": "some_test_name"
				},
				"measurements": {
					"ms": [
						{
							"value": "min",
							"measurement": 1.2
						},
						{
							"value": "max",
							"measurement": 2.4
						},
						{
							"value": "median",
							"measurement": 1.5
						}
					]
				}
			}
		]
	}`

const incoming = `{
	"build_id": "3567162",
	"build_flavor": "marlin-userdebug",
	"device_name":"coral",
	"sdk_release_name":"API_29_R",
	"jit": "disabled",
	"metrics": {
		"android.platform.systemui.tests.jank.LauncherJankTests#testAppSwitchGMailtoHome": {
			"frame-fps": "9.328892269753897",
			"frame-avg-jank": "8.4",
			"frame-max-frame-duration": "7.834711093388444",
			"frame-max-jank": "10"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testHomeScreenSwipe": {
			"gfx-avg-slow-ui-thread": "0.10191099340499558",
			"gfx-max-slow-bitmap-uploads": "0.0",
			"gfx-max-frame-time-95": "8",
			"gfx-max-frame-time-50": "5",
			"gfx-max-slow-ui-thread": "0.25510204081632654",
			"gfx-avg-frame-time-50": "5.0",
			"gfx-max-jank": "0.26",
			"gfx-avg-slow-draw": "0.0",
			"gfx-avg-frame-time-95": "7.4",
			"gfx-max-frame-time-90": "7",
			"gfx-avg-frame-time-90": "6.8",
			"gfx-avg-jank": "0.10200000000000001",
			"gfx-max-missed-vsync": "0.0",
			"gfx-avg-slow-bitmap-uploads": "0.0",
			"gfx-max-high-input-latency": "0.0",
			"gfx-max-frame-time-99": "12",
			"gfx-avg-missed-vsync": "0.0",
			"gfx-avg-frame-time-99": "10.4",
			"gfx-max-slow-draw": "0.0",
			"gfx-avg-high-input-latency": "0.0"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testWidgetsContainerFling": {
			"gfx-avg-slow-ui-thread": "0.0968528680643497",
			"gfx-max-slow-bitmap-uploads": "0.0",
			"gfx-max-frame-time-95": "9",
			"gfx-max-frame-time-50": "5",
			"gfx-max-slow-ui-thread": "0.24271844660194172",
			"gfx-avg-frame-time-50": "5.0",
			"gfx-max-jank": "0.5",
			"gfx-avg-slow-draw": "0.0",
			"gfx-avg-frame-time-95": "8.2",
			"gfx-max-frame-time-90": "8",
			"gfx-avg-frame-time-90": "7.2",
			"gfx-avg-jank": "0.294",
			"gfx-max-missed-vsync": "0.24271844660194172",
			"gfx-avg-slow-bitmap-uploads": "0.0",
			"gfx-max-high-input-latency": "0.0",
			"gfx-max-frame-time-99": "15",
			"gfx-avg-missed-vsync": "0.14539655738473806",
			"gfx-avg-frame-time-99": "11.0",
			"gfx-max-slow-draw": "0.0",
			"gfx-avg-high-input-latency": "0.0"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testAllAppsContainerSwipe": {
			"gfx-avg-slow-ui-thread": "0.07554138508437006",
			"gfx-max-slow-bitmap-uploads": "0.07598784194528875",
			"gfx-max-frame-time-95": "9",
			"gfx-max-frame-time-50": "5",
			"gfx-max-slow-ui-thread": "0.1508295625942685",
			"gfx-avg-frame-time-50": "5.0",
			"gfx-max-jank": "0.3",
			"gfx-avg-slow-draw": "0.045592705167173245",
			"gfx-avg-frame-time-95": "8.2",
			"gfx-max-frame-time-90": "8",
			"gfx-avg-frame-time-90": "7.4",
			"gfx-avg-jank": "0.16599999999999998",
			"gfx-max-missed-vsync": "0.15232292460015232",
			"gfx-avg-slow-bitmap-uploads": "0.01519756838905775",
			"gfx-max-high-input-latency": "0.0",
			"gfx-max-frame-time-99": "11",
			"gfx-avg-missed-vsync": "0.07567937219606331",
			"gfx-avg-frame-time-99": "10.6",
			"gfx-max-slow-draw": "0.22796352583586624",
			"gfx-avg-high-input-latency": "0.0"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testAppSwitchChrometoHome": {
			"frame-fps": "9.059377622237943",
			"frame-avg-jank": "8.6",
			"frame-max-frame-duration": "11.048077785923113",
			"frame-max-jank": "9"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testAppSwitchPhotostoHome": {
			"frame-fps": "9.342556065428203",
			"frame-avg-jank": "8.0",
			"frame-max-frame-duration": "7.633792937351717",
			"frame-max-jank": "9"
		},
		"android.platform.systemui.tests.jank.LauncherJankTests#testOpenAllAppsContainer": {
			"gfx-avg-slow-ui-thread": "5.040813095770279",
			"gfx-max-slow-bitmap-uploads": "0.0",
			"gfx-max-frame-time-95": "24",
			"gfx-max-frame-time-50": "7",
			"gfx-max-slow-ui-thread": "5.352112676056338",
			"gfx-avg-frame-time-50": "7.0",
			"gfx-max-jank": "8.17",
			"gfx-avg-slow-draw": "1.5528189571212099",
			"gfx-avg-frame-time-95": "22.4",
			"gfx-max-frame-time-90": "14",
			"gfx-avg-frame-time-90": "12.8",
			"gfx-avg-jank": "7.148000000000001",
			"gfx-max-missed-vsync": "3.867403314917127",
			"gfx-avg-slow-bitmap-uploads": "0.0",
			"gfx-max-high-input-latency": "0.0",
			"gfx-max-frame-time-99": "61",
			"gfx-avg-missed-vsync": "3.4349243386335053",
			"gfx-avg-frame-time-99": "51.4",
			"gfx-max-slow-draw": "2.2535211267605635",
			"gfx-avg-high-input-latency": "0.0"
		}
	},
	"branch": "google-marlin-marlin-O"
}`

const incoming2 = `{
   "build_id" : "3842951",
   "metrics" : {
      "coremark" : {
         "score" : "5439.620216"
      }
   },
   "results_name" : "coremarkcom.google.android.performance.CoreMarkTest#coremark",
   "build_flavor" : "angler-userdebug",
   "branch" : "google-angler-angler-O"
}`

const nometrics = `{
	"build_id" : "3842951",
	"results_name" : "coremarkcom.google.android.performance.CoreMarkTest#coremark",
	"build_flavor" : "angler-userdebug",
	"branch" : "google-angler-angler-O"
 }`

// incoming_presubmit is a file with a build_id that begins with "P", which
// means it is a presubmit result and can be ignored.
const incoming_presubmit = `{
	"build_id" : "P3842951"
 }`