From 8ed547204c85d877d39a15e4858b2a6b04091aa3 Mon Sep 17 00:00:00 2001
From: ds5678 <49847914+ds5678@users.noreply.github.com>
Date: Mon, 24 Feb 2025 17:05:40 -0800
Subject: [PATCH] Add TpkTests

---
 Source/AssetRipper.Tests/TpkTests.cs          | 25 +++++++++++++++++++
 .../TypeTreeNodeStructTests.cs                | 18 ++++++-------
 2 files changed, 34 insertions(+), 9 deletions(-)
 create mode 100644 Source/AssetRipper.Tests/TpkTests.cs

diff --git a/Source/AssetRipper.Tests/TpkTests.cs b/Source/AssetRipper.Tests/TpkTests.cs
new file mode 100644
index 000000000..41686bb47
--- /dev/null
+++ b/Source/AssetRipper.Tests/TpkTests.cs
@@ -0,0 +1,25 @@
+using AssetRipper.SourceGenerated;
+using AssetRipper.Tpk;
+using AssetRipper.Tpk.EngineAssets;
+using AssetRipper.Tpk.TypeTrees;
+
+namespace AssetRipper.Tests;
+
+internal class TpkTests
+{
+	[Test]
+	public void EngineAssetsTpkSuccessfullyExtracts()
+	{
+		TpkFile tpk = TpkFile.FromStream(EngineAssetsTpk.GetStream());
+		TpkEngineAssetsBlob blob = (TpkEngineAssetsBlob)tpk.GetDataBlob();
+		Assert.That(blob.Data, Is.Not.Empty);
+	}
+
+	[Test]
+	public void TypeTreeTpkSuccessfullyExtracts()
+	{
+		TpkFile tpk = TpkFile.FromStream(SourceTpk.GetStream());
+		TpkTypeTreeBlob blob = (TpkTypeTreeBlob)tpk.GetDataBlob();
+		Assert.That(blob.ClassInformation, Is.Not.Empty);
+	}
+}
diff --git a/Source/AssetRipper.Tests/TypeTreeNodeStructTests.cs b/Source/AssetRipper.Tests/TypeTreeNodeStructTests.cs
index 9cd19fde0..41127b72c 100644
--- a/Source/AssetRipper.Tests/TypeTreeNodeStructTests.cs
+++ b/Source/AssetRipper.Tests/TypeTreeNodeStructTests.cs
@@ -1,5 +1,5 @@
-using AssetRipper.Import.Structure.Assembly.TypeTrees;
-using AssetRipper.IO.Files.SerializedFiles.Parser.TypeTrees;
+using AssetRipper.Import.Structure.Assembly.TypeTrees;
+using AssetRipper.IO.Files.SerializedFiles.Parser.TypeTrees;
 
 namespace AssetRipper.Tests;
 
@@ -8,13 +8,13 @@ internal class TypeTreeNodeStructTests
 	[Test]
 	public void NamedVectorTest()
 	{
-		TypeTree tree = new();
-		tree.Nodes.Add(new TypeTreeNode("MonoBehaviour", "Base", 0, true));
-		tree.Nodes.Add(new TypeTreeNode("SerializableClass", "fieldName", 1, true));
-		tree.Nodes.Add(new TypeTreeNode("Array", "Array", 2, false));
-		tree.Nodes.Add(new TypeTreeNode("int", "size", 3, false));
-		tree.Nodes.Add(new TypeTreeNode("SerializableClass", "data", 3, true));
-		tree.Nodes.Add(new TypeTreeNode("float", "subFieldName1", 4, false));
+		TypeTree tree = new();
+		tree.Nodes.Add(new TypeTreeNode("MonoBehaviour", "Base", 0, true));
+		tree.Nodes.Add(new TypeTreeNode("SerializableClass", "fieldName", 1, true));
+		tree.Nodes.Add(new TypeTreeNode("Array", "Array", 2, false));
+		tree.Nodes.Add(new TypeTreeNode("int", "size", 3, false));
+		tree.Nodes.Add(new TypeTreeNode("SerializableClass", "data", 3, true));
+		tree.Nodes.Add(new TypeTreeNode("float", "subFieldName1", 4, false));
 		tree.Nodes.Add(new TypeTreeNode("bool", "subFieldName2", 4, true));
 
 		Assert.That(TypeTreeNodeStruct.TryMakeFromTypeTree(tree, out TypeTreeNodeStruct rootNode));