diff --git a/src/Database.Tests/ModelTests.cs b/src/Database.Tests/ModelTests.cs
index c271ca85f7cfc2fd0bea4b725f5d7a960e729d37..ab7e65638701573b8c16faa9b6cc11f4a6fd3b83 100644
--- a/src/Database.Tests/ModelTests.cs
+++ b/src/Database.Tests/ModelTests.cs
@@ -1,5 +1,7 @@
-using NUnit.Framework;
+using Coscine.Database.Models;
+using NUnit.Framework;
using System;
+using System.Collections.Generic;
namespace Coscine.Database.Tests
{
diff --git a/src/Database/Models/ProjectModel.cs b/src/Database/Models/ProjectModel.cs
index 707e6d772368130d1c40d77c4f959c46bdfa585b..8ead1a9d66898e3a851a4344606274794b593821 100644
--- a/src/Database/Models/ProjectModel.cs
+++ b/src/Database/Models/ProjectModel.cs
@@ -131,7 +131,7 @@ namespace Coscine.Database.Models
});
}
- public Project StoreFromObject(ProjectObject projectObject, User user, bool isRWTHMember = false)
+ public Project StoreFromObject(ProjectObject projectObject, User user, IDictionary<string, int> defaultQuota)
{
if (!projectObject.Disciplines.Any() || !projectObject.Organizations.Any())
{
@@ -158,7 +158,7 @@ namespace Coscine.Database.Models
{
SetDisciplines(project, projectObject.Disciplines);
SetOrganizations(project, projectObject.Organizations);
- SetQuotas(project, isRWTHMember);
+ SetQuotas(project, defaultQuota);
}
catch (Exception)
{
@@ -252,22 +252,17 @@ namespace Coscine.Database.Models
}
}
- private void SetQuotas(Project project, bool IsRWTHMember = false)
+ public void SetQuotas(Project project, IDictionary<string, int> defaultQuotas)
{
ProjectQuotaModel projectQuotaModel = new ProjectQuotaModel();
ResourceTypeModel resourceTypeModel = new ResourceTypeModel();
foreach (var resourceType in resourceTypeModel.GetAll())
{
- var quota = 25;
- if ((resourceType.DisplayName.Equals("rds") && !IsRWTHMember) || resourceType.DisplayName.Equals("rdss3"))
- {
- quota = 0;
- }
projectQuotaModel.Insert(new ProjectQuota
{
ProjectId = project.Id,
ResourceTypeId = resourceType.Id,
- Quota = quota
+ Quota = defaultQuotas != null && defaultQuotas.ContainsKey(resourceType.DisplayName) ? defaultQuotas[resourceType.DisplayName] : 0
});
}
}