Exemple #1
0
    def test_connect(self):
        sqlite3_mock = MagicMock()
        row_factory_mock = PropertyMock(return_value=None)

        type(sqlite3_mock).row_factory = row_factory_mock
        sqlite3.connect = sqlite3_mock

        db = Database('my_db')
        db.connect()

        sqlite3.connect.assert_called_with('my_db')
        self.assertIsInstance(db.conn, MagicMock)