X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/91fe9916122adaee2cf1695040f906d709e1aa1c..87fb552dd37301bc0834afc09aa99cadd6310b98:/images_test.py?ds=inline

diff --git a/images_test.py b/images_test.py
index 8ada1ee..3f3fd13 100644
--- a/images_test.py
+++ b/images_test.py
@@ -8,7 +8,7 @@
 import unittest
 
 from fetch import FakeFetcher
-from images import ImageStore
+from images import DiskImageStore
 
 
 class TestImageStore(unittest.TestCase):
@@ -22,7 +22,7 @@ class TestImageStore(unittest.TestCase):
             'https://example.com/other_images/carol': b'CAROLINA'})
 
     def testFetchOnce(self) -> None:
-        store = ImageStore('istest_fetch_once', self._fetcher)
+        store = DiskImageStore('istest_fetch_once', self._fetcher)
         self.assertEqual(self._fetcher.request_count(), 0)
         a1 = store.get_image('https://example.com/images/alice.png')
         self.assertEqual(self._fetcher.request_count(), 1)
@@ -48,7 +48,7 @@ class TestImageStore(unittest.TestCase):
         self.assertEqual(a1, a3)
 
     def testNameCollision(self) -> None:
-        store = ImageStore('istest_name_collision', self._fetcher)
+        store = DiskImageStore('istest_name_collision', self._fetcher)
         self.assertEqual(self._fetcher.request_count(), 0)
         b1 = store.get_image('https://example.com/images/bob.jpeg')
         self.assertEqual(self._fetcher.request_count(), 1)