class Test(unittest.TestCase):
def setUp(self):
test_dir = os.getcwd() + "\\index_test"
if not os.path.exists(test_dir):
os.makedirs(test_dir)
self.subject = CsvRepository(test_dir, mock_fetcher)
def tearDown(self):
self.subject.clear()
def test_get_fails_on_invalid_key(self):
with self.assertRaises(KeyError):
self.subject.get("INVALID")
def test_delete_no_fail_on_invalid_key(self):
self.subject.delete("INVALID")
def test_getting_existing(self):
key = self.subject.fetch(None, None, None)
self.assertIsNotNone(self.subject.get(key))
def test_get_gives_generator(self):
key = self.subject.fetch(None, None, None)
data = self.subject.get(key)
self.assertEquals((datetime(2012, 11, 01), 1.0), data.next())
self.assertEquals((datetime(2013, 01, 02), 2.5), data.next())
def test_get_fails_after_delete(self):
key = self.subject.fetch(None, None, None)
data = self.subject.get(key)
self.assertEqual(2, len(list(data)))
self.subject.delete(key)
with self.assertRaises(KeyError):
self.subject.get(key)
class Test(unittest.TestCase):
def setUp(self):
test_dir = os.getcwd() + "\\index_test"
if not os.path.exists(test_dir):
os.makedirs(test_dir)
self.subject = CsvRepository(test_dir, mock_fetcher)
def tearDown(self):
self.subject.clear()
def test_get_fails_on_invalid_key(self):
with self.assertRaises(KeyError):
self.subject.get("INVALID")
def test_delete_no_fail_on_invalid_key(self):
self.subject.delete("INVALID")
def test_getting_existing(self):
key = self.subject.fetch(None, None, None)
self.assertIsNotNone(self.subject.get(key))
def test_get_gives_generator(self):
key = self.subject.fetch(None, None, None)
data = self.subject.get(key)
self.assertEquals((datetime(2012,11,01),1.0), data.next())
self.assertEquals((datetime(2013,01,02),2.5), data.next())
def test_get_fails_after_delete(self):
key = self.subject.fetch(None, None, None)
data = self.subject.get(key)
self.assertEqual(2, len(list(data)))
self.subject.delete(key)
with self.assertRaises(KeyError):
self.subject.get(key)
