package main

import (
	"context"
	"fmt"
	"os"
	"testing"

	"errors"
	"github.com/sirupsen/logrus"
	"github.com/zcash/lightwalletd/common"
	"google.golang.org/grpc"
	"google.golang.org/grpc/peer"
)

var step int

func testhandler(ctx context.Context, req interface{}) (interface{}, error) {
	step++
	switch step {
	case 1:
		return nil, errors.New("test error")
	case 2:
		return nil, nil
	}
	return nil, nil
}

func TestLogInterceptor(t *testing.T) {
	output, err := os.OpenFile("test-log", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
	if err != nil {
		os.Stderr.WriteString(fmt.Sprint("Cannot open test-log:", err))
		os.Exit(1)
	}
	logger := logrus.New()
	logger.SetOutput(output)
	common.Log = logger.WithFields(logrus.Fields{
		"app": "test",
	})
	var req interface{}
	resp, err := logInterceptor(peer.NewContext(context.Background(), &peer.Peer{}),
		&req, &grpc.UnaryServerInfo{}, testhandler)
	if err == nil {
		t.Fatal("unexpected success")
	}
	if resp != nil {
		t.Fatal("unexpected response", resp)
	}
	resp, err = logInterceptor(context.Background(), &req, &grpc.UnaryServerInfo{}, testhandler)
	if err != nil {
		t.Fatal("unexpected error", err)
	}
	if resp != nil {
		t.Fatal("unexpected response", resp)
	}
	os.Remove("test-log")
	step = 0
}

func TestFileExists(t *testing.T) {
	if fileExists("nonexistent-file") {
		t.Fatal("fileExists unexpected success")
	}
	// If the path exists but is a directory, should return false
	if fileExists(".") {
		t.Fatal("fileExists unexpected success")
	}
	// The following file should exist, it's what's being tested
	if !fileExists("main.go") {
		t.Fatal("fileExists failed")
	}
}