mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-03 13:36:07 +00:00
585523750e
This patch loosen the parsing requirement to allow parsing not only JSON dictionaries but also valid JSON type (integer, float, string, bool, array, null). Signed-off-by: Med Ismail Bennani <ismail@bennani.ma>
304 lines
12 KiB
Python
304 lines
12 KiB
Python
"""
|
|
Test some SBStructuredData API.
|
|
"""
|
|
|
|
|
|
import lldb
|
|
from lldbsuite.test.decorators import *
|
|
from lldbsuite.test.lldbtest import *
|
|
from lldbsuite.test import lldbutil
|
|
|
|
import json
|
|
|
|
|
|
class TestStructuredDataAPI(TestBase):
|
|
NO_DEBUG_INFO_TESTCASE = True
|
|
|
|
def test(self):
|
|
self.structured_data_api_test()
|
|
|
|
def structured_data_api_test(self):
|
|
error = lldb.SBError()
|
|
s = lldb.SBStream()
|
|
|
|
dict_str = json.dumps(
|
|
{
|
|
"key_dict": {
|
|
"key_string": "STRING",
|
|
"key_uint": 0xFFFFFFFF00000000,
|
|
"key_sint": -42,
|
|
"key_float": 2.99,
|
|
"key_bool": True,
|
|
"key_array": ["23", "arr"],
|
|
}
|
|
}
|
|
)
|
|
s.Print(dict_str)
|
|
example = lldb.SBStructuredData()
|
|
|
|
# Check SetFromJSON API for dictionaries, integers, floating point
|
|
# values, strings and arrays
|
|
error = example.SetFromJSON(s)
|
|
if not error.Success():
|
|
self.fail("FAILED: " + error.GetCString())
|
|
|
|
# Tests for invalid data type
|
|
self.invalid_struct_test(example)
|
|
|
|
# Test that GetDescription works:
|
|
s.Clear()
|
|
error = example.GetDescription(s)
|
|
self.assertSuccess(error, "GetDescription works")
|
|
if not "key_float" in s.GetData():
|
|
self.fail("FAILED: could not find key_float in description output")
|
|
|
|
dict_struct = lldb.SBStructuredData()
|
|
dict_struct = example.GetValueForKey("key_dict")
|
|
|
|
# Tests for dictionary data type
|
|
self.dictionary_struct_test(example)
|
|
|
|
# Tests for string data type
|
|
self.string_struct_test(dict_struct)
|
|
|
|
# Tests for integer data type
|
|
self.uint_struct_test(dict_struct)
|
|
|
|
# Tests for integer data type
|
|
self.sint_struct_test(dict_struct)
|
|
|
|
# Tests for floating point data type
|
|
self.double_struct_test(dict_struct)
|
|
|
|
# Tests for boolean data type
|
|
self.bool_struct_test(dict_struct)
|
|
|
|
# Tests for array data type
|
|
self.array_struct_test(dict_struct)
|
|
|
|
s.Clear()
|
|
self.assertSuccess(example.GetAsJSON(s))
|
|
py_obj = json.loads(s.GetData())
|
|
self.assertTrue(py_obj)
|
|
self.assertIn("key_dict", py_obj)
|
|
|
|
py_dict = py_obj["key_dict"]
|
|
self.assertEqual(py_dict["key_string"], "STRING")
|
|
self.assertEqual(py_dict["key_uint"], 0xFFFFFFFF00000000)
|
|
self.assertEqual(py_dict["key_sint"], -42)
|
|
self.assertEqual(py_dict["key_float"], 2.99)
|
|
self.assertEqual(py_dict["key_bool"], True)
|
|
self.assertEqual(py_dict["key_array"], ["23", "arr"])
|
|
|
|
class MyRandomClass:
|
|
payload = "foo"
|
|
|
|
py_dict["key_generic"] = MyRandomClass()
|
|
|
|
stp = lldb.SBScriptObject(py_dict, lldb.eScriptLanguagePython)
|
|
self.assertEqual(stp.ptr, py_dict)
|
|
|
|
sd = lldb.SBStructuredData(stp, self.dbg)
|
|
self.assertTrue(sd.IsValid())
|
|
self.assertEqual(sd.GetSize(), len(py_dict))
|
|
|
|
generic_sd = sd.GetValueForKey("key_generic")
|
|
self.assertTrue(generic_sd.IsValid())
|
|
self.assertEqual(generic_sd.GetType(), lldb.eStructuredDataTypeGeneric)
|
|
|
|
my_random_class = generic_sd.GetGenericValue()
|
|
self.assertTrue(my_random_class)
|
|
self.assertEqual(my_random_class.payload, MyRandomClass.payload)
|
|
|
|
example = lldb.SBStructuredData()
|
|
self.assertSuccess(example.SetFromJSON("1"))
|
|
self.assertEqual(example.GetType(), lldb.eStructuredDataTypeInteger)
|
|
self.assertEqual(example.GetIntegerValue(), 1)
|
|
|
|
self.assertSuccess(example.SetFromJSON("4.19"))
|
|
self.assertEqual(example.GetType(), lldb.eStructuredDataTypeFloat)
|
|
self.assertEqual(example.GetFloatValue(), 4.19)
|
|
|
|
self.assertSuccess(example.SetFromJSON('"Bonjour, 123!"'))
|
|
self.assertEqual(example.GetType(), lldb.eStructuredDataTypeString)
|
|
self.assertEqual(example.GetStringValue(42), "Bonjour, 123!")
|
|
|
|
self.assertSuccess(example.SetFromJSON("true"))
|
|
self.assertEqual(example.GetType(), lldb.eStructuredDataTypeBoolean)
|
|
self.assertTrue(example.GetBooleanValue())
|
|
|
|
self.assertSuccess(example.SetFromJSON("null"))
|
|
self.assertEqual(example.GetType(), lldb.eStructuredDataTypeNull)
|
|
|
|
example_arr = [1, 2.3, "4", {"5": False}]
|
|
arr_str = json.dumps(example_arr)
|
|
s.Clear()
|
|
s.Print(arr_str)
|
|
self.assertSuccess(example.SetFromJSON(s))
|
|
|
|
s.Clear()
|
|
self.assertSuccess(example.GetAsJSON(s))
|
|
sb_data = json.loads(s.GetData())
|
|
self.assertEqual(sb_data, example_arr)
|
|
|
|
def invalid_struct_test(self, example):
|
|
invalid_struct = lldb.SBStructuredData()
|
|
invalid_struct = example.GetValueForKey("invalid_key")
|
|
if invalid_struct.IsValid():
|
|
self.fail("An invalid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not invalid_struct.GetType() == lldb.eStructuredDataTypeInvalid:
|
|
self.fail("Wrong type returned: " + str(invalid_struct.GetType()))
|
|
|
|
def dictionary_struct_test(self, example):
|
|
# Check API returning a valid SBStructuredData of 'dictionary' type
|
|
dict_struct = lldb.SBStructuredData()
|
|
dict_struct = example.GetValueForKey("key_dict")
|
|
if not dict_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not dict_struct.GetType() == lldb.eStructuredDataTypeDictionary:
|
|
self.fail("Wrong type returned: " + str(dict_struct.GetType()))
|
|
|
|
# Check Size API for 'dictionary' type
|
|
if not dict_struct.GetSize() == 6:
|
|
self.fail("Wrong no of elements returned: " + str(dict_struct.GetSize()))
|
|
|
|
def string_struct_test(self, dict_struct):
|
|
string_struct = lldb.SBStructuredData()
|
|
string_struct = dict_struct.GetValueForKey("key_string")
|
|
if not string_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not string_struct.GetType() == lldb.eStructuredDataTypeString:
|
|
self.fail("Wrong type returned: " + str(string_struct.GetType()))
|
|
|
|
# Check API returning 'string' value
|
|
output = string_struct.GetStringValue(25)
|
|
if not "STRING" in output:
|
|
self.fail("wrong output: " + output)
|
|
|
|
# Calling wrong API on a SBStructuredData
|
|
# (e.g. getting an integer from a string type structure)
|
|
output = string_struct.GetIntegerValue()
|
|
if output:
|
|
self.fail(
|
|
"Valid integer value " + str(output) + " returned for a string object"
|
|
)
|
|
|
|
def uint_struct_test(self, dict_struct):
|
|
# Check a valid SBStructuredData containing an unsigned integer.
|
|
# We intentionally make this larger than what an int64_t can hold but
|
|
# still small enough to fit a uint64_t
|
|
uint_struct = lldb.SBStructuredData()
|
|
uint_struct = dict_struct.GetValueForKey("key_uint")
|
|
if not uint_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not uint_struct.GetType() == lldb.eStructuredDataTypeInteger:
|
|
self.fail("Wrong type returned: " + str(uint_struct.GetType()))
|
|
|
|
# Check API returning unsigned integer value
|
|
output = uint_struct.GetUnsignedIntegerValue()
|
|
if not output == 0xFFFFFFFF00000000:
|
|
self.fail("wrong output: " + str(output))
|
|
|
|
# Calling wrong API on a SBStructuredData
|
|
# (e.g. getting a string value from an integer type structure)
|
|
output = uint_struct.GetStringValue(25)
|
|
if output:
|
|
self.fail("Valid string " + output + " returned for an integer object")
|
|
|
|
def sint_struct_test(self, dict_struct):
|
|
# Check a valid SBStructuredData containing an signed integer.
|
|
# We intentionally make this smaller than what an uint64_t can hold but
|
|
# still small enough to fit a int64_t
|
|
sint_struct = lldb.SBStructuredData()
|
|
sint_struct = dict_struct.GetValueForKey("key_sint")
|
|
if not sint_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not sint_struct.GetType() == lldb.eStructuredDataTypeSignedInteger:
|
|
self.fail("Wrong type returned: " + str(sint_struct.GetType()))
|
|
|
|
# Check API returning signed integer value
|
|
output = sint_struct.GetSignedIntegerValue()
|
|
if not output == -42:
|
|
self.fail("wrong output: " + str(output))
|
|
|
|
# Calling wrong API on a SBStructuredData
|
|
# (e.g. getting a string value from an integer type structure)
|
|
output = sint_struct.GetStringValue(69)
|
|
if output:
|
|
self.fail("Valid string " + output + " returned for an integer object")
|
|
|
|
def double_struct_test(self, dict_struct):
|
|
floating_point_struct = lldb.SBStructuredData()
|
|
floating_point_struct = dict_struct.GetValueForKey("key_float")
|
|
if not floating_point_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not floating_point_struct.GetType() == lldb.eStructuredDataTypeFloat:
|
|
self.fail("Wrong type returned: " + str(floating_point_struct.GetType()))
|
|
|
|
# Check API returning 'double' value
|
|
output = floating_point_struct.GetFloatValue()
|
|
if not output == 2.99:
|
|
self.fail("wrong output: " + str(output))
|
|
|
|
def bool_struct_test(self, dict_struct):
|
|
bool_struct = lldb.SBStructuredData()
|
|
bool_struct = dict_struct.GetValueForKey("key_bool")
|
|
if not bool_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not bool_struct.GetType() == lldb.eStructuredDataTypeBoolean:
|
|
self.fail("Wrong type returned: " + str(bool_struct.GetType()))
|
|
|
|
# Check API returning 'bool' value
|
|
output = bool_struct.GetBooleanValue()
|
|
if not output:
|
|
self.fail("wrong output: " + str(output))
|
|
|
|
def array_struct_test(self, dict_struct):
|
|
# Check API returning a valid SBStructuredData of 'array' type
|
|
array_struct = lldb.SBStructuredData()
|
|
array_struct = dict_struct.GetValueForKey("key_array")
|
|
if not array_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
|
|
# Check Type API
|
|
if not array_struct.GetType() == lldb.eStructuredDataTypeArray:
|
|
self.fail("Wrong type returned: " + str(array_struct.GetType()))
|
|
|
|
# Check Size API for 'array' type
|
|
if not array_struct.GetSize() == 2:
|
|
self.fail("Wrong no of elements returned: " + str(array_struct.GetSize()))
|
|
|
|
# Check API returning a valid SBStructuredData for different 'array'
|
|
# indices
|
|
string_struct = array_struct.GetItemAtIndex(0)
|
|
if not string_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
if not string_struct.GetType() == lldb.eStructuredDataTypeString:
|
|
self.fail("Wrong type returned: " + str(string_struct.GetType()))
|
|
output = string_struct.GetStringValue(5)
|
|
if not output == "23":
|
|
self.fail("wrong output: " + str(output))
|
|
|
|
string_struct = array_struct.GetItemAtIndex(1)
|
|
if not string_struct.IsValid():
|
|
self.fail("A valid object should have been returned")
|
|
if not string_struct.GetType() == lldb.eStructuredDataTypeString:
|
|
self.fail("Wrong type returned: " + str(string_struct.GetType()))
|
|
output = string_struct.GetStringValue(5)
|
|
if not output == "arr":
|
|
self.fail("wrong output: " + str(output))
|