示例#1
0
class ServerScannerTestCase(TestCase):
    def setUp(self):
        self.server = Mock()
        self.queue = Queue()
        self.scanner = ServerScanner(self.queue, server=self.server)

    @patch('scanner.api.user.add_user')
    def test_block_add(self, add_user):
        self.scanner.add_user('mike')
        self.assertFalse(add_user.called)

    @patch('scanner.api.user.add_user')
    def test_unblock_add(self, add_user):
        scanner = ServerScanner(queue=self.queue,
                                server=self.server,
                                add_users=True)
        scanner.add_user('mike')
        add_user.assert_called_once_with('mike', self.server.hostname)

    @patch('scanner.api.user.remove_user')
    def test_block_remove(self, remove_user):
        self.scanner.remove_user('mike')
        self.assertFalse(remove_user.called)

    @patch('scanner.api.user.remove_user')
    def test_unblock_remove(self, remove_user):
        scanner = ServerScanner(queue=self.queue,
                                server=self.server,
                                remove_users=True)
        scanner.remove_user('mike')
        remove_user.assert_called_once_with('mike', self.server.hostname)
示例#2
0
class ServerScannerTestCase(TestCase):

    def setUp(self):
        self.server = Mock()
        self.queue = Queue()
        self.scanner = ServerScanner(self.queue, server=self.server)

    @patch('scanner.api.user.add_user')
    def test_block_add(self, add_user):
        self.scanner.add_user('mike')
        self.assertFalse(add_user.called)

    @patch('scanner.api.user.add_user')
    def test_unblock_add(self, add_user):
        scanner = ServerScanner(queue=self.queue, server=self.server, add_users=True)
        scanner.add_user('mike')
        add_user.assert_called_once_with('mike', self.server.hostname)

    @patch('scanner.api.user.remove_user')
    def test_block_remove(self, remove_user):
        self.scanner.remove_user('mike')
        self.assertFalse(remove_user.called)

    @patch('scanner.api.user.remove_user')
    def test_unblock_remove(self, remove_user):
        scanner = ServerScanner(queue=self.queue, server=self.server, remove_users=True)
        scanner.remove_user('mike')
        remove_user.assert_called_once_with('mike', self.server.hostname)
示例#3
0
 def test_unblock_add(self, add_user):
     scanner = ServerScanner(queue=self.queue,
                             server=self.server,
                             add_users=True)
     scanner.add_user('mike')
     add_user.assert_called_once_with('mike', self.server.hostname)
示例#4
0
 def test_unblock_add(self, add_user):
     scanner = ServerScanner(queue=self.queue, server=self.server, add_users=True)
     scanner.add_user('mike')
     add_user.assert_called_once_with('mike', self.server.hostname)