package goricochet

import (
  "testing"
  "github.com/s-rah/go-ricochet/utils"
  "net"
  "time"
)


func SimpleServer() {
    ln,_ := net.Listen("tcp", "127.0.0.1:11000")
    conn,_ := ln.Accept()
    b := make([]byte, 4)
    n,err := conn.Read(b)
    if n == 4 && err == nil {
        conn.Write([]byte{0x01})
    }
    conn.Close()
}

func BadVersionNegotiation() {
    ln,_ := net.Listen("tcp", "127.0.0.1:11001")
    conn,_ := ln.Accept()
    // We are already testing negotiation bytes, we don't care, just send a termination.
    conn.Write([]byte{0x00})
    conn.Close()
}

func NotRicochetServer() {
    ln,_ := net.Listen("tcp", "127.0.0.1:11002")
    conn,_ := ln.Accept()
    conn.Close()
}

func TestRicochet(t *testing.T) {
    go SimpleServer()
    // Wait for Server to Initialize
    time.Sleep(time.Second)

    rc,err := Open("127.0.0.1:11000|abcdefghijklmno.onion")
    if err == nil {
        if rc.IsInbound {
            t.Errorf("RicochetConnection declares itself as an Inbound connection after an Outbound attempt...that shouldn't happen")
        }
        return
    }
    t.Errorf("RicochetProtocol: Open Failed: %v", err)
}

func TestBadVersionNegotiation(t*testing.T) {
    go BadVersionNegotiation()
    time.Sleep(time.Second)

    _,err := Open("127.0.0.1:11001|abcdefghijklmno.onion")
    if err != utils.VersionNegotiationFailed {
        t.Errorf("RicochetProtocol: Server Had No Correct Version - Should Have Failed: err = %v", err)
    }
}


func TestNotARicochetServer(t*testing.T) {
    go NotRicochetServer()
    time.Sleep(time.Second)

    _,err := Open("127.0.0.1:11002|abcdefghijklmno.onion")
    if err != utils.VersionNegotiationError {
        t.Errorf("RicochetProtocol: Server Had No Correct Version - Should Have Failed: err = %v", err)
    }
}