def test_add_view(self):
"""Test adding a view to a sketch."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
view = sketch.add_view(
view_name="MockView", analyzer_name="Test", query_string="test"
)
self.assertIsInstance(view, View)
def test_get_all_instances(self):
"""Test get all indices from a sketch."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
indices = sketch.get_all_indices()
self.assertIsInstance(indices, list)
self.assertEqual(len(indices), 1)
self.assertEqual(indices[0], "test")
def test_event(self):
"""Tests creating an Event object."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
datastore = MockDataStore('127.0.0.1', 4711)
valid_event = dict(
_id='1', _type='test', _index='test', _source='test')
invalid_event = dict(_id='1')
event = interface.Event(valid_event, datastore, sketch=None)
sketch_event = interface.Event(valid_event, datastore, sketch=sketch)
self.assertIsInstance(event.datastore, MockDataStore)
self.assertIsNone(event.sketch)
self.assertIsInstance(sketch_event.sketch, interface.Sketch)
self.assertRaises(KeyError, interface.Event, invalid_event, datastore)
def test_add_story(self):
"""Test adding a story to a sketch."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
story = sketch.add_story(title="test")
view = sketch.add_view(
view_name="MockView", analyzer_name="Test", query_string="test"
)
sql_story = story.story
story.add_text("test")
story.add_view(view)
content_json = sql_story.content
content_list = json.loads(sql_story.content)
self.assertIsInstance(sql_story, Story)
self.assertIsInstance(content_json, str)
self.assertIsInstance(content_list, list)
self.assertIsInstance(content_list[0], dict)
self.assertIsInstance(content_list[1], dict)
def test_event(self):
"""Tests creating an Event object."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
datastore = MockDataStore("127.0.0.1", 4711)
valid_event = dict(
_id="1",
_type="test",
_index="test",
_source={"__ts_timeline_id": 1, "test": True},
)
invalid_event = dict(_id="1")
event = interface.Event(valid_event, datastore, sketch=None)
sketch_event = interface.Event(valid_event, datastore, sketch=sketch)
self.assertIsInstance(event.datastore, MockDataStore)
self.assertIsNone(event.sketch)
self.assertIsInstance(sketch_event.sketch, interface.Sketch)
self.assertRaises(KeyError, interface.Event, invalid_event, datastore)
def test_sketch(self):
"""Test creating a sketch object."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
self.assertEqual(sketch.id, self.SKETCH_ID)
self.assertIsInstance(sketch.sql_sketch, Sketch)
def test_add_view(self):
"""Test adding a view to a sketch."""
sketch = interface.Sketch(sketch_id=self.SKETCH_ID)
view = sketch.add_view('MockView', query_string='test')
self.assertIsInstance(view, View)
