@@ -13,7 +13,7 @@ import (
13
13
)
14
14
15
15
func TestPoolInitialization (t * testing.T ) {
16
- urls := []string {"ws://127.0.0.1:9944" }
16
+ urls := []string {getUrlBasedOnEnv () }
17
17
mgr := NewManager (urls ... )
18
18
defer mgr .Close ()
19
19
@@ -28,7 +28,8 @@ func TestPoolInitialization(t *testing.T) {
28
28
}
29
29
30
30
func TestConnectionReuse (t * testing.T ) {
31
- mgr := NewManager ("ws://127.0.0.1:9944" )
31
+ urls := []string {getUrlBasedOnEnv ()}
32
+ mgr := NewManager (urls ... )
32
33
defer mgr .Close ()
33
34
34
35
// Wait for pool initialization
@@ -59,7 +60,8 @@ func TestConnectionReuse(t *testing.T) {
59
60
}
60
61
61
62
func TestConcurrentAccess (t * testing.T ) {
62
- mgr := NewManager ("ws://127.0.0.1:9944" )
63
+ urls := []string {getUrlBasedOnEnv ()}
64
+ mgr := NewManager (urls ... )
63
65
defer mgr .Close ()
64
66
65
67
ctx , cancel := context .WithTimeout (context .Background (), 5 * time .Second )
@@ -87,7 +89,8 @@ func TestConcurrentAccess(t *testing.T) {
87
89
}
88
90
89
91
func TestFailover (t * testing.T ) {
90
- mgr := NewManager ("ws://fail1" , "ws://127.0.0.1:9944" )
92
+ urls := []string {"ws://fail1" , getUrlBasedOnEnv ()}
93
+ mgr := NewManager (urls ... )
91
94
defer mgr .Close ()
92
95
93
96
sub1 , err := mgr .GetConnection (context .Background ())
@@ -96,12 +99,13 @@ func TestFailover(t *testing.T) {
96
99
sub2 , err := mgr .GetConnection (context .Background ())
97
100
require .NoError (t , err )
98
101
defer sub2 .Release ()
99
- assert .Equal (t , sub1 .conn .url , "ws://127.0.0.1:9944" )
100
- assert .Equal (t , sub2 .conn .url , "ws://127.0.0.1:9944" )
102
+ assert .Equal (t , sub1 .conn .url , urls [ 1 ] )
103
+ assert .Equal (t , sub2 .conn .url , urls [ 1 ] )
101
104
}
102
105
103
106
func TestHealthChecking (t * testing.T ) {
104
- mgr := NewManager ("ws://127.0.0.1:9944" )
107
+ urls := []string {getUrlBasedOnEnv ()}
108
+ mgr := NewManager (urls ... )
105
109
defer mgr .Close ()
106
110
107
111
sub , err := mgr .GetConnection (context .Background ())
@@ -131,7 +135,8 @@ func TestStressWithFailures(t *testing.T) {
131
135
ConnectionTimeout : time .Second ,
132
136
}
133
137
134
- mgr := NewManagerWithConfig (config , "ws://127.0.0.1:9944" )
138
+ urls := []string {getUrlBasedOnEnv ()}
139
+ mgr := NewManagerWithConfig (config , urls ... )
135
140
defer mgr .Close ()
136
141
137
142
ctx , cancel := context .WithTimeout (context .Background (), 3 * time .Second )
0 commit comments