package goricochet

import "testing"
import "time"
import "log"

// The purpose of this test is to exercise the Unauthorized Error flows that occur
// when a client attempts to open a Chat Channel or Send a Contact Reuqest before Authentication
// itself with the Service.

type TestUnauthorizedService struct {
	StandardRicochetService
	FailedToOpen int
}

func (ts *TestUnauthorizedService) OnConnect(oc *OpenConnection) {
	if oc.Client {
		log.Printf("Attempting Authentication Not Authorized")
		oc.IsAuthed = true // Connections to Servers are Considered Authenticated by Default
		// REMOVED Authenticate
		oc.OpenChatChannel(5)
		oc.SendContactRequest(3, "test", "test")
	}
}

func (ts *TestUnauthorizedService) OnFailedChannelOpen(oc *OpenConnection, channelID int32, errorType string) {
	oc.UnsetChannel(channelID)
	if errorType == "UnauthorizedError" {
		ts.FailedToOpen++
	}
}

func TestUnauthorizedClientReject(t *testing.T) {
	ricochetService := new(TestService)
	err := ricochetService.Init("./private_key")

	if err != nil {
		t.Errorf("Could not initate ricochet service: %v", err)
	}

	go ricochetService.Listen(ricochetService, 9880)

	time.Sleep(time.Second * 2)

	ricochetService2 := new(TestUnauthorizedService)
	err = ricochetService2.Init("./private_key")

	if err != nil {
		t.Errorf("Could not initate ricochet service: %v", err)
	}

	go ricochetService2.Listen(ricochetService2, 9881)
	err = ricochetService2.Connect("127.0.0.1:9880|kwke2hntvyfqm7dr")
	if err != nil {
		t.Errorf("Could not connect to ricochet service:  %v", err)
	}

	time.Sleep(time.Second * 2)
	if ricochetService2.FailedToOpen != 2 {
		t.Errorf("Test server did not reject open channels with unauthorized error")
	}

}