File: test_flask_seeder.py

package info (click to toggle)
python-flask-seeder 1.2.0-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 292 kB
  • sloc: python: 1,062; makefile: 2
file content (18 lines) | stat: -rw-r--r-- 517 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from unittest import TestCase
from unittest.mock import MagicMock
from flask_seeder import FlaskSeeder

class TestFlaskSeeder(TestCase):

    def test_init_app_without_db(self):
        """ FlaskSeeder should use db object when passed via constructor.
        """
        db = MagicMock()
        app = MagicMock()
        ext = {}
        app.extensions.__setitem__.side_effect = ext.__setitem__
        seeder = FlaskSeeder(db=db)

        seeder.init_app(app)

        self.assertEqual(ext["flask_seeder"].db, db)