def test_files_offset_limit(self, m_FileExt): offset = randint(1000, 2000) limit = randint(0, 1000) api_files.list(offset=offset, limit=limit) m_FileExt.query_find_by_name().limit.assert_called_once_with(limit) m_offset = m_FileExt.query_find_by_name().limit().offset m_offset.assert_called_once_with(offset)
def test_files_raise_none_None(self): name = "whatever" hash = "something" with self.assertRaises(ValueError) as context: api_files.list(name=name, hash=hash) self.assertEqual(str(context.exception), "Can't find using both name and hash")
def test_files_by_sha256(self, m_FileExt): h = hashlib.md5() data = "something".encode("utf-8") h.update(data) hash_val = h.hexdigest() api_files.list(hash=hash_val) m_FileExt.query_find_by_hash.assert_called_once_with( "md5", hash_val, None, self.session)
def test_files_by_sha256(self, m_FileExt): h = hashlib.md5() data = "something".encode("utf-8") h.update(data) hash_val = h.hexdigest() api_files.list(hash=hash_val) m_FileExt.query_find_by_hash.assert_called_once_with("md5", hash_val, None, self.session)
def test_files_by_tags(self, m_FileExt): tag_list = ["tag1", "tag2"] api_files.list(tags=tag_list) m_FileExt.query_find_by_name.assert_called_once_with( "", tag_list, self.session)
def test_files_all(self, m_FileExt): api_files.list() m_FileExt.query_find_by_name.assert_called_once_with( "", None, self.session)
def test_files_by_name(self, m_FileExt): name = "something" api_files.list(name=name) m_FileExt.query_find_by_name.assert_called_once_with( name, None, self.session)
def test_files_raise_h_type_None(self): hash = "something" with self.assertRaises(ValueError) as context: api_files.list(hash=hash) self.assertEqual(str(context.exception), "Hash not supported")
def test_files_by_tags(self, m_FileExt): tag_list = ["tag1", "tag2"] api_files.list(tags=tag_list) m_FileExt.query_find_by_name.assert_called_once_with("", tag_list, self.session)
def test_files_all(self, m_FileExt): api_files.list() m_FileExt.query_find_by_name.assert_called_once_with("", None, self.session)
def test_files_by_name(self, m_FileExt): name = "something" api_files.list(name=name) m_FileExt.query_find_by_name.assert_called_once_with(name, None, self.session)