Add tests

inputfalken · Jun 15, 2024 · 8eaf26a · 8eaf26a
1 parent 45802e6
commit 8eaf26a
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/...s/DynamoDBGenerator.SourceGenerator.Tests/DynamoDBDocumentTests/PrimitiveArgumentTests.cs b/...s/DynamoDBGenerator.SourceGenerator.Tests/DynamoDBDocumentTests/PrimitiveArgumentTests.cs
@@ -8,6 +8,36 @@ namespace DynamoDBGenerator.SourceGenerator.Tests.DynamoDBDocumentTests;
 [DynamoDBMarshaller(EntityType = typeof((Guid, Guid)), AccessName = "INT", ArgumentType = typeof(int))]
 public partial class PrimitiveArgumentTests
 {
+    [Fact]
+    public void UnmarshallArgument_GUID_ShouldThrowInvalidOperationException()
+    {
+        var act = () => GUID.UnmarshallArgument(new Dictionary<string, AttributeValue>());
+
+        act.Should()
+          .Throw<InvalidOperationException>()
+          .WithMessage("Could not unmarshall the type 'Guid'*");
+    }
+
+    [Fact]
+    public void UnmarshallArgument_INT_ShouldThrowInvalidOperationException()
+    {
+        var act = () => INT.UnmarshallArgument(new Dictionary<string, AttributeValue>());
+
+        act.Should()
+          .Throw<InvalidOperationException>()
+          .WithMessage("Could not unmarshall the type 'Int32'*");
+    }
+
+    [Fact]
+    public void UnmarshallArgument_STRING_ShouldThrowInvalidOperationException()
+    {
+        var act = () => STRING.UnmarshallArgument(new Dictionary<string, AttributeValue>());
+
+        act.Should()
+          .Throw<InvalidOperationException>()
+          .WithMessage("Could not unmarshall the type 'String'*");
+    }
+
     [Fact]
     public void ExpressionValueTracker_STRING_ShouldBeExpandedCorrectly()
     {