int64_test.go

/*
 * TencentBlueKing is pleased to support the open source community by making
 * 蓝鲸智云-gopkg available.
 * Copyright (C) 2017-2022 THL A29 Limited, a Tencent company. All rights reserved.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at http://opensource.org/licenses/MIT
 * 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 set_test

import (
	. "github.com/onsi/ginkgo/v2"
	"github.com/stretchr/testify/assert"

	"github.com/TencentBlueKing/gopkg/collection/set"
)

var _ = Describe("Int64", func() {
	var s *set.Int64Set

	BeforeEach(func() {
		s = set.NewInt64Set()
	})

	It("NewInt64Set", func() {
		// s := util.NewInt64Set()
		assert.Len(GinkgoT(), s.Data, 0)
		assert.Equal(GinkgoT(), 0, s.Size())
	})

	It("NewInt64SetWithValues", func() {
		s1 := set.NewInt64SetWithValues([]int64{123, 456})

		assert.Len(GinkgoT(), s1.Data, 2)
		assert.Equal(GinkgoT(), 2, s1.Size())

		assert.True(GinkgoT(), s1.Has(123))
	})

	It("NewFixedLengthInt64Set", func() {
		s1 := set.NewFixedLengthInt64Set(2)

		assert.Len(GinkgoT(), s1.Data, 0)
		assert.Equal(GinkgoT(), 0, s1.Size())
	})

	It("Add one, check size", func() {
		s.Add(123)

		assert.Len(GinkgoT(), s.Data, 1)
		assert.Equal(GinkgoT(), 1, s.Size())
	})

	It("Append", func() {
		s.Append([]int64{123, 456}...)
		s.Append([]int64{456, 789}...)

		assert.Len(GinkgoT(), s.Data, 3)
		assert.Equal(GinkgoT(), 3, s.Size())

		assert.True(GinkgoT(), s.Has(int64(123)))
		assert.True(GinkgoT(), s.Has(int64(456)))
		assert.True(GinkgoT(), s.Has(int64(789)))
	})

	It("Has 123", func() {
		assert.False(GinkgoT(), s.Has(123))
		s.Add(123)
		assert.True(GinkgoT(), s.Has(123))
	})

	It("ToSlice", func() {
		s.Add(123)
		sli1 := s.ToSlice()
		assert.Len(GinkgoT(), sli1, 1)

		s.Add(456)

		sli2 := s.ToSlice()
		assert.Len(GinkgoT(), sli2, 2)
	})
})