// Copyright (C) The Arvados Authors. All rights reserved.
//
// SPDX-License-Identifier: AGPL-3.0

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"strings"
	"testing"

	"git.arvados.org/arvados.git/sdk/go/arvados"
	. "gopkg.in/check.v1"
)

// Gocheck boilerplate
func Test(t *testing.T) {
	TestingT(t)
}

type TestSuite struct {
	cfg   *ConfigParams
	ac    *arvados.Client
	users map[string]arvados.User
}

func (s *TestSuite) SetUpTest(c *C) {
	s.ac = arvados.NewClientFromEnv()
	u, err := s.ac.CurrentUser()
	c.Assert(err, IsNil)
	c.Assert(u.IsAdmin, Equals, true)

	s.users = make(map[string]arvados.User)
	ul := arvados.UserList{}
	s.ac.RequestAndDecode(&ul, "GET", "/arvados/v1/users", nil, arvados.ResourceListParams{})
	c.Assert(ul.ItemsAvailable, Not(Equals), 0)
	s.users = make(map[string]arvados.User)
	for _, u := range ul.Items {
		s.users[u.UUID] = u
	}

	// Set up command config
	os.Args = []string{"cmd", "somefile.csv"}
	config, err := GetConfig()
	c.Assert(err, IsNil)
	s.cfg = &config
}

func (s *TestSuite) TearDownTest(c *C) {
	var dst interface{}
	// Reset database to fixture state after every test run.
	err := s.cfg.Client.RequestAndDecode(&dst, "POST", "/database/reset", nil, nil)
	c.Assert(err, IsNil)
}

var _ = Suite(&TestSuite{})

// MakeTempCSVFile creates a temp file with data as comma separated values
func MakeTempCSVFile(data [][]string) (f *os.File, err error) {
	f, err = ioutil.TempFile("", "test_sync_users")
	if err != nil {
		return
	}
	for _, line := range data {
		fmt.Fprintf(f, "%s\n", strings.Join(line, ","))
	}
	err = f.Close()
	return
}

func ListUsers(ac *arvados.Client) ([]arvados.User, error) {
	var ul arvados.UserList
	err := ac.RequestAndDecode(&ul, "GET", "/arvados/v1/users", nil, arvados.ResourceListParams{})
	if err != nil {
		return nil, err
	}
	return ul.Items, nil
}

func (s *TestSuite) TestParseFlagsWithoutPositionalArgument(c *C) {
	os.Args = []string{"cmd", "-verbose"}
	err := ParseFlags(&ConfigParams{})
	c.Assert(err, NotNil)
}

func (s *TestSuite) TestParseFlagsWithPositionalArgument(c *C) {
	cfg := ConfigParams{}
	os.Args = []string{"cmd", "/tmp/somefile.csv"}
	err := ParseFlags(&cfg)
	c.Assert(err, IsNil)
	c.Assert(cfg.Path, Equals, "/tmp/somefile.csv")
	c.Assert(cfg.Verbose, Equals, false)
	c.Assert(cfg.DeactivateUnlisted, Equals, false)
}

func (s *TestSuite) TestParseFlagsWithOptionalFlags(c *C) {
	cfg := ConfigParams{}
	os.Args = []string{"cmd", "-verbose", "-deactivate-unlisted", "/tmp/somefile.csv"}
	err := ParseFlags(&cfg)
	c.Assert(err, IsNil)
	c.Assert(cfg.Path, Equals, "/tmp/somefile.csv")
	c.Assert(cfg.Verbose, Equals, true)
	c.Assert(cfg.DeactivateUnlisted, Equals, true)
}

func (s *TestSuite) TestGetConfig(c *C) {
	os.Args = []string{"cmd", "/tmp/somefile.csv"}
	cfg, err := GetConfig()
	c.Assert(err, IsNil)
	c.Assert(cfg.AnonUserUUID, Not(Equals), "")
	c.Assert(cfg.SysUserUUID, Not(Equals), "")
	c.Assert(cfg.CurrentUser, Not(Equals), "")
	c.Assert(cfg.ClusterID, Not(Equals), "")
	c.Assert(cfg.Client, NotNil)
}

func (s *TestSuite) TestFailOnEmptyFields(c *C) {
	records := [][]string{
		{"", "first-name", "last-name", "1", "0"},
		{"user@example", "", "last-name", "1", "0"},
		{"user@example", "first-name", "", "1", "0"},
		{"user@example", "first-name", "last-name", "", "0"},
		{"user@example", "first-name", "last-name", "1", ""},
	}
	for _, record := range records {
		data := [][]string{record}
		tmpfile, err := MakeTempCSVFile(data)
		c.Assert(err, IsNil)
		defer os.Remove(tmpfile.Name())
		s.cfg.Path = tmpfile.Name()
		err = doMain(s.cfg)
		c.Assert(err, NotNil)
		c.Assert(err, ErrorMatches, ".*fields cannot be empty.*")
	}
}

