From 11b919b14c9bf97fd04729002252f302b3d1d000 Mon Sep 17 00:00:00 2001
From: Yunus Emre Kalkan <emre.kalkan@volosoft.com>
Date: Fri, 24 Jan 2020 09:54:11 +0300
Subject: [PATCH] fix test

---
 .../MongoDB/BookManagementMongoDbContext.cs                | 7 ++++++-
 .../MongoDB/BookManagementMongoDbContextExtensions.cs      | 6 ++++++
 .../MongoDB/BookManagementMongoDbModule.cs                 | 1 +
 .../MongoDB/IBookManagementMongoDbContext.cs               | 6 +++++-
 4 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContext.cs b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContext.cs
index 9ee4ae0379..46bd6e13b1 100644
--- a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContext.cs
+++ b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContext.cs
@@ -1,4 +1,6 @@
-using Volo.Abp.Data;
+using Acme.BookStore.BookManagement.Books;
+using MongoDB.Driver;
+using Volo.Abp.Data;
 using Volo.Abp.MongoDB;
 
 namespace Acme.BookStore.BookManagement.MongoDB
@@ -8,6 +10,8 @@ namespace Acme.BookStore.BookManagement.MongoDB
     {
         public static string CollectionPrefix { get; set; } = BookManagementConsts.DefaultDbTablePrefix;
 
+        public IMongoCollection<Book> Books => Collection<Book>();
+
         /* Add mongo collections here. Example:
          * public IMongoCollection<Question> Questions => Collection<Question>();
          */
@@ -21,5 +25,6 @@ namespace Acme.BookStore.BookManagement.MongoDB
                 options.CollectionPrefix = CollectionPrefix;
             });
         }
+
     }
 }
\ No newline at end of file
diff --git a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContextExtensions.cs b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContextExtensions.cs
index f32ad21f3a..76c4f16ff9 100644
--- a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContextExtensions.cs
+++ b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbContextExtensions.cs
@@ -1,4 +1,5 @@
 using System;
+using Acme.BookStore.BookManagement.Books;
 using Volo.Abp;
 using Volo.Abp.MongoDB;
 
@@ -15,6 +16,11 @@ namespace Acme.BookStore.BookManagement.MongoDB
             var options = new BookManagementMongoModelBuilderConfigurationOptions();
 
             optionsAction?.Invoke(options);
+
+            builder.Entity<Book>(b =>
+            {
+                b.CollectionName = options.CollectionPrefix + "Books";
+            });
         }
     }
 }
\ No newline at end of file
diff --git a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbModule.cs b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbModule.cs
index 3f01a07f55..dd1c852c01 100644
--- a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbModule.cs
+++ b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/BookManagementMongoDbModule.cs
@@ -14,6 +14,7 @@ namespace Acme.BookStore.BookManagement.MongoDB
         {
             context.Services.AddMongoDbContext<BookManagementMongoDbContext>(options =>
             {
+                options.AddDefaultRepositories();
                 /* Add custom repositories here. Example:
                  * options.AddRepository<Question, MongoQuestionRepository>();
                  */
diff --git a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/IBookManagementMongoDbContext.cs b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/IBookManagementMongoDbContext.cs
index 5df9220350..5d1c9e6525 100644
--- a/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/IBookManagementMongoDbContext.cs
+++ b/samples/BookStore-Modular/modules/book-management/src/Acme.BookStore.BookManagement.MongoDB/MongoDB/IBookManagementMongoDbContext.cs
@@ -1,4 +1,6 @@
-using Volo.Abp.Data;
+using Acme.BookStore.BookManagement.Books;
+using MongoDB.Driver;
+using Volo.Abp.Data;
 using Volo.Abp.MongoDB;
 
 namespace Acme.BookStore.BookManagement.MongoDB
@@ -6,6 +8,8 @@ namespace Acme.BookStore.BookManagement.MongoDB
     [ConnectionStringName("BookManagement")]
     public interface IBookManagementMongoDbContext : IAbpMongoDbContext
     {
+        IMongoCollection<Book> Books { get; }
+
         /* Define mongo collections here. Example:
          * IMongoCollection<Question> Questions { get; }
          */