Add extra test for AwaitSchemaAgreement

scylladb · Sep 5, 2024 · 8e45a42 · 8e45a42
1 parent d31335c
commit 8e45a42
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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)
+	}
+
+}