// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package main

import (
	"fmt"
	"strings"
	"testing"

	"go.chromium.org/luci/vpython/api/vpython"

	. "github.com/smartystreets/goconvey/convey"
	. "go.chromium.org/luci/common/testing/assertions"
)

func TestPEP425TagSelector(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		tags     []*vpython.PEP425Tag
		template map[string]string
	}{
		{
			[]*vpython.PEP425Tag{
				{Python: "py2", Abi: "none", Platform: "any"},
				{Python: "py27", Abi: "none", Platform: "any"},
				{Python: "cp27", Abi: "cp27mu", Platform: "linux_x86_64"},
				{Python: "cp27", Abi: "cp27mu", Platform: "manylinux1_x86_64"},
				{Python: "cp27", Abi: "none", Platform: "manylinux1_x86_64"},
			},
			map[string]string{
				"platform":         "linux-amd64",
				"py_tag":           "cp27-cp27mu-manylinux1_x86_64",
				"py_python":        "cp27",
				"py_version":       "cp27",
				"py_abi":           "cp27mu",
				"py_platform":      "manylinux1_x86_64",
				"py_arch":          "manylinux1_x86_64",
				"vpython_platform": "linux-amd64_cp27_cp27mu",
			},
		},

		{
			[]*vpython.PEP425Tag{
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_12_x86_64"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_12_fat64"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_12_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_12_intel"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_10_intel"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_9_fat64"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_9_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_9_universal"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_8_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_8_universal"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_6_intel"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_6_fat64"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_6_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_6_universal"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_5_universal"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_4_intel"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_4_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_1_universal"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_0_fat32"},
				{Python: "cp27", Abi: "cp27m", Platform: "macosx_10_0_universal"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_12_x86_64"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_12_intel"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_12_fat64"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_9_universal"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_8_x86_64"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_8_intel"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_7_intel"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_7_fat64"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_7_fat32"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_6_universal"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_5_x86_64"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_5_intel"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_3_fat32"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_3_universal"},
				{Python: "cp27", Abi: "none", Platform: "macosx_10_2_fat32"},
				{Python: "py2", Abi: "none", Platform: "macosx_10_4_intel"},
				{Python: "cp27", Abi: "none", Platform: "any"},
			},
			map[string]string{
				"platform":         "mac-amd64",
				"py_tag":           "cp27-cp27m-macosx_10_4_intel",
				"py_python":        "cp27",
				"py_version":       "cp27",
				"py_abi":           "cp27m",
				"py_platform":      "macosx_10_4_intel",
				"py_arch":          "macosx_10_4_intel",
				"vpython_platform": "mac-amd64_cp27_cp27m",
			},
		},

		{
			[]*vpython.PEP425Tag{
				{Python: "py27", Abi: "none", Platform: "any"},
				{Python: "py27", Abi: "none", Platform: "linux_i686"},
			},
			map[string]string{
				"platform":         "linux-386",
				"py_tag":           "py27-none-linux_i686",
				"py_python":        "py27",
				"py_version":       "py27",
				"py_abi":           "none",
				"py_platform":      "linux_i686",
				"py_arch":          "linux_i686",
				"vpython_platform": "linux-386_py27_none",
			},
		},

		{
			[]*vpython.PEP425Tag{
				{Python: "py27", Abi: "none", Platform: "any"},
				{Python: "py27", Abi: "none", Platform: "linux_x86_64"},
			},
			map[string]string{
				"platform":         "linux-amd64",
				"py_tag":           "py27-none-linux_x86_64",
				"py_python":        "py27",
				"py_version":       "py27",
				"py_abi":           "none",
				"py_platform":      "linux_x86_64",
				"py_arch":          "linux_x86_64",
				"vpython_platform": "linux-amd64_py27_none",
			},
		},

		{
			// Tags taken from actual output running vpython3 on Linux x64.
			[]*vpython.PEP425Tag{
				{Python: "cp3", Abi: "none", Platform: "any"},
				{Python: "cp32", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp32", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp33", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp33", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp34", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp34", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp35", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp35", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp36", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp36", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp37", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp37", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp38", Abi: "abi3", Platform: "linux_x86_64"},
				{Python: "cp38", Abi: "abi3", Platform: "manylinux1_x86_64"},
				{Python: "cp38", Abi: "cp38", Platform: "linux_x86_64"},
				{Python: "cp38", Abi: "cp38", Platform: "manylinux1_x86_64"},
				{Python: "cp38", Abi: "none", Platform: "any"},
				{Python: "cp38", Abi: "none", Platform: "linux_x86_64"},
				{Python: "cp38", Abi: "none", Platform: "manylinux1_x86_64"},
				{Python: "py3", Abi: "none", Platform: "any"},
				{Python: "py3", Abi: "none", Platform: "linux_x86_64"},
				{Python: "py3", Abi: "none", Platform: "manylinux1_x86_64"},
				{Python: "py30", Abi: "none", Platform: "any"},
				{Python: "py31", Abi: "none", Platform: "any"},
				{Python: "py32", Abi: "none", Platform: "any"},
				{Python: "py33", Abi: "none", Platform: "any"},
				{Python: "py34", Abi: "none", Platform: "any"},
				{Python: "py35", Abi: "none", Platform: "any"},
				{Python: "py36", Abi: "none", Platform: "any"},
				{Python: "py37", Abi: "none", Platform: "any"},
				{Python: "py38", Abi: "none", Platform: "any"},
			},
			map[string]string{
				"platform":         "linux-amd64",
				"py_tag":           "cp38-cp38-manylinux1_x86_64",
				"py_python":        "cp38",
				"py_version":       "cp38",
				"py_abi":           "cp38",
				"py_platform":      "manylinux1_x86_64",
				"py_arch":          "manylinux1_x86_64",
				"vpython_platform": "linux-amd64_cp38_cp38",
			},
		},
	}

	Convey(`Testing PEP425 tag selection`, t, func() {
		for i, tc := range testCases {
			tagsStr := make([]string, len(tc.tags))
			for i, tag := range tc.tags {
				tagsStr[i] = tag.TagString()
			}
			t.Logf("Test case #%d, using tags: %v", i, tagsStr)

			tagsList := strings.Join(tagsStr, ", ")
			Convey(fmt.Sprintf(`Generates template for [%s]`, tagsList), func() {
				tag := pep425TagSelector(tc.tags)

				template, err := getPEP425CIPDTemplateForTag(tag)
				So(err, ShouldBeNil)
				So(template, ShouldResemble, tc.template)
			})
		}

		Convey(`Returns an error when no tag is selected.`, func() {
			tag := pep425TagSelector(nil)
			So(tag, ShouldBeNil)

			_, err := getPEP425CIPDTemplateForTag(tag)
			So(err, ShouldErrLike, "no PEP425 tag")
		})

		Convey(`Returns an error when an unknown platform is selected.`, func() {
			tag := pep425TagSelector([]*vpython.PEP425Tag{
				{Python: "py27", Abi: "none", Platform: "any"},
				{Python: "py27", Abi: "foo", Platform: "bar"},
			})
			So(tag, ShouldResemble, &vpython.PEP425Tag{Python: "py27", Abi: "foo", Platform: "bar"})

			_, err := getPEP425CIPDTemplateForTag(tag)
			So(err, ShouldErrLike, "failed to infer CIPD platform for tag")
		})
	})
}