save_everything

Chatter/Repository/MessageRepository.cs

@@ -20,24 +20,19 @@ namespace Chatter.Repository
     {
         private readonly IDbConnector _connector;
 
-        public MessageRepository()
+        public MessageRepository(IDbConnector connector)
         {
+            _connector = connector;
         }
 
         public void AddMessage(MessageDto msg)
         {
-            using (var connector = new SqLiteConnector())
-            {
-                connector.AddMessage(msg);
-            }
+            _connector.AddMessage(msg);
         }
 
         public IEnumerable<MessageDto> GetMessages()
         {
-            using (var connector = new SqLiteConnector())
-            {
-                return connector.GetMessages();
-            }
+            return _connector.GetMessages();
         }
     }
 }