Selaa lähdekoodia

Added sample test case

pull/147/head
Johannes 8 vuotta sitten
vanhempi
commit
73e0726434
4 muutettua tiedostoa jossa 32 lisäystä ja 3 poistoa
  1. +3
    -1
      .travis.yml
  2. +7
    -1
      modules/core/core.py
  3. +3
    -1
      modules/logs/__init__.py
  4. +19
    -0
      tests/logs/test_log_view.py

+ 3
- 1
.travis.yml Näytä tiedosto

@@ -3,4 +3,6 @@ python:
- "2.7"
# command to install dependencies
install:
- pip install -r requirements.txt
- pip install -r requirements.txt

script: make test

+ 7
- 1
modules/core/core.py Näytä tiedosto

@@ -10,6 +10,7 @@ from functools import wraps, update_wrapper
from importlib import import_module
from time import localtime, strftime
import time
import os.path
from flask import Flask, redirect, json, g, make_response
from flask_socketio import SocketIO
@@ -275,11 +276,16 @@ class FermentationCore(object):
class CraftBeerPI(object):
_logger_configuration_file = './config/logger.yaml'
cache = {}
eventbus = {}
def __init__(self):
logging.config.dictConfig(yaml.load(open('./config/logger.yaml', 'r')))
if os.path.isfile(self._logger_configuration_file):
logging.config.dictConfig(yaml.load(open(self._logger_configuration_file, 'r')))
self.logger = logging.getLogger(__name__)
self.logger.info("Logger got initialized.")


+ 3
- 1
modules/logs/__init__.py Näytä tiedosto

@@ -7,6 +7,8 @@ from modules.core.core import cbpi
class LogView(FlaskView):
_log_directory = "./logs"
@route('/', methods=['GET'])
def get_all_logfiles(self):
"""
@@ -19,7 +21,7 @@ class LogView(FlaskView):
description: List of all log files
"""
result = []
for filename in os.listdir("./logs"):
for filename in os.listdir(self._log_directory):
if filename.endswith(".log"):
result.append(filename)
return json.dumps(result)


+ 19
- 0
tests/logs/test_log_view.py Näytä tiedosto

@@ -0,0 +1,19 @@
import unittest

from modules.logs import LogView


class TestLogView(unittest.TestCase):

def test_get_all_logfiles(self):
LogView._log_directory = "../../logs"

log_view = LogView()
logfiles = log_view.get_all_logfiles()

print(logfiles)
self.assertTrue(len(logfiles) > 0)


if __name__ == '__main__':
unittest.main()

Loading…
Peruuta
Tallenna