pulsar/internal/checksum_test.go

//
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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 internal

import (
    "bytes"
    "hash/crc32"
    "testing"
)

func TestFrameChecksum(t *testing.T) {
    input := []byte{1, 2, 3, 4, 5}
    var f CheckSum

    if got := f.compute(); got != nil {
        t.Fatalf("compute() = %v; expected nil", got)
    }

    if _, err := f.Write(input); err != nil {
        t.Fatalf("Write() err = %v; expected nil", err)
    }

    h := crc32.New(crc32.MakeTable(crc32.Castagnoli))
    if _, err := h.Write(input); err != nil {
        t.Fatal(err)
    }

    if got, expected := f.compute(), h.Sum(nil); !bytes.Equal(got, expected) {
        t.Fatalf("compute() = %x; expected %x", got, expected)
    } else {
        t.Logf("compute() = 0x%x", got)
    }
}