wire_test.go

/*
 * Copyright 2022 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package fastpb

import (
	"fmt"
	"testing"

	"google.golang.org/protobuf/encoding/protowire"
)

func TestAppendTag(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 2
	expectS := "fa0f"

	n := AppendTag(buf, protowire.Number(255), protowire.BytesType)
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendTag[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestAppendVarint(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 5
	expectS := "ffffffff0f"

	n := AppendVarint(buf, 0xFFFFFFFF)
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendVarint[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestAppendFixed32(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 4
	expectS := "ffffffff"

	n := AppendFixed32(buf, 0xFFFFFFFF)
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendFixed32[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestAppendFixed64(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 8
	expectS := "ffffffff00000000"

	n := AppendFixed64(buf, 0xFFFFFFFF)
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendFixed64[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestAppendBytes(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 12
	expectS := "0b68656c6c6f20776f726c64"

	n := AppendBytes(buf, []byte("hello world"))
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendBytes[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestAppendString(t *testing.T) {
	buf := make([]byte, 64)
	expectN := 12
	expectS := "0b68656c6c6f20776f726c64"

	n := AppendString(buf, "hello world")
	gotS := fmt.Sprintf("%x", buf[:n])
	if n != expectN || gotS != expectS {
		panic(fmt.Errorf("AppendString[%d][%x] != [%d][%s]\n", n, buf[:n], expectN, expectS))
	}
}

func TestEnforceUTF8(t *testing.T) {
	t.Skip()
	if !EnforceUTF8() {
		panic(fmt.Errorf("EnforceUTF8 must be true"))
	}
}