@@ -30,13 +30,26 @@ func TestRun(t *testing.T) {
3030 require .Contains (t , string (output ), "default" )
3131}
3232
33- func TestRun_PutGet (t * testing.T ) {
34- ctx := context .Background ()
33+ func TestPutGet (t * testing.T ) {
34+ t .Run ("single_node" , func (t * testing.T ) {
35+ ctr , err := etcd .Run (context .Background (), "gcr.io/etcd-development/etcd:v3.5.14" )
36+ testPutGet (t , ctr , err )
37+ })
38+ t .Run ("multiple_nodes" , func (t * testing.T ) {
39+ ctr , err := etcd .Run (context .Background (), "gcr.io/etcd-development/etcd:v3.5.14" , etcd .WithNodes ("etcd-1" , "etcd-2" , "etcd-3" ))
40+ testPutGet (t , ctr , err )
41+ })
42+ }
43+
44+ func testPutGet (t * testing.T , ctr * etcd.EtcdContainer , err error ) {
45+ t .Helper ()
3546
36- ctr , err := etcd .Run (ctx , "gcr.io/etcd-development/etcd:v3.5.14" , etcd .WithNodes ("etcd-1" , "etcd-2" , "etcd-3" ))
3747 testcontainers .CleanupContainer (t , ctr )
48+
3849 require .NoError (t , err )
3950
51+ ctx := context .Background ()
52+
4053 clientEndpoints , err := ctr .ClientEndpoints (ctx )
4154 require .NoError (t , err )
4255
@@ -45,7 +58,9 @@ func TestRun_PutGet(t *testing.T) {
4558 DialTimeout : 5 * time .Second ,
4659 })
4760 require .NoError (t , err )
48- defer cli .Close ()
61+ defer func () {
62+ require .NoError (t , cli .Close ())
63+ }()
4964
5065 ctx , cancel := context .WithTimeout (ctx , 2 * time .Second )
5166 defer cancel ()
0 commit comments