From c2b7bb20e6a31243c91005d3bd0eabf380c1b5cf Mon Sep 17 00:00:00 2001
From: Collin Smith <collinsmith70@gmail.com>
Date: Fri, 8 Jan 2021 17:16:25 -0800
Subject: [PATCH] Created ArmType schema and table test

---
 .../java/com/riiablo/table/schema/ArmType.java   | 16 ++++++++++++++++
 .../test/java/com/riiablo/table/TablesTest.java  | 12 ++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 core/src/main/java/com/riiablo/table/schema/ArmType.java

diff --git a/core/src/main/java/com/riiablo/table/schema/ArmType.java b/core/src/main/java/com/riiablo/table/schema/ArmType.java
new file mode 100644
index 00000000..bf69d5bd
--- /dev/null
+++ b/core/src/main/java/com/riiablo/table/schema/ArmType.java
@@ -0,0 +1,16 @@
+package com.riiablo.table.schema;
+
+import com.riiablo.table.annotation.PrimaryKey;
+import com.riiablo.table.annotation.Schema;
+
+@Schema
+public class ArmType {
+  @Override
+  public String toString() {
+    return Name;
+  }
+
+  @PrimaryKey
+  public String Token;
+  public String Name;
+}
diff --git a/core/src/test/java/com/riiablo/table/TablesTest.java b/core/src/test/java/com/riiablo/table/TablesTest.java
index e8b93714..f96a1276 100644
--- a/core/src/test/java/com/riiablo/table/TablesTest.java
+++ b/core/src/test/java/com/riiablo/table/TablesTest.java
@@ -22,6 +22,7 @@ import com.riiablo.table.schema.RarePrefix;
 import com.riiablo.table.schema.RareSuffix;
 import com.riiablo.table.schema.Sounds;
 import com.riiablo.table.schema.Weapons;
+import com.riiablo.table.table.ArmTypeTable;
 import com.riiablo.table.table.BodyLocsTable;
 import com.riiablo.table.table.ItemStatCostTable;
 import com.riiablo.table.table.MagicPrefixTable;
@@ -425,4 +426,15 @@ public class TablesTest extends RiiabloTest {
     Assert.assertEquals("TRAMPLE", table.get(106).name);
     Assert.assertEquals("FLANGE", table.get(154).name);
   }
+
+  @Test
+  public void armtype() {
+    LogManager.setLevel("com.riiablo.table.table.ArmTypeTable", Level.TRACE);
+    TableManifest.armtype.parser = null;
+    FileHandle handle = Gdx.files.internal("test/armtype.txt");
+    ArmTypeTable table = Tables.loadTsv(TableManifest.armtype, handle);
+    Assert.assertEquals("LIT", table.get(0).Token);
+    Assert.assertEquals("MED", table.get(1).Token);
+    Assert.assertEquals("HVY", table.get(2).Token);
+  }
 }