mirror of https://github.com/ErsatzTV/ErsatzTV.git
Jason Dove
6 months ago
committed by
GitHub
33 changed files with 14167 additions and 74 deletions
@ -0,0 +1,34 @@
@@ -0,0 +1,34 @@
|
||||
using ErsatzTV.Core.Interfaces.Scheduling; |
||||
using ErsatzTV.Core.Scheduling; |
||||
using NSubstitute; |
||||
using NUnit.Framework; |
||||
using Shouldly; |
||||
|
||||
namespace ErsatzTV.Core.Tests.Scheduling; |
||||
|
||||
[TestFixture] |
||||
public class CountExpressionTests |
||||
{ |
||||
[Test] |
||||
[TestCase("2", 2)] |
||||
[TestCase("count", 10)] |
||||
[TestCase("count / 2", 5)] |
||||
[TestCase("count * 2", 20)] |
||||
[TestCase("count + 1", 11)] |
||||
[TestCase("count - 1", 9)] |
||||
[TestCase("random % 4 + 1", 3)] |
||||
[TestCase("invalid", 0)] |
||||
[TestCase("count / 0", 0)] |
||||
public void Should_Evaluate_Expression(string expression, int expected) |
||||
{ |
||||
var enumerator = Substitute.For<IMediaCollectionEnumerator>(); |
||||
enumerator.Count.Returns(10); |
||||
|
||||
var random = Substitute.For<Random>(); |
||||
random.Next().Returns(2); |
||||
|
||||
int result = CountExpression.Evaluate(expression, enumerator, random, CancellationToken.None); |
||||
|
||||
result.ShouldBe(expected); |
||||
} |
||||
} |
||||
@ -0,0 +1,47 @@
@@ -0,0 +1,47 @@
|
||||
using ErsatzTV.Core.Interfaces.Scheduling; |
||||
using NCalc; |
||||
|
||||
namespace ErsatzTV.Core.Scheduling; |
||||
|
||||
public static class CountExpression |
||||
{ |
||||
public static int Evaluate( |
||||
string countExpression, |
||||
IMediaCollectionEnumerator enumerator, |
||||
Random random, |
||||
CancellationToken cancellationToken) |
||||
{ |
||||
int enumeratorCount = enumerator is PlaylistEnumerator playlistEnumerator |
||||
? playlistEnumerator.CountForRandom |
||||
: enumerator.Count; |
||||
var expression = new Expression(countExpression); |
||||
expression.EvaluateParameter += (name, e) => |
||||
{ |
||||
e.Result = name switch |
||||
{ |
||||
"count" => enumeratorCount, |
||||
"random" => enumeratorCount > 0 ? random.Next() % enumeratorCount : 0, |
||||
_ => e.Result |
||||
}; |
||||
}; |
||||
|
||||
object expressionResult = 0; |
||||
try |
||||
{ |
||||
expressionResult = expression.Evaluate(cancellationToken); |
||||
} |
||||
catch (Exception) |
||||
{ |
||||
// do nothing
|
||||
} |
||||
|
||||
return expressionResult switch |
||||
{ |
||||
double d when double.IsInfinity(d) || double.IsNaN(d) => 0, |
||||
double doubleResult => (int)Math.Floor(doubleResult), |
||||
int intResult => intResult, |
||||
long longResult => (int)longResult, |
||||
_ => 0 |
||||
}; |
||||
} |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,38 @@
@@ -0,0 +1,38 @@
|
||||
using Microsoft.EntityFrameworkCore.Migrations; |
||||
|
||||
#nullable disable |
||||
|
||||
namespace ErsatzTV.Infrastructure.MySql.Migrations |
||||
{ |
||||
/// <inheritdoc />
|
||||
public partial class Change_ProgramScheduleItemMultipleCountString : Migration |
||||
{ |
||||
/// <inheritdoc />
|
||||
protected override void Up(MigrationBuilder migrationBuilder) |
||||
{ |
||||
migrationBuilder.AlterColumn<string>( |
||||
name: "Count", |
||||
table: "ProgramScheduleMultipleItem", |
||||
type: "longtext", |
||||
nullable: true, |
||||
oldClrType: typeof(int), |
||||
oldType: "int") |
||||
.Annotation("MySql:CharSet", "utf8mb4"); |
||||
} |
||||
|
||||
/// <inheritdoc />
|
||||
protected override void Down(MigrationBuilder migrationBuilder) |
||||
{ |
||||
migrationBuilder.AlterColumn<int>( |
||||
name: "Count", |
||||
table: "ProgramScheduleMultipleItem", |
||||
type: "int", |
||||
nullable: false, |
||||
defaultValue: 0, |
||||
oldClrType: typeof(string), |
||||
oldType: "longtext", |
||||
oldNullable: true) |
||||
.OldAnnotation("MySql:CharSet", "utf8mb4"); |
||||
} |
||||
} |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,36 @@
@@ -0,0 +1,36 @@
|
||||
using Microsoft.EntityFrameworkCore.Migrations; |
||||
|
||||
#nullable disable |
||||
|
||||
namespace ErsatzTV.Infrastructure.Sqlite.Migrations |
||||
{ |
||||
/// <inheritdoc />
|
||||
public partial class Change_ProgramScheduleItemMultipleCountString : Migration |
||||
{ |
||||
/// <inheritdoc />
|
||||
protected override void Up(MigrationBuilder migrationBuilder) |
||||
{ |
||||
migrationBuilder.AlterColumn<string>( |
||||
name: "Count", |
||||
table: "ProgramScheduleMultipleItem", |
||||
type: "TEXT", |
||||
nullable: true, |
||||
oldClrType: typeof(int), |
||||
oldType: "INTEGER"); |
||||
} |
||||
|
||||
/// <inheritdoc />
|
||||
protected override void Down(MigrationBuilder migrationBuilder) |
||||
{ |
||||
migrationBuilder.AlterColumn<int>( |
||||
name: "Count", |
||||
table: "ProgramScheduleMultipleItem", |
||||
type: "INTEGER", |
||||
nullable: false, |
||||
defaultValue: 0, |
||||
oldClrType: typeof(string), |
||||
oldType: "TEXT", |
||||
oldNullable: true); |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue