ceph/src/arrow/go/arrow/memory/cgo_allocator_test.go

   1 // Licensed to the Apache Software Foundation (ASF) under one
   2 // or more contributor license agreements.  See the NOTICE file
   3 // distributed with this work for additional information
   4 // regarding copyright ownership.  The ASF licenses this file
   5 // to you under the Apache License, Version 2.0 (the
   6 // "License"); you may not use this file except in compliance
   7 // with the License.  You may obtain a copy of the License at
   8 //
   9 // http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16
  17 // +build cgo
  18 // +build ccalloc
  19
  20 package memory
  21
  22 import (
  23         "testing"
  24
  25         "github.com/stretchr/testify/assert"
  26 )
  27
  28 func TestCgoArrowAllocator_Allocate(t *testing.T) {
  29         tests := []struct {
  30                 name string
  31                 sz   int
  32         }{
  33                 {"lt alignment", 33},
  34                 {"gt alignment unaligned", 65},
  35                 {"eq alignment", 64},
  36                 {"large unaligned", 4097},
  37                 {"large aligned", 8192},
  38         }
  39
  40         for _, test := range tests {
  41                 t.Run(test.name, func(t *testing.T) {
  42                         alloc := NewCgoArrowAllocator()
  43                         buf := alloc.Allocate(test.sz)
  44                         assert.NotNil(t, buf)
  45                         assert.Len(t, buf, test.sz)
  46
  47                         alloc.AssertSize(t, test.sz)
  48                         defer alloc.AssertSize(t, 0)
  49                         defer alloc.Free(buf)
  50                 })
  51         }
  52 }
  53
  54 func TestCgoArrowAllocator_Reallocate(t *testing.T) {
  55         tests := []struct {
  56                 name     string
  57                 sz1, sz2 int
  58         }{
  59                 {"smaller", 200, 100},
  60                 {"same", 200, 200},
  61                 {"larger", 200, 300},
  62         }
  63         for _, test := range tests {
  64                 t.Run(test.name, func(t *testing.T) {
  65                         alloc := NewCgoArrowAllocator()
  66                         buf := alloc.Allocate(test.sz1)
  67                         for i := range buf {
  68                                 buf[i] = byte(i & 0xFF)
  69                         }
  70
  71                         exp := make([]byte, test.sz2)
  72                         copy(exp, buf)
  73
  74                         newBuf := alloc.Reallocate(test.sz2, buf)
  75                         assert.Equal(t, exp, newBuf)
  76
  77                         alloc.AssertSize(t, test.sz2)
  78                         defer alloc.AssertSize(t, 0)
  79                         defer alloc.Free(newBuf)
  80                 })
  81         }
  82 }