Add vhost

Author: Gsantomaggio

pkg/stream/enviroment.go

@@ -262,6 +262,15 @@ func (envOptions *EnvironmentOptions) SetHost(host string) *EnvironmentOptions {
 	return envOptions
 }
 
+func (envOptions *EnvironmentOptions) SetVHost(vhost string) *EnvironmentOptions {
+	if len(envOptions.ConnectionParameters) == 0 {
+		envOptions.ConnectionParameters = append(envOptions.ConnectionParameters, &Broker{Vhost: vhost})
+	} else {
+		envOptions.ConnectionParameters[0].Vhost = vhost
+	}
+	return envOptions
+}
+
 func (envOptions *EnvironmentOptions) SetTLSConfig(config *tls.Config) *EnvironmentOptions {
 	if len(envOptions.ConnectionParameters) == 0 {
 		envOptions.ConnectionParameters = append(envOptions.ConnectionParameters, &Broker{tlsConfig: config})

pkg/stream/enviroment_test.go

@@ -188,6 +188,12 @@ var _ = Describe("Environment test", func() {
 			Expect(err).To(Equal(VirtualHostAccessFailure))
 		})
 
+
+		It("Connection Vhost not exist", func() {
+			_, err := NewEnvironment(NewEnvironmentOptions().SetVHost("VHOSTNOEXIST"))
+			Expect(err).To(Equal(VirtualHostAccessFailure))
+		})
+
 		It("Connection Vhost exists", func() {
 			_, err := NewEnvironment(NewEnvironmentOptions().
 				SetUri("rabbitmq-stream://guest:guest@localhost:5552/" + testVhost))