forked from Katello/katello
-
Notifications
You must be signed in to change notification settings - Fork 0
/
uebercert_generate_test.py
38 lines (28 loc) · 1.16 KB
/
uebercert_generate_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import unittest
from mock import Mock
import urlparse
from cli_test_utils import CLIOptionTestCase, CLIActionTestCase
import katello.client.core.repo
from katello.client.core.organization import GenerateDebugCert
class RequiredCLIOptionsTests(CLIOptionTestCase):
def setUp(self):
self.set_action(GenerateDebugCert())
self.mock_options()
def test_missing_name_generates_error(self):
self.assertRaises(Exception, self.action.process_options, ['uebercert'])
def test_no_error_if_required_options_provided(self):
self.action.process_options(['uebercert', '--name=org'])
self.assertEqual(len(self.action.optErrors), 0)
class CreateUebercertTest(CLIActionTestCase):
NAME = 'ORG'
def setUp(self):
self.set_action(GenerateDebugCert())
self.set_module(katello.client.core.organization)
self.mock_printer()
self.mock_options({ 'name': self.NAME })
self.action.api.uebercert = Mock()
def tearDown(self):
self.restore_mocks()
def test_generates_uebercert_in_cp(self):
self.action.run()
self.action.api.uebercert.assert_called_once_with(self.NAME, None)