blob: 9560f353e12cc8ea719a36351eb6cb91cd27fc54 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using Timeline.Entities;
using Timeline.Models;
using Timeline.Services;
namespace Timeline.Tests.Helpers
{
public class TestDatabase : IDisposable
{
// currently password service is thread safe, so we share a static one.
private static PasswordService PasswordService { get; } = new PasswordService();
private static User CreateEntityFromMock(MockUser user)
{
return new User
{
Name = user.Username,
EncryptedPassword = PasswordService.HashPassword(user.Password),
RoleString = UserRoleConvert.ToString(user.Administrator),
Avatar = null
};
}
private static IEnumerable<User> CreateDefaultMockEntities()
{
// emmmmmmm. Never reuse the user instances because EF Core uses them, which will cause strange things.
yield return CreateEntityFromMock(MockUser.User);
yield return CreateEntityFromMock(MockUser.Admin);
}
private static void InitDatabase(DatabaseContext context)
{
context.Database.EnsureCreated();
context.Users.AddRange(CreateDefaultMockEntities());
context.SaveChanges();
}
public SqliteConnection Connection { get; }
public DatabaseContext Context { get; }
public TestDatabase()
{
Connection = new SqliteConnection("Data Source=:memory:;");
Connection.Open();
var options = new DbContextOptionsBuilder<DevelopmentDatabaseContext>()
.UseSqlite(Connection)
.Options;
Context = new DevelopmentDatabaseContext(options);
InitDatabase(Context);
}
private List<MockUser> _extraMockUsers;
public IReadOnlyList<MockUser> ExtraMockUsers => _extraMockUsers;
public void CreateExtraMockUsers(int count)
{
if (count <= 0)
throw new ArgumentOutOfRangeException(nameof(count), count, "Additional user count must be bigger than 0.");
if (_extraMockUsers != null)
throw new InvalidOperationException("Already create mock users.");
_extraMockUsers = new List<MockUser>();
for (int i = 0; i < count; i++)
{
_extraMockUsers.Add(new MockUser($"user{i}", $"password", false));
}
Context.AddRange(_extraMockUsers.Select(u => CreateEntityFromMock(u)));
Context.SaveChanges();
}
public void Dispose()
{
Context.Dispose();
Connection.Close();
Connection.Dispose();
}
}
}
|
