58 lines
2.8 KiB
C#
58 lines
2.8 KiB
C#
using Microsoft.Data.Sqlite;
|
|
using VoiceCat.Server.Data;
|
|
|
|
namespace VoiceCat.Tests;
|
|
|
|
public sealed class AccountStoreTests
|
|
{
|
|
[Fact]
|
|
public void UnsupportedSchemaIsRejectedWithoutCreatingAccountTables()
|
|
{
|
|
string path = Path.Combine(Path.GetTempPath(), "voicecat-future-" + Guid.NewGuid().ToString("N") + ".db");
|
|
try
|
|
{
|
|
SQLitePCL.Batteries_V2.Init();
|
|
using var connection = new SqliteConnection(new SqliteConnectionStringBuilder { DataSource = path, Pooling = false }.ToString());
|
|
connection.Open();
|
|
using var command = connection.CreateCommand();
|
|
command.CommandText = "CREATE TABLE server_meta (key TEXT PRIMARY KEY,value TEXT NOT NULL); INSERT INTO server_meta VALUES ('schema_version','99');";
|
|
command.ExecuteNonQuery();
|
|
Assert.Throws<InvalidDataException>(() => new AccountStore(path));
|
|
command.CommandText = "SELECT COUNT(*) FROM sqlite_master WHERE name='accounts'";
|
|
Assert.Equal(0L, command.ExecuteScalar());
|
|
command.CommandText = "SELECT value FROM server_meta WHERE key='schema_version'";
|
|
Assert.Equal("99", command.ExecuteScalar());
|
|
}
|
|
finally { File.Delete(path); File.Delete(path + "-wal"); File.Delete(path + "-shm"); }
|
|
}
|
|
|
|
[Fact]
|
|
public async Task AccountsSurviveRestartAndFailedAuthDoesNotChangeLastLogin()
|
|
{
|
|
string directory = Path.Combine(Path.GetTempPath(), "voicecat-db-" + Guid.NewGuid().ToString("N"));
|
|
Directory.CreateDirectory(directory);
|
|
string path = Path.Combine(directory, "voicecat.db");
|
|
try
|
|
{
|
|
Account account;
|
|
using (var store = new AccountStore(path)) account = await store.CreateAccountAsync("admin'", "secret", true);
|
|
using (var store = new AccountStore(path))
|
|
{
|
|
Assert.Null(await store.AuthenticateAsync("admin'", "wrong"));
|
|
Assert.Null(await store.AuthenticateAsync("missing", "secret"));
|
|
using var connection = new SqliteConnection(new SqliteConnectionStringBuilder { DataSource = path, Pooling = false }.ToString());
|
|
connection.Open();
|
|
using var command = connection.CreateCommand();
|
|
command.CommandText = "SELECT last_login FROM accounts WHERE id=$id";
|
|
command.Parameters.AddWithValue("$id", account.Id);
|
|
Assert.Equal(0L, command.ExecuteScalar());
|
|
Account authenticated = Assert.IsType<Account>(await store.AuthenticateAsync("admin'", "secret"));
|
|
Assert.Equal(account.Id, authenticated.Id);
|
|
Assert.True(authenticated.IsAdmin);
|
|
Assert.True(authenticated.LastLogin > 0);
|
|
}
|
|
}
|
|
finally { Directory.Delete(directory, true); }
|
|
}
|
|
}
|