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)
