Add assert for failed getBucketForElement().

phetsims · Mar 9, 2020 · 5104eb3 · 5104eb3
1 parent 75a45ad
commit 5104eb3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/js/common/model/Kit.js b/js/common/model/Kit.js
@@ -165,10 +165,11 @@ class Kit {
    * @returns {Bucket}
    */
   getBucketForElement( element ) {
-    //TODO: Add assert for failed _find()
-    return _.find( this.buckets, bucket => {
+    const elementBucket = _.find( this.buckets, bucket => {
       return bucket.element.isSameElement( element );
     } );
+    assert && assert( elementBucket !== null, 'Element does not have an associated bucket.' );
+    return elementBucket
   }
 
   /**