1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from unittest import TestCase
from unittest.mock import MagicMock, mock_open, patch
from flask_seeder.generator import Name
MOCK_CONTENTS = [
"name1",
"name2"
]
class TestNameGenerator(TestCase):
def setUp(self):
self.rnd_mock = MagicMock()
self.generator = Name(rnd=self.rnd_mock)
@patch("flask_seeder.generator.read_resource", return_value=MOCK_CONTENTS)
def test_generate_name(self, m_read_resource):
result = self.generator.generate()
self.rnd_mock.choice.assert_called_once_with(MOCK_CONTENTS)
|
