// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"bytes"
	"fmt"
	"os"
	"strings"
	"testing"

	"android/soong/tools/compliance"
)

func TestMain(m *testing.M) {
	// Change into the parent directory before running the tests
	// so they can find the testdata directory.
	if err := os.Chdir(".."); err != nil {
		fmt.Printf("failed to change to testdata directory: %s\n", err)
		os.Exit(1)
	}
	os.Exit(m.Run())
}

func Test_plaintext(t *testing.T) {
	tests := []struct {
		condition   string
		name        string
		outDir      string
		roots       []string
		ctx         context
		expectedOut []string
	}{
		{
			condition:   "firstparty",
			name:        "apex",
			roots:       []string{"highest.apex.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition: "firstparty",
			name:      "apex_trimmed",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/firstparty/bin/bin1.meta_lic"},
				stripPrefix: []string{"testdata/firstparty/"},
			},
			expectedOut: []string{},
		},
		{
			condition:   "firstparty",
			name:        "container",
			roots:       []string{"container.zip.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "firstparty",
			name:        "application",
			roots:       []string{"application.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "firstparty",
			name:        "binary",
			roots:       []string{"bin/bin1.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "firstparty",
			name:        "library",
			roots:       []string{"lib/libd.so.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "notice",
			name:        "apex",
			roots:       []string{"highest.apex.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition: "notice",
			name:      "apex_trimmed",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/notice/bin/bin1.meta_lic"},
				stripPrefix: []string{"testdata/notice/"},
			},
			expectedOut: []string{},
		},
		{
			condition:   "notice",
			name:        "container",
			roots:       []string{"container.zip.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "notice",
			name:        "application",
			roots:       []string{"application.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "notice",
			name:        "binary",
			roots:       []string{"bin/bin1.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "notice",
			name:        "library",
			roots:       []string{"lib/libd.so.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "reciprocal",
			name:        "apex",
			roots:       []string{"highest.apex.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition: "reciprocal",
			name:      "apex_trimmed",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/reciprocal/bin/bin1.meta_lic"},
				stripPrefix: []string{"testdata/reciprocal/"},
			},
			expectedOut: []string{},
		},
		{
			condition:   "reciprocal",
			name:        "container",
			roots:       []string{"container.zip.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "reciprocal",
			name:        "application",
			roots:       []string{"application.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "reciprocal",
			name:        "binary",
			roots:       []string{"bin/bin1.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "reciprocal",
			name:        "library",
			roots:       []string{"lib/libd.so.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition: "restricted",
			name:      "apex",
			roots:     []string{"highest.apex.meta_lic"},
			expectedOut: []string{
				"testdata/restricted/lib/liba.so.meta_lic restricted_if_statically_linked",
				"testdata/restricted/lib/libb.so.meta_lic restricted",
			},
		},
		{
			condition: "restricted",
			name:      "apex_trimmed_bin1",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/restricted/bin/bin1.meta_lic"},
				stripPrefix: []string{"testdata/restricted/"},
			},
			expectedOut: []string{"lib/liba.so.meta_lic restricted_if_statically_linked"},
		},
		{
			condition: "restricted",
			name:      "apex_trimmed_bin2",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/restricted/bin/bin2.meta_lic"},
				stripPrefix: []string{"testdata/restricted/"},
			},
			expectedOut: []string{"lib/libb.so.meta_lic restricted"},
		},
		{
			condition: "restricted",
			name:      "container",
			roots:     []string{"container.zip.meta_lic"},
			expectedOut: []string{
				"testdata/restricted/lib/liba.so.meta_lic restricted_if_statically_linked",
				"testdata/restricted/lib/libb.so.meta_lic restricted",
			},
		},
		{
			condition:   "restricted",
			name:        "application",
			roots:       []string{"application.meta_lic"},
			expectedOut: []string{"testdata/restricted/lib/liba.so.meta_lic restricted_if_statically_linked"},
		},
		{
			condition:   "restricted",
			name:        "binary",
			roots:       []string{"bin/bin1.meta_lic"},
			expectedOut: []string{"testdata/restricted/lib/liba.so.meta_lic restricted_if_statically_linked"},
		},
		{
			condition:   "restricted",
			name:        "library",
			roots:       []string{"lib/libd.so.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "proprietary",
			name:        "apex",
			roots:       []string{"highest.apex.meta_lic"},
			expectedOut: []string{"testdata/proprietary/lib/libb.so.meta_lic restricted"},
		},
		{
			condition: "proprietary",
			name:      "apex_trimmed_bin1",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/proprietary/bin/bin1.meta_lic"},
				stripPrefix: []string{"testdata/proprietary/"},
			},
			expectedOut: []string{},
		},
		{
			condition: "proprietary",
			name:      "apex_trimmed_bin2",
			roots:     []string{"highest.apex.meta_lic"},
			ctx: context{
				sources:     []string{"testdata/proprietary/bin/bin2.meta_lic"},
				stripPrefix: []string{"testdata/proprietary/"},
			},
			expectedOut: []string{"lib/libb.so.meta_lic restricted"},
		},
		{
			condition:   "proprietary",
			name:        "container",
			roots:       []string{"container.zip.meta_lic"},
			expectedOut: []string{"testdata/proprietary/lib/libb.so.meta_lic restricted"},
		},
		{
			condition:   "proprietary",
			name:        "application",
			roots:       []string{"application.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "proprietary",
			name:        "binary",
			roots:       []string{"bin/bin1.meta_lic"},
			expectedOut: []string{},
		},
		{
			condition:   "proprietary",
			name:        "library",
			roots:       []string{"lib/libd.so.meta_lic"},
			expectedOut: []string{},
		},
	}
	for _, tt := range tests {
		t.Run(tt.condition+" "+tt.name, func(t *testing.T) {
			expectedOut := &bytes.Buffer{}
			for _, eo := range tt.expectedOut {
				expectedOut.WriteString(eo)
				expectedOut.WriteString("\n")
			}
			fmt.Fprintf(expectedOut, "restricted conditions trace to %d targets\n", len(tt.expectedOut))
			if 0 == len(tt.expectedOut) {
				fmt.Fprintln(expectedOut, "  (check for typos in project names or metadata files)")
			}

			stdout := &bytes.Buffer{}
			stderr := &bytes.Buffer{}

			rootFiles := make([]string, 0, len(tt.roots))
			for _, r := range tt.roots {
				rootFiles = append(rootFiles, "testdata/"+tt.condition+"/"+r)
			}
			if len(tt.ctx.sources) < 1 {
				tt.ctx.sources = rootFiles
			}
			_, err := traceRestricted(&tt.ctx, stdout, stderr, compliance.GetFS(tt.outDir), rootFiles...)
			t.Logf("rtrace: stderr = %v", stderr)
			t.Logf("rtrace: stdout = %v", stdout)
			if err != nil {
				t.Fatalf("rtrace: error = %v", err)
				return
			}
			if stderr.Len() > 0 {
				t.Errorf("rtrace: gotStderr = %v, want none", stderr)
			}
			out := stdout.String()
			expected := expectedOut.String()
			if out != expected {
				outList := strings.Split(out, "\n")
				expectedList := strings.Split(expected, "\n")
				startLine := 0
				for startLine < len(outList) && startLine < len(expectedList) && outList[startLine] == expectedList[startLine] {
					startLine++
				}
				t.Errorf("rtrace: gotStdout = %v, want %v, somewhere near line %d Stdout = %v, want %v",
					out, expected, startLine+1, outList[startLine], expectedList[startLine])
			}
		})
	}
}