157 lines
6.0 KiB
C#
157 lines
6.0 KiB
C#
using System.Collections.Generic;
|
|
using System.Data.Entity;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text.Json;
|
|
using System.Threading;
|
|
|
|
using ASC.Common.Security.Authentication;
|
|
using ASC.Core;
|
|
using ASC.Core.Common.EF;
|
|
using ASC.Core.Common.EF.Context;
|
|
using ASC.Core.Tenants;
|
|
using ASC.Core.Users;
|
|
using ASC.Files.Helpers;
|
|
using ASC.Files.Model;
|
|
using ASC.Files.Tests.Infrastructure;
|
|
using ASC.Web.Files.Classes;
|
|
using ASC.Web.Files.Services.WCFService;
|
|
using ASC.Web.Files.Services.WCFService.FileOperations;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.AspNetCore.TestHost;
|
|
using Microsoft.EntityFrameworkCore.Infrastructure;
|
|
using Microsoft.EntityFrameworkCore.Migrations;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Options;
|
|
|
|
namespace ASC.Files.Tests
|
|
{
|
|
public class BaseFilesTests
|
|
{
|
|
protected FilesControllerHelper<int> FilesControllerHelper { get; set; }
|
|
protected TestServer TestServer { get; set; }
|
|
|
|
public BaseDbContext baseDbContext;
|
|
|
|
|
|
protected GlobalFolderHelper GlobalFolderHelper { get; set; }
|
|
protected FileStorageService<int> FileStorageService { get; set; }
|
|
protected UserManager UserManager { get; set; }
|
|
protected Tenant CurrentTenant { get; set; }
|
|
protected UserInfo User { get; set; }
|
|
protected SecurityContext SecurityContext { get; set; }
|
|
protected UserOptions UserOptions { get; set; }
|
|
protected IAccount Account { get; set; }
|
|
protected IConfiguration Configuration { get; set; }
|
|
protected IServiceScope scope { get; set; }
|
|
|
|
const string con = "Server=localhost;Database=onlyoffice_test;User ID = root; Password=root;Pooling=true;";
|
|
public virtual void SetUp()
|
|
{
|
|
var scope1 = Program.CreateHostBuilder(new string[] { "--pathToConf" ,"..\\..\\..\\..\\..\\..\\config", "--ConnectionStrings:default:connectionString", con, "--migration:enabled", "true" }).Build();
|
|
|
|
scope = scope1.Services.CreateScope();
|
|
var TenantDbContext = scope.ServiceProvider.GetService<DbContextManager<TenantDbContext>>();
|
|
using (var db = TenantDbContext)
|
|
{
|
|
var migrator = db.Value.GetInfrastructure().GetRequiredService<IMigrator>();
|
|
migrator.Migrate("20201222152946_TestUser");
|
|
}
|
|
var tenantManager = scope.ServiceProvider.GetService<TenantManager>();
|
|
var tenant = tenantManager.GetTenant(1);
|
|
tenantManager.SetCurrentTenant(tenant);
|
|
CurrentTenant = tenant;
|
|
|
|
FilesControllerHelper = scope.ServiceProvider.GetService<FilesControllerHelper<int>>();
|
|
GlobalFolderHelper = scope.ServiceProvider.GetService<GlobalFolderHelper>();
|
|
UserManager = scope.ServiceProvider.GetService<UserManager>();
|
|
SecurityContext = scope.ServiceProvider.GetService<SecurityContext>();
|
|
UserOptions = scope.ServiceProvider.GetService<IOptions<UserOptions>>().Value;
|
|
FileStorageService = scope.ServiceProvider.GetService<FileStorageService<int>>();
|
|
|
|
|
|
SecurityContext.AuthenticateMe(CurrentTenant.OwnerId);
|
|
|
|
}
|
|
|
|
|
|
public void DeleteFolder(int folder, bool deleteAfter, bool DeleteEmmediatly)
|
|
{
|
|
FilesControllerHelper.DeleteFolder(folder, deleteAfter, DeleteEmmediatly);
|
|
while (true)
|
|
{
|
|
var statuses = FileStorageService.GetTasksStatuses();
|
|
|
|
if (statuses.TrueForAll(r => r.Finished))
|
|
break;
|
|
Thread.Sleep(100);
|
|
}
|
|
}
|
|
public void DeleteFile(int file, bool deleteAfter, bool DeleteEmmediatly)
|
|
{
|
|
FilesControllerHelper.DeleteFile(file, deleteAfter, DeleteEmmediatly);
|
|
while (true)
|
|
{
|
|
var statuses = FileStorageService.GetTasksStatuses();
|
|
|
|
if (statuses.TrueForAll(r => r.Finished))
|
|
break;
|
|
Thread.Sleep(100);
|
|
}
|
|
}
|
|
private void Configure(WebHostBuilderContext hostingContext, IConfigurationBuilder config)
|
|
{
|
|
var path = "..\\..\\..\\..\\..\\..\\config";
|
|
|
|
|
|
if (!Path.IsPathRooted(path))
|
|
{
|
|
path = Path.GetFullPath(Path.Combine(hostingContext.HostingEnvironment.ContentRootPath, path));
|
|
}
|
|
|
|
config.SetBasePath(path);
|
|
config
|
|
.AddInMemoryCollection(new Dictionary<string, string>
|
|
{
|
|
{"pathToConf", path}
|
|
})
|
|
.AddJsonFile("appsettings.json")
|
|
.AddJsonFile($"appsettings.{hostingContext.HostingEnvironment.EnvironmentName}.json", true)
|
|
.AddJsonFile("storage.json")
|
|
.AddJsonFile("kafka.json")
|
|
.AddJsonFile($"kafka.{hostingContext.HostingEnvironment.EnvironmentName}.json", true)
|
|
.AddEnvironmentVariables();
|
|
|
|
Configuration = config.Build();
|
|
}
|
|
|
|
public BatchModel GetBatchModel(string text)
|
|
{
|
|
var json = text;
|
|
|
|
var jsonDocument = JsonDocument.Parse(json);
|
|
var root = jsonDocument.RootElement;
|
|
var folderIds = root[0].GetProperty("folderIds").EnumerateArray().ToList();
|
|
var fileIds = root[1].GetProperty("fileIds").EnumerateArray().ToList();
|
|
var destFolderdId = root[2];
|
|
|
|
var batchModel = new BatchModel
|
|
{
|
|
FolderIds = folderIds,
|
|
FileIds = fileIds,
|
|
DestFolderId = destFolderdId,
|
|
DeleteAfter = false,
|
|
ConflictResolveType = FileConflictResolveType.Overwrite
|
|
};
|
|
|
|
return batchModel;
|
|
}
|
|
|
|
public virtual void TearDown()
|
|
{
|
|
Database.Delete(con);
|
|
}
|
|
}
|
|
}
|