// Copyright (C) 2022 The Android Open Source Project
//
// 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 local

import (
 "bytes"
 "context"
 "reflect"
 "testing"

 "tools/treble/build/report/app"
)

// Test cases for local GIT.
type TestCmd struct {
 err  error
 text string
}
type gitTestCli struct {
 revParse   *TestCmd
 remoteUrl  *TestCmd
 tree       *TestCmd
 commit     *TestCmd
 diffBranch *TestCmd
}

func (g *gitTestCli) ProjectInfo(ctx context.Context, gitDir, workDir string) (*bytes.Buffer, error) {
 return bytes.NewBufferString(g.revParse.text), g.revParse.err
}
func (g *gitTestCli) RemoteUrl(ctx context.Context, gitDir, workDir, remote string) (*bytes.Buffer, error) {
 return bytes.NewBufferString(g.remoteUrl.text), g.remoteUrl.err
}
func (g *gitTestCli) Tree(ctx context.Context, gitDir, workDir, revision string) (*bytes.Buffer, error) {
 return bytes.NewBufferString(g.tree.text), g.tree.err
}
func (g *gitTestCli) CommitInfo(ctx context.Context, gitDir, workDir, sha string) (*bytes.Buffer, error) {
 return bytes.NewBufferString(g.commit.text), g.tree.err
}
func (g *gitTestCli) DiffBranches(ctx context.Context, gitDir, workDir, upstream, sha string) (*bytes.Buffer, error) {
 return bytes.NewBufferString(g.diffBranch.text), g.tree.err
}

func Test_git(t *testing.T) {

 type projectTest struct {
  revCmd    *TestCmd
  remoteCmd *TestCmd
  treeCmd   *TestCmd
  res       *app.GitProject
 }

 type commitTest struct {
  sha string
  cmd *TestCmd
  res *app.GitCommit
 }

 tests := []struct {
  path     string
  gitDir   string
  remote   string
  revision string
  getFiles bool
  project  projectTest
  commit   commitTest
 }{
  {
   path:     "work/dir",
   gitDir:   "",
   remote:   "origin",
   revision: "sha_revision",
   getFiles: true,
   project: projectTest{
    revCmd:    &TestCmd{text: "/abs/path/to/work/dir\nsha_revision\n", err: nil},
    remoteCmd: &TestCmd{text: "http://url/workdir", err: nil},
    treeCmd:   &TestCmd{text: "", err: nil},
    res: &app.GitProject{
     RepoDir:   "work/dir",
     WorkDir:   "/abs/path/to/work/dir",
     GitDir:    ".git",
     Remote:    "origin",
     RemoteUrl: "http://url/workdir",
     Revision:  "sha_revision",
     Files:     make(map[string]*app.GitTreeObj)},
   },
   // Test empty commit
   commit: commitTest{
    sha: "commit_sha",
    cmd: &TestCmd{text: "commit_sha", err: nil},
    res: &app.GitCommit{Sha: "commit_sha", Files: []app.GitCommitFile{}},
   },
  },
  {
   path:     "work/dir",
   gitDir:   "",
   remote:   "origin",
   revision: "sha_revision",
   getFiles: true,
   project: projectTest{
    revCmd:    &TestCmd{text: "/abs/path/to/work/dir\nsha_revision\n", err: nil},
    remoteCmd: &TestCmd{text: "http://url/workdir", err: nil},
    treeCmd:   &TestCmd{text: "100644 blob 0000000000000000000000000000000000000001 file.1\n", err: nil},
    res: &app.GitProject{
     RepoDir:   "work/dir",
     WorkDir:   "/abs/path/to/work/dir",
     GitDir:    ".git",
     Remote:    "origin",
     RemoteUrl: "http://url/workdir",
     Revision:  "sha_revision",
     Files: map[string]*app.GitTreeObj{"file.1": &app.GitTreeObj{Permissions: "100644", Type: "blob",
      Sha: "0000000000000000000000000000000000000001", Filename: "file.1"}}},
   },
   commit: commitTest{
    sha: "HEAD",
    cmd: &TestCmd{text: "sha_for_head\nR removed.1\nA added.1\nM modified.1\n", err: nil},
    res: &app.GitCommit{
     Sha: "sha_for_head",
     Files: []app.GitCommitFile{
      {Filename: "removed.1", Type: app.GitFileRemoved},
      {Filename: "added.1", Type: app.GitFileAdded},
      {Filename: "modified.1", Type: app.GitFileModified},
     },
    },
   },
  },
 }
 for _, test := range tests {
  git := &gitCli{git: &gitTestCli{
   revParse:  test.project.revCmd,
   remoteUrl: test.project.remoteCmd,
   tree:      test.project.treeCmd,
   commit:    test.commit.cmd,
  }}

  proj, err := git.Project(nil, test.path, test.gitDir, test.remote, test.revision)
  if err != nil {
   t.Fatal("Failed to parse project")
  }
  if test.getFiles {
   _ = git.PopulateFiles(nil, proj, "")
  }
  if !reflect.DeepEqual(*proj, *test.project.res) {
   t.Errorf("Project = %+v; want %+v", *proj, *test.project.res)
  }
  if test.commit.cmd != nil {
   c, err := git.CommitInfo(nil, proj, test.commit.sha)
   if err != nil {
    t.Errorf("Failed to get; %v", test)
   } else {
    if !reflect.DeepEqual(*c, *test.commit.res) {
     t.Errorf("Commit = %v; want %v", c, *test.commit.res)
    }
   }
  }
 }

}