From 65474994e075bbc2fdcc1ffa5880cc43bb0d9865 Mon Sep 17 00:00:00 2001
From: cha0s <git@cha0s.io>
Date: Mon, 4 Jan 2021 06:44:32 -0600
Subject: [PATCH] refactor: API

---
 packages/entity/src/entity.js    | 8 ++++----
 packages/entity/test/entity.js   | 2 +-
 packages/entity/test/existent.js | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/packages/entity/src/entity.js b/packages/entity/src/entity.js
index 3e4c658..d365c48 100644
--- a/packages/entity/src/entity.js
+++ b/packages/entity/src/entity.js
@@ -151,10 +151,6 @@ export default (latus) => class Entity extends decorate(JsonResource) {
     }
   }
 
-  allTraitInstances() {
-    return this._traits;
-  }
-
   allTraitTypes() {
     return Object.keys(this._traits);
   }
@@ -362,4 +358,8 @@ export default (latus) => class Entity extends decorate(JsonResource) {
     return this._traits[type];
   }
 
+  get traits() {
+    return this._traits;
+  }
+
 };
diff --git a/packages/entity/test/entity.js b/packages/entity/test/entity.js
index ae3bec5..44b2dea 100644
--- a/packages/entity/test/entity.js
+++ b/packages/entity/test/entity.js
@@ -19,7 +19,7 @@ describe(name, () => {
   });
   it('has sane defaults', () => {
     const entity = new Entity();
-    expect(entity.allTraitInstances()).to.deep.equal({});
+    expect(entity.traits).to.deep.equal({});
     expect(entity.allTraitTypes()).to.deep.equal([]);
   });
   it('can add and remove traits', async () => {
diff --git a/packages/entity/test/existent.js b/packages/entity/test/existent.js
index 0893e0e..b2361c5 100644
--- a/packages/entity/test/existent.js
+++ b/packages/entity/test/existent.js
@@ -53,7 +53,7 @@ describe(name, () => {
           }),
         ]);
         entity.destroy();
-        expect(entity.allTraitInstances()).to.deep.equal({});
+        expect(entity.traits).to.deep.equal({});
         return promise;
       });
     });