From 8e45a42f18472dd5b90ac52d53f42fb75c412724 Mon Sep 17 00:00:00 2001 From: sylwiaszunejko Date: Thu, 5 Sep 2024 15:21:41 +0200 Subject: [PATCH] Add extra test for AwaitSchemaAgreement --- integration_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/integration_test.go b/integration_test.go index f548a829f..d87bfeacb 100644 --- a/integration_test.go +++ b/integration_test.go @@ -233,3 +233,13 @@ func TestSessionAwaitSchemaAgreement(t *testing.T) { t.Fatalf("expected session.AwaitSchemaAgreement to not return an error but got '%v'", err) } } + +func TestSessionAwaitSchemaAgreementSessionClosed(t *testing.T) { + session := createSession(t) + session.Close() + + if err := session.AwaitSchemaAgreement(context.Background()); err != ErrConnectionClosed { + t.Fatalf("expected session.AwaitSchemaAgreement to return ErrConnectionClosed but got '%v'", err) + } + +}