def test_count(self, mock_get): ShellContext.current_path = Path('/foo') mock_get.return_value = { 'foos': { 'count': 3 } } result = cmds.count() self.assertEqual(result, 3) ShellContext.current_path = Path('/') result = cmds.count(resource='foo') self.assertEqual(result, 3) ShellContext.current_path = Path('/foo/%s' % uuid.uuid4()) result = cmds.count() self.assertEqual(result, None)
def test_count(self, mock_request): p = Path('foo') mock_request.return_value = { 'foos': { 'count': 3 } } result = cmds.count(p) self.assertEqual(result, 3) p = Path() result = cmds.count(p, 'foo') self.assertEqual(result, 3) p = Path('foo/%s' % uuid.uuid4()) result = cmds.count(p) self.assertEqual(result, None)