cache many components per container

The way the cache was set up, we could only have one render
element per test. Which, mind you, is the most common use-case,
but it can create surprises.

Fixes #205

Author: yanick

src/__tests__/auto-cleanup.test.js

@@ -13,3 +13,17 @@ describe('auto-cleanup', () => {
     expect(document.body.innerHTML).toEqual('')
   })
 })
+
+describe('cleanup of two components', () => {
+  // This just verifies that by importing STL in an
+  // environment which supports afterEach (like jest)
+  // we'll get automatic cleanup between tests.
+  test('first', () => {
+    render(Comp, { props: { name: 'world' } })
+    render(Comp, { props: { name: 'universe' } })
+  })
+
+  test('second', () => {
+    expect(document.body.innerHTML).toEqual('')
+  })
+})

src/pure.js

@@ -5,7 +5,7 @@ import {
 } from '@testing-library/dom'
 import { tick } from 'svelte'
 
-const containerCache = new Map()
+const containerCache = new Set()
 const componentCache = new Set()
 
 const svelteComponentOptions = [
@@ -59,7 +59,7 @@ const render = (
     ...checkProps(options),
   })
 
-  containerCache.set(container, { target, component })
+  containerCache.add({ container, target, component })
   componentCache.add(component)
 
   component.$$.on_destroy.push(() => {
@@ -79,7 +79,7 @@ const render = (
         ...checkProps(options),
       })
 
-      containerCache.set(container, { target, component })
+      containerCache.add({ container, target, component })
       componentCache.add(component)
 
       component.$$.on_destroy.push(() => {
@@ -93,16 +93,16 @@ const render = (
   }
 }
 
-const cleanupAtContainer = (container) => {
-  const { target, component } = containerCache.get(container)
+const cleanupAtContainer = (cached) => {
+  const { target, component } = cached
 
   if (componentCache.has(component)) component.$destroy()
 
   if (target.parentNode === document.body) {
     document.body.removeChild(target)
   }
 
-  containerCache.delete(container)
+  containerCache.delete(cached)
 }
 
 const cleanup = () => {