package app

import (
  "fmt"
  "os/user"
  "testing"

  "github.com/stretchr/testify/mock"
  "github.com/pkg/errors"

)

type mockValues struct {
  mock.Mock
}

func (m *mockValues) GetCurrentUser() (*user.User, error) {
  fmt.Println("Mocked Get User")
  args := m.Called()

  return args.Get(0).(*user.User), args.Error(1)
}

func TestGetDirPath(t *testing.T) {
  fakeUser := user.User{}
  u := &fakeUser

  myMocks := new(mockValues)

  myMocks.On("GetCurrentUser").Return(&fakeUser, nil)
  myMocks.On("GetPath", u, "", "test_image").Return("", "")

  imageClient := ImageClient{myMocks, "test_image", "test_meta"}
  _, dir, err := imageClient.GetDirPath("")
  if err != nil {
    t.Errorf("Path was incorrect, got: %q, want: %q.", dir, "some_path")
  }
}

func (m *mockValues) DBCreate(name string, key ImageKey, meta string, c Image) error{
    fmt.Println("Mocked Create image in Mongo")
    args := m.Called(name, key, meta, c)

    return args.Error(0)
}

func (m *mockValues) DBRead(name string, key ImageKey, meta string) ([]byte, error) {
    fmt.Println("Mocked Mongo DB Read Operations")
    args := m.Called(name, key, meta)

    return args.Get(0).([]byte), args.Error(1)
}

func (m *mockValues) DBUnmarshal(value []byte, c Image) error {
    fmt.Println("Mocked Mongo DB Unmarshal Operation")
    args := m.Called(value, c)

    return args.Error(0)
}

func (m *mockValues) GetPath(u *user.User, imageName string, storeName string) (string, string) {
    args := m.Called(u, "", "test_image")

    return args.String(0), args.String(1)
}

func (m *mockValues) OSMakeDir(dirPath string, perm int) error {
    fmt.Println("Mocked OS Create Directory Operation")
    args := m.Called(dirPath, perm)
    
    return args.Error(0)
}

func (m *mockValues) OSCreateFile(filePath string) error {
    fmt.Println("Mocked Create File Operation")
    args := m.Called(filePath)

    return args.Error(0)
}

func (m *mockValues) DBDelete(name string, key ImageKey, meta string) error {
    fmt.Println("Mocked Mongo DB Delete")
    args := m.Called(name, key, meta)

    return args.Error(0)

}

func (m *mockValues) OSRemove(filePath string) error {
    fmt.Println("Mocked OS File Remove")
    args := m.Called(filePath)

    return args.Error(0)
}

func (m *mockValues) DBUpdate(s string, k ImageKey, t string, c Image) error {
    fmt.Println("Mocked Mongo DB Update")
    args := m.Called(s, k, t, c)

    return args.Error(0)
}

func TestCreate(t *testing.T) {
    image := Image{}
    arr_data := []byte{}
    key := ImageKey{}
    myMocks := new(mockValues)
    // just to get an error value
    err1 := errors.New("math: square root of negative number")

    fakeUser := user.User{}
    u := &fakeUser

    myMocks.On("DBCreate", "test_image", key, "test_meta", image).Return(nil)
    myMocks.On("DBRead", "test_image", key, "test_meta").Return(arr_data, err1)
    myMocks.On("DBUnmarshal", arr_data, image).Return(nil)
    myMocks.On("GetCurrentUser").Return(&fakeUser, nil)
    myMocks.On("GetPath", u, "", "test_image").Return("", "")
    myMocks.On("OSMakeDir", "", 0744).Return(nil)
    myMocks.On("OSCreateFile", "").Return(nil)

    imageClient := ImageClient{myMocks, "test_image", "test_meta"}
    _, err := imageClient.Create(image)
    if err != nil {
        t.Errorf("Some error occured!")
    }
}

func TestDelete(t *testing.T) {
    key := ImageKey{}
    fakeUser := user.User{}
    u := &fakeUser

    myMocks := new(mockValues)

    myMocks.On("DBDelete", "test_image", key, "test_meta").Return(nil)
    myMocks.On("GetCurrentUser").Return(&fakeUser, nil)
    myMocks.On("GetPath", u, "", "test_image").Return("", "")
    myMocks.On("OSRemove", "").Return(nil)

    imageClient := ImageClient{myMocks, "test_image", "test_meta"}
    err := imageClient.Delete("")
    if err != nil {
        t.Errorf("Some error occured!")
    }

}

func TestUpdate(t *testing.T) {
    image := Image{}
    key := ImageKey{}
    arr_data := []byte{} 

    myMocks := new(mockValues)

    myMocks.On("DBRead", "test_image", key, "test_meta").Return(arr_data, nil)
    myMocks.On("DBUnmarshal", arr_data, image).Return(nil)
    myMocks.On("DBUpdate", "test_image", key, "test_meta", image).Return(nil)
    imageClient := ImageClient{myMocks, "test_image", "test_meta"}
    _, err := imageClient.Update("", image)
    if err != nil {
        t.Errorf("Some error occured!")
    }
}