132 строки
3.9 KiB
Go
132 строки
3.9 KiB
Go
package servicebus
|
|
|
|
// MIT License
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
// copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Azure/azure-service-bus-go/internal/test"
|
|
"github.com/Azure/go-autorest/autorest/azure"
|
|
"github.com/joho/godotenv"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type (
|
|
serviceBusSuite struct {
|
|
test.BaseSuite
|
|
}
|
|
)
|
|
|
|
func TestSB(t *testing.T) {
|
|
suite.Run(t, new(serviceBusSuite))
|
|
}
|
|
|
|
func (suite *serviceBusSuite) TestCreateNamespaceFromConnectionString() {
|
|
_ = godotenv.Load()
|
|
|
|
connStr := os.Getenv("SERVICEBUS_CONNECTION_STRING") // `Endpoint=sb://XXXX.servicebus.windows.net/;SharedAccessKeyName=XXXX;SharedAccessKey=XXXX`
|
|
if connStr == "" {
|
|
suite.FailNow("environment variable SERVICEBUS_CONNECTION_STRING was not set")
|
|
}
|
|
|
|
ns, err := NewNamespace(NamespaceWithConnectionString(connStr))
|
|
if suite.NoError(err) {
|
|
suite.Contains(connStr, ns.Name)
|
|
}
|
|
}
|
|
|
|
func TestNewNamespaceWithAzureEnvironment(t *testing.T) {
|
|
ns, err := NewNamespace(NamespaceWithAzureEnvironment("namespaceName", "AzureGermanCloud"))
|
|
if err != nil {
|
|
t.Fatalf("unexpected error creating namespace: %s", err)
|
|
}
|
|
if ns.Environment != azure.GermanCloud {
|
|
t.Fatalf("expected namespace environment to be %q but was %q", azure.GermanCloud, ns.Environment)
|
|
}
|
|
if !strings.EqualFold(ns.Suffix, azure.GermanCloud.ServiceBusEndpointSuffix) {
|
|
t.Fatalf("expected suffix to be %q but was %q", azure.GermanCloud.ServiceBusEndpointSuffix, ns.Suffix)
|
|
}
|
|
if ns.Name != "namespaceName" {
|
|
t.Fatalf("expected namespace name to be %q but was %q", "namespaceName", ns.Name)
|
|
}
|
|
}
|
|
|
|
// TearDownSuite destroys created resources during the run of the suite
|
|
func (suite *serviceBusSuite) TearDownSuite() {
|
|
suite.BaseSuite.TearDownSuite()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
defer cancel()
|
|
suite.deleteAllTaggedQueues(ctx)
|
|
suite.deleteAllTaggedTopics(ctx)
|
|
}
|
|
|
|
func (suite *serviceBusSuite) deleteAllTaggedQueues(ctx context.Context) {
|
|
ns := suite.getNewSasInstance()
|
|
qm := ns.NewQueueManager()
|
|
|
|
qs, err := qm.List(ctx)
|
|
if err != nil {
|
|
suite.T().Fatal(err)
|
|
}
|
|
|
|
for _, q := range qs {
|
|
if strings.HasSuffix(q.Name, suite.TagID) {
|
|
err := qm.Delete(ctx, q.Name)
|
|
if err != nil {
|
|
suite.T().Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (suite *serviceBusSuite) deleteAllTaggedTopics(ctx context.Context) {
|
|
ns := suite.getNewSasInstance()
|
|
tm := ns.NewTopicManager()
|
|
|
|
topics, err := tm.List(ctx)
|
|
if err != nil {
|
|
suite.T().Fatal(err)
|
|
}
|
|
|
|
for _, topic := range topics {
|
|
if strings.HasSuffix(topic.Name, suite.TagID) {
|
|
err := tm.Delete(ctx, topic.Name)
|
|
if err != nil {
|
|
suite.T().Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (suite *serviceBusSuite) getNewSasInstance(opts ...NamespaceOption) *Namespace {
|
|
ns, err := NewNamespace(append(opts, NamespaceWithConnectionString(suite.ConnStr))...)
|
|
if err != nil {
|
|
suite.T().Fatal(err)
|
|
}
|
|
return ns
|
|
}
|