kopia lustrzana https://github.com/glidernet/ogn-python
24 wiersze
593 B
Python
24 wiersze
593 B
Python
import unittest
|
|
import os
|
|
|
|
from tests.base import TestBaseDB, db
|
|
|
|
from ogn_python.model import DeviceInfo
|
|
from ogn_python.commands.database import import_file
|
|
|
|
from ogn_python import app
|
|
|
|
|
|
class TestDatabase(TestBaseDB):
|
|
def test_import_ddb_file(self):
|
|
runner = app.test_cli_runner()
|
|
result = runner.invoke(import_file, [os.path.dirname(__file__) + '/../custom_ddb.txt'])
|
|
self.assertEqual(result.exit_code, 0)
|
|
|
|
device_infos = db.session.query(DeviceInfo).all()
|
|
self.assertEqual(len(device_infos), 6)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|