From ee62abdbc40c8bd6368f5456589276c6ae6710d7 Mon Sep 17 00:00:00 2001
From: Sahil Kumar <xdsahil@gmail.com>
Date: Fri, 5 May 2023 02:21:06 +0530
Subject: [PATCH] test: add test.

Signed-off-by: xsahil03x <xdsahil@gmail.com>
---
 .../lib/src/stream_chat_persistence_client.dart     |  1 +
 .../test/stream_chat_persistence_client_test.dart   | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/packages/stream_chat_persistence/lib/src/stream_chat_persistence_client.dart b/packages/stream_chat_persistence/lib/src/stream_chat_persistence_client.dart
index b19977db..3d5a5bdb 100644
--- a/packages/stream_chat_persistence/lib/src/stream_chat_persistence_client.dart
+++ b/packages/stream_chat_persistence/lib/src/stream_chat_persistence_client.dart
@@ -90,6 +90,7 @@ class StreamChatPersistenceClient extends ChatPersistenceClient {
         'disconnect the previous instance before connecting again.',
       );
     }
+    _logger.info('connect');
     db = databaseProvider?.call(userId, _connectionMode) ??
         await _defaultDatabaseProvider(userId, _connectionMode);
   }
diff --git a/packages/stream_chat_persistence/test/stream_chat_persistence_client_test.dart b/packages/stream_chat_persistence/test/stream_chat_persistence_client_test.dart
index 8d7ab8be..c99668fd 100644
--- a/packages/stream_chat_persistence/test/stream_chat_persistence_client_test.dart
+++ b/packages/stream_chat_persistence/test/stream_chat_persistence_client_test.dart
@@ -55,6 +55,15 @@ void main() {
     expect(client.db, isNull);
   });
 
+  test('client function throws stateError if db is not yet connected', () {
+    final client = StreamChatPersistenceClient(logLevel: Level.ALL);
+    expect(
+      // Running a function that requires db connection.
+      () => client.getReplies('testParentId'),
+      throwsA(isA<StateError>()),
+    );
+  });
+
   group('client functions', () {
     const userId = 'testUserId';
     final mockDatabase = MockChatDatabase();
@@ -66,6 +75,10 @@ void main() {
       await client.connect(userId, databaseProvider: _mockDatabaseProvider);
     });
 
+    tearDown(() async {
+      await client.disconnect();
+    });
+
     test('getReplies', () async {
       const parentId = 'testParentId';
       final replies = List.generate(3, (index) => Message(id: 'testId$index'));