mirror of https://bitbucket.org/ausocean/av.git
132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
/*
|
|
NAME
|
|
rtmp_test.go
|
|
|
|
DESCRIPTION
|
|
See Readme.md
|
|
|
|
AUTHOR
|
|
Saxon Nelson-Milton <saxon@ausocean.org>
|
|
|
|
LICENSE
|
|
rtmp_test.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)
|
|
|
|
It is free software: you can redistribute it and/or modify them
|
|
under the terms of the GNU General Public License as published by the
|
|
Free Software Foundation, either version 3 of the License, or (at your
|
|
option) any later version.
|
|
|
|
It is distributed in the hope that it will be useful, but WITHOUT
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
*/
|
|
|
|
package rtmp
|
|
|
|
import (
|
|
"testing"
|
|
"unsafe"
|
|
)
|
|
|
|
const (
|
|
arrStart = 0
|
|
arrEnd = 5
|
|
arrSize = 6
|
|
inc = 3
|
|
dec = 3
|
|
)
|
|
|
|
const (
|
|
byteSize = 1
|
|
int32Size = 4
|
|
int64Size = 8
|
|
)
|
|
|
|
var (
|
|
byteArr = [arrSize]byte{0x01, 0x02, 0x03, 0x04, 0x05, 0x06}
|
|
int32Arr = [arrSize]int32{1, 2, 3, 4, 5, 6}
|
|
int64Arr = [arrSize]int64{1, 2, 3, 4, 5, 6}
|
|
errMsg = "Obtained: %v, but wanted: %v"
|
|
)
|
|
|
|
func TestMemset(t *testing.T) {
|
|
size := 10
|
|
setNum := 5
|
|
testVal := byte('A')
|
|
mem := make([]byte, size)
|
|
memset(&mem[0], testVal, setNum)
|
|
for i := 0; i < size; i++ {
|
|
if i > setNum-1 {
|
|
testVal = byte(0)
|
|
}
|
|
if mem[i] != testVal {
|
|
t.Errorf("mem doesn't match expected values at: %v", i)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGoStrToCStr(t *testing.T) {
|
|
goStr := "string\000"
|
|
bStr := goStrToCStr(goStr)
|
|
|
|
testData := []byte{'s', 't', 'r', 'i', 'n', 'g', '\000'}
|
|
|
|
for i := 0; i < len(goStr); i++ {
|
|
val := (*[_Gi]byte)(unsafe.Pointer(bStr))[i]
|
|
testVal := testData[i]
|
|
if val != testVal {
|
|
t.Errorf("Wanted: %v, but got: %v", testVal, val)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIncPtr(t *testing.T) {
|
|
// Test how it deals with bytes
|
|
bytePtr := unsafe.Pointer(&byteArr[arrStart])
|
|
valueByte := *(*byte)(incPtr(bytePtr, inc, byteSize))
|
|
if valueByte != byteArr[inc] {
|
|
t.Errorf(errMsg, valueByte, byteArr[inc])
|
|
}
|
|
|
|
// Test how it deals with int32s
|
|
int32Ptr := unsafe.Pointer(&int32Arr[arrStart])
|
|
valueInt32 := *(*int32)(incPtr(int32Ptr, inc, int32Size))
|
|
if valueInt32 != int32Arr[inc] {
|
|
t.Errorf(errMsg, valueInt32, int32Arr[inc])
|
|
}
|
|
|
|
// Test how it deals with int64
|
|
int64Ptr := unsafe.Pointer(&int64Arr[arrStart])
|
|
valueInt64 := *(*int64)(incPtr(int64Ptr, inc, int64Size))
|
|
if valueInt64 != int64Arr[inc] {
|
|
t.Errorf(errMsg, valueInt64, int64Arr[inc])
|
|
}
|
|
}
|
|
|
|
func TestDecPtr(t *testing.T) {
|
|
// Test how it deals with bytes
|
|
bytePtr := unsafe.Pointer(&byteArr[arrEnd])
|
|
valueByte := *(*byte)(decPtr(bytePtr, dec, byteSize))
|
|
if valueByte != byteArr[arrEnd-dec] {
|
|
t.Errorf(errMsg, valueByte, byteArr[inc])
|
|
}
|
|
|
|
// Test how it deals with ints
|
|
int32Ptr := unsafe.Pointer(&int32Arr[arrEnd])
|
|
valueInt32 := *(*int32)(decPtr(int32Ptr, dec, int32Size))
|
|
if valueInt32 != int32Arr[arrEnd-inc] {
|
|
t.Errorf(errMsg, valueInt32, int32Arr[inc])
|
|
}
|
|
|
|
// Test how it deals with int64
|
|
int64Ptr := unsafe.Pointer(&int64Arr[arrEnd])
|
|
valueInt64 := *(*int64)(decPtr(int64Ptr, dec, int64Size))
|
|
if valueInt64 != int64Arr[arrEnd-dec] {
|
|
t.Errorf(errMsg, valueInt64, int64Arr[inc])
|
|
}
|
|
}
|