def testGetFormatVersion(self):
"""Tests the GetFormatVersion function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
format_version = test_reader.GetFormatVersion()
self.assertIsNone(format_version)
def testGetSerializationFormat(self):
"""Tests the GetSerializationFormat function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
serialization_format = test_reader.GetSerializationFormat()
self.assertIsNone(serialization_format)
def testGetStorageType(self):
"""Tests the GetStorageType function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
storage_type = test_reader.GetStorageType()
self.assertEqual(storage_type, 'session')
def testHasAttributeContainers(self):
"""Tests the HasAttributeContainers function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
test_reader._store.Open()
try:
event_source = event_sources.EventSource()
test_reader._store.AddAttributeContainer(event_source)
result = test_reader.HasAttributeContainers(event_source.CONTAINER_TYPE)
self.assertTrue(result)
finally:
test_reader._store.Close()
def testGetNumberOfAttributeContainers(self):
"""Tests the GetNumberOfAttributeContainers function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
test_reader._store.Open()
try:
event_source = event_sources.EventSource()
test_reader._store.AddAttributeContainer(event_source)
number_of_containers = test_reader.GetNumberOfAttributeContainers(
event_source.CONTAINER_TYPE)
self.assertEqual(number_of_containers, 1)
finally:
test_reader._store.Close()
def testGetAttributeContainers(self):
"""Tests the GetAttributeContainers function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
test_reader._store.Open()
try:
event_source = event_sources.EventSource()
test_reader._store.AddAttributeContainer(event_source)
test_generator = test_reader.GetAttributeContainers(
event_source.CONTAINER_TYPE)
test_containers = list(test_generator)
self.assertEqual(len(test_containers), 1)
finally:
test_reader._store.Close()
def testGetAttributeContainerByIndex(self):
"""Tests the GetAttributeContainerByIndex function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
test_reader._store.Open()
try:
event_source = event_sources.EventSource()
test_reader._store.AddAttributeContainer(event_source)
test_container = test_reader.GetAttributeContainerByIndex(
event_source.CONTAINER_TYPE, 0)
self.assertIsNotNone(test_container)
test_container = test_reader.GetAttributeContainerByIndex(
event_source.CONTAINER_TYPE, 99)
self.assertIsNone(test_container)
finally:
test_reader._store.Close()
def testInitialization(self):
"""Tests the __init__ function."""
test_reader = reader.StorageReader()
self.assertIsNotNone(test_reader)
def testSetStorageProfiler(self):
"""Tests the SetStorageProfiler function."""
test_reader = reader.StorageReader()
test_reader._store = fake_store.FakeStore()
test_reader.SetStorageProfiler(None)