func (s *TestSuite) TestIgnoreSpaces(c *C) {
	// Make sure users aren't already there from fixtures
	for _, user := range s.users {
		e := user.Email
		found := e == "user1@example.com" || e == "user2@example.com" || e == "user3@example.com"
		c.Assert(found, Equals, false)
	}
	// Use CSV data with leading/trailing whitespaces, confirm that they get ignored
	data := [][]string{
		{" user1@example.com", "  Example", "   User1", "1", "0"},
		{"user2@example.com ", "Example  ", "User2   ", "1", "0"},
		{" user3@example.com ", "  Example  ", "   User3   ", "1", "0"},
	}
	tmpfile, err := MakeTempCSVFile(data)
	c.Assert(err, IsNil)
	defer os.Remove(tmpfile.Name())
	s.cfg.Path = tmpfile.Name()
	err = doMain(s.cfg)
	c.Assert(err, IsNil)
	users, err := ListUsers(s.cfg.Client)
	c.Assert(err, IsNil)
	for _, userNr := range []int{1, 2, 3} {
		found := false
		for _, user := range users {
			if user.Email == fmt.Sprintf("user%d@example.com", userNr) &&
				user.LastName == fmt.Sprintf("User%d", userNr) &&
				user.FirstName == "Example" && user.IsActive == true {
				found = true
				break
			}
		}
		c.Assert(found, Equals, true)
	}
}

// Error out when records have != 5 records
func (s *TestSuite) TestWrongNumberOfFields(c *C) {
	for _, testCase := range [][][]string{
		{{"user1@example.com", "Example", "User1", "1"}},
		{{"user1@example.com", "Example", "User1", "1", "0", "extra data"}},
	} {
		tmpfile, err := MakeTempCSVFile(testCase)
		c.Assert(err, IsNil)
		defer os.Remove(tmpfile.Name())
		s.cfg.Path = tmpfile.Name()
		err = doMain(s.cfg)
		c.Assert(err, NotNil)
		c.Assert(err, ErrorMatches, ".*expected 5 fields, found.*")
	}
}

// Error out when records have incorrect data types
func (s *TestSuite) TestWrongDataFields(c *C) {
	for _, testCase := range [][][]string{
		{{"user1@example.com", "Example", "User1", "yep", "0"}},
		{{"user1@example.com", "Example", "User1", "1", "nope"}},
	} {
		tmpfile, err := MakeTempCSVFile(testCase)
		c.Assert(err, IsNil)
		defer os.Remove(tmpfile.Name())
		s.cfg.Path = tmpfile.Name()
		err = doMain(s.cfg)
		c.Assert(err, NotNil)
		c.Assert(err, ErrorMatches, ".*parsing error at line.*[active|admin] status not recognized.*")
	}
}

// Activate and deactivate users
func (s *TestSuite) TestUserCreationAndUpdate(c *C) {
	testCases := []struct {
		Email     string
		FirstName string
		LastName  string
		Active    bool
		Admin     bool
	}{{
		Email:     "user1@example.com",
		FirstName: "Example",
		LastName:  "User1",
		Active:    true,
		Admin:     false,
	}, {
		Email:     "admin1@example.com",
		FirstName: "Example",
		LastName:  "Admin1",
		Active:    true,
		Admin:     true,
	}}
	// Make sure users aren't already there from fixtures
	for _, user := range s.users {
		e := user.Email
		found := e == testCases[0].Email || e == testCases[1].Email
		c.Assert(found, Equals, false)
	}
	// User creation
	data := [][]string{
		{testCases[0].Email, testCases[0].FirstName, testCases[0].LastName, fmt.Sprintf("%t", testCases[0].Active), fmt.Sprintf("%t", testCases[0].Admin)},
		{testCases[1].Email, testCases[1].FirstName, testCases[1].LastName, fmt.Sprintf("%t", testCases[1].Active), fmt.Sprintf("%t", testCases[1].Admin)},
	}
	tmpfile, err := MakeTempCSVFile(data)
	c.Assert(err, IsNil)
	defer os.Remove(tmpfile.Name())
	s.cfg.Path = tmpfile.Name()
	err = doMain(s.cfg)
	c.Assert(err, IsNil)

	users, err := ListUsers(s.cfg.Client)
	c.Assert(err, IsNil)
	for _, tc := range testCases {
		var foundUser arvados.User
		for _, user := range users {
			if user.Email == tc.Email {
				foundUser = user
				break
			}
		}
		c.Assert(foundUser, NotNil)
		c.Logf("Checking recently created user %q", foundUser.Email)
		c.Assert(foundUser.FirstName, Equals, tc.FirstName)
		c.Assert(foundUser.LastName, Equals, tc.LastName)
		c.Assert(foundUser.IsActive, Equals, true)
		c.Assert(foundUser.IsAdmin, Equals, tc.Admin)
	}
	// User deactivation
	testCases[0].Active = false
	testCases[1].Active = false
	data = [][]string{
		{testCases[0].Email, testCases[0].FirstName, testCases[0].LastName, fmt.Sprintf("%t", testCases[0].Active), fmt.Sprintf("%t", testCases[0].Admin)},
		{testCases[1].Email, testCases[1].FirstName, testCases[1].LastName, fmt.Sprintf("%t", testCases[1].Active), fmt.Sprintf("%t", testCases[1].Admin)},
	}
	tmpfile, err = MakeTempCSVFile(data)
	c.Assert(err, IsNil)
	defer os.Remove(tmpfile.Name())
	s.cfg.Path = tmpfile.Name()
	err = doMain(s.cfg)
	c.Assert(err, IsNil)

	users, err = ListUsers(s.cfg.Client)
	c.Assert(err, IsNil)
	for _, tc := range testCases {
		var foundUser arvados.User
		for _, user := range users {
			if user.Email == tc.Email {
				foundUser = user
				break
			}
		}
		c.Assert(foundUser, NotNil)
		c.Logf("Checking recently deactivated user %q", foundUser.Email)
		c.Assert(foundUser.FirstName, Equals, tc.FirstName)
		c.Assert(foundUser.LastName, Equals, tc.LastName)
		c.Assert(foundUser.IsActive, Equals, false)
		c.Assert(foundUser.IsAdmin, Equals, false) // inactive users cannot be admins
	}
}