1 files changed, 11 insertions, 1 deletions
diff --git a/pokemodr/models/RootModel.cpp b/pokemodr/models/RootModel.cpp
index 7dac9b39..40759f7f 100644
--- a/pokemodr/models/RootModel.cpp
+++ b/pokemodr/models/RootModel.cpp
@@ -26,6 +26,9 @@
 
 // Test code includes
 #include <QFile>
+#include "../../pokemod/Ability.h"
+#include "../../pokemod/AbilityEffect.h"
+#include "AbilityModel.h"
 
 RootModel::RootModel(const QList<QVariant>& pokemods) :
         GroupModel(NULL, NULL)
@@ -71,6 +74,13 @@ void RootModel::setupData(const QList<QVariant>& pokemods)
     QDomDocument xml;
     xml.setContent(&fin);
     fin.close();
-    m_objects.append(new PokemodModel(this, new Pokemod(xml.documentElement())));
+    Pokemod* pokemod = new Pokemod();
+    Ability* ability = pokemod->newAbility();
+    ability->setName("foo");
+    ability->newEffect()->setEffect(AbilityEffect::E_PreventDamage);
+    m_objects.append(new AbilityModel(this, ability));
+    m_objects.append(new AbilityGroupModel(this, pokemod));
+    Pokemod* full = new Pokemod(xml.documentElement());
+    m_objects.append(new PokemodModel(this, full));
     // TODO: make sub models from data
 }