misc_test.go

package gota

import (
	"bytes"
	"io"
	"testing"
	"time"
)

func TestStatistic_AddSentBytes(t *testing.T) {
	s := &Statistic{}
	sent := uint64(1024)
	s.AddSentBytes(sent)
	if s.SentBytes != 1024 {
		t.Errorf("Error SentBytes: %d", s.SentBytes)
	}
}

func TestStatistic_AddReceivedBytes(t *testing.T) {
	s := &Statistic{}
	sent := uint64(1024)
	s.AddReceivedBytes(sent)
	if s.ReceivedBytes != 1024 {
		t.Errorf("Error ReceivedBytes: %d", s.ReceivedBytes)
	}
}

func TestStatistic_SendSpeedSecond(t *testing.T) {
	s := NewStatistic(3)
	s.AddSentBytes(10)
	s.AddSentBytes(10)
	time.Sleep(1 * time.Second)
	s.AddSentBytes(20)
	s.AddSentBytes(10)
	time.Sleep(1 * time.Second)
	s.AddSentBytes(30)
	s.AddSentBytes(10)
	ss := s.SendSpeedSecond(3)
	if ss != 30 {
		t.Errorf("Error speed: %d", ss)
	}
}

func TestStatistic_ReceiveSpeedSecond(t *testing.T) {
	s := NewStatistic(3)
	s.AddReceivedBytes(10)
	s.AddReceivedBytes(10)
	time.Sleep(1 * time.Second)
	s.AddReceivedBytes(20)
	s.AddReceivedBytes(10)
	time.Sleep(1 * time.Second)
	s.AddReceivedBytes(30)
	s.AddReceivedBytes(10)
	ss := s.ReceiveSpeedSecond(3)
	if ss != 30 {
		t.Errorf("Error speed: %d", ss)
	}
}

func TestReadNBytes(t *testing.T) {
	str := "Hello world, I'm Jim"
	lens := 20
	var testBuf bytes.Buffer
	testBuf.Write([]byte(str))

	full, err := ReadNBytes(&testBuf, lens)
	if err != nil && err != io.EOF {
		t.Errorf("ReadNBytes Error: %s", err)
	}

	if string(full) != str {
		t.Error("ReadNBytes Error, different string")
	}

	testBuf.Write([]byte(str))
	partial, err := ReadNBytes(&testBuf, 5)
	t.Logf("%s", partial)

	if err != nil {
		t.Errorf("ReadNBytes Error: %s", err)
	}

	if string(partial) != str[:5] {
		t.Error("ReadNBytes Error, different string")
	}
}

func TestWriteNBytes(t *testing.T) {
	str := "Hello world, I'm Jim"
	lens := 20

	var testWrite bytes.Buffer
	err := WriteNBytes(&testWrite, lens, []byte(str))

	if err != nil && err != io.EOF {
		t.Errorf("WriteNBytes Error: %s", err)
	}

	if string(testWrite.Bytes()) != str {
		t.Errorf("WriteNBytes Error, different string")
	}
}