def test_dbfeeds(self): """Tests event storage on a database""" self.test_filename = mkstemp(".sqlite", "dbfeedstest")[1] # Register an event using the DBFeed configuration = Configuration({ "feed": "DBFeed", "db_engine": "sqlite:///%s" % self.test_filename }) feed = DBFeed(configuration) event = Event("Test event") event.session = Session(Queue(), "test", "127.0.0.1", 3200, "127.0.0.1", 3201) feed.log(event) feed.stop() # Now check the event in the database conn = sqlite3.connect(self.test_filename) cursor = conn.cursor() cursor.execute('SELECT * FROM events') results = cursor.fetchall() self.assertEqual(len(results), 1) self.assertEqual(results[0][1], str(event.session.uuid)) self.assertEqual(results[0][2], str(event.timestamp)) self.assertEqual(results[0][3], repr(event))
def test_dbfeeds(self): """Tests event storage on a database""" self.test_filename = mkstemp(".sqlite", "dbfeedstest")[1] # Register an event using the DBFeed configuration = Configuration({"feed": "DBFeed", "db_engine": "sqlite:///%s" % self.test_filename}) feed = DBFeed(configuration) event = Event("Test event") event.session = Session(Queue(), "test", "127.0.0.1", 3200, "127.0.0.1", 3201) feed.log(event) feed.stop() # Now check the event in the database conn = sqlite3.connect(self.test_filename) cursor = conn.cursor() cursor.execute('SELECT * FROM events') results = cursor.fetchall() self.assertEqual(len(results), 1) self.assertEqual(results[0][1], str(event.session.uuid)) self.assertEqual(results[0][2], str(event.timestamp)) self.assertEqual(results[0][3], repr(event))
def test_logfeeds(self): self.test_filename = mkstemp(".log", "logfeedstest")[1] # Register an event using the LogFeed configuration = Configuration({"feed": "LogFeed", "log_filename": self.test_filename}) feed = LogFeed(configuration) event = Event("Test event") event.session = Session(Queue(), "test", "127.0.0.1", 3200, "127.0.0.1", 3201) feed.log(event) feed.stop() self.assertIs(path.exists(self.test_filename), True)
def test_hpfeeds(self): """Tests the HPFeed by connecting to honeynet's HPFriends service. """ # Register an event using the HPFeed configuration = Configuration({"feed": "HPFeed", "feed_host": self.test_host, "feed_port": self.test_port, "feed_ident": self.test_ident, "feed_secret": self.test_secret, "channels": [self.test_channel]}) feed = HPFeed(configuration) event = Event("Test event") event.session = Session(Queue(), "test", "127.0.0.1", 3200, "127.0.0.1", 3201) feed.log(event) feed.stop()
def test_logfeeds(self): self.test_filename = mkstemp(".log", "logfeedstest")[1] # Register an event using the LogFeed configuration = Configuration({ "feed": "LogFeed", "log_filename": self.test_filename }) feed = LogFeed(configuration) event = Event("Test event") event.session = Session(Queue(), "test", "127.0.0.1", 3200, "127.0.0.1", 3201) feed.log(event) feed.stop() self.assertIs(path.exists(self.test_filename), True)