// Copyright © 2023 OpenIM. All rights reserved. // // 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 discoveryregister import ( "os" ) func setupTestEnvironment() { os.Setenv("ZOOKEEPER_SCHEMA", "openim") os.Setenv("ZOOKEEPER_ADDRESS", "172.28.0.1") os.Setenv("ZOOKEEPER_PORT", "12181") os.Setenv("ZOOKEEPER_USERNAME", "") os.Setenv("ZOOKEEPER_PASSWORD", "") } //func TestNewDiscoveryRegister(t *testing.T) { // setupTestEnvironment() // conf := config.NewGlobalConfig() // tests := []struct { // envType string // gatewayName string // expectedError bool // expectedResult bool // }{ // {"zookeeper", "MessageGateway", false, true}, // {"k8s", "MessageGateway", false, true}, // {"direct", "MessageGateway", false, true}, // {"invalid", "MessageGateway", true, false}, // } // // for _, test := range tests { // conf.Envs.Discovery = test.envType // conf.RpcRegisterName.OpenImMessageGatewayName = test.gatewayName // client, err := NewDiscoveryRegister(conf) // // if test.expectedError { // assert.Error(t, err) // } else { // assert.NoError(t, err) // if test.expectedResult { // assert.Implements(t, (*discovery.SvcDiscoveryRegistry)(nil), client) // } else { // assert.Nil(t, client) // } // } // } //}