Add tests for DatabaseTest