122c8efcdSZachary Turner //===-- PythonDataObjectsTests.cpp ------------------------------*- C++ -*-===// 222c8efcdSZachary Turner // 322c8efcdSZachary Turner // The LLVM Compiler Infrastructure 422c8efcdSZachary Turner // 522c8efcdSZachary Turner // This file is distributed under the University of Illinois Open Source 622c8efcdSZachary Turner // License. See LICENSE.TXT for details. 722c8efcdSZachary Turner // 822c8efcdSZachary Turner //===----------------------------------------------------------------------===// 922c8efcdSZachary Turner 1022c8efcdSZachary Turner #include "gtest/gtest.h" 1122c8efcdSZachary Turner 129c40264fSZachary Turner #include "lldb/Host/File.h" 139c40264fSZachary Turner #include "lldb/Host/FileSystem.h" 1422c8efcdSZachary Turner #include "lldb/Host/HostInfo.h" 1522c8efcdSZachary Turner #include "Plugins/ScriptInterpreter/Python/lldb-python.h" 1622c8efcdSZachary Turner #include "Plugins/ScriptInterpreter/Python/PythonDataObjects.h" 1722c8efcdSZachary Turner #include "Plugins/ScriptInterpreter/Python/ScriptInterpreterPython.h" 1822c8efcdSZachary Turner 19c946d462SZachary Turner #include "PythonTestSuite.h" 20c946d462SZachary Turner 2122c8efcdSZachary Turner using namespace lldb_private; 2222c8efcdSZachary Turner 23c946d462SZachary Turner class PythonDataObjectsTest : public PythonTestSuite 2422c8efcdSZachary Turner { 2522c8efcdSZachary Turner public: 2622c8efcdSZachary Turner void 2722c8efcdSZachary Turner SetUp() override 2822c8efcdSZachary Turner { 29c946d462SZachary Turner PythonTestSuite::SetUp(); 30a1405147SZachary Turner 31a1405147SZachary Turner PythonString sys_module("sys"); 32a1405147SZachary Turner m_sys_module.Reset(PyRefType::Owned, PyImport_Import(sys_module.get())); 33a1405147SZachary Turner m_main_module = PythonModule::MainModule(); 34a1405147SZachary Turner m_builtins_module = PythonModule::BuiltinsModule(); 3522c8efcdSZachary Turner } 3622c8efcdSZachary Turner 3722c8efcdSZachary Turner void 3822c8efcdSZachary Turner TearDown() override 3922c8efcdSZachary Turner { 40a1405147SZachary Turner m_sys_module.Reset(); 41a1405147SZachary Turner m_main_module.Reset(); 42a1405147SZachary Turner m_builtins_module.Reset(); 43f8b22f8fSZachary Turner 44c946d462SZachary Turner PythonTestSuite::TearDown(); 4522c8efcdSZachary Turner } 46f8b22f8fSZachary Turner 47a1405147SZachary Turner protected: 48a1405147SZachary Turner PythonModule m_sys_module; 49a1405147SZachary Turner PythonModule m_main_module; 50a1405147SZachary Turner PythonModule m_builtins_module; 5122c8efcdSZachary Turner }; 5222c8efcdSZachary Turner 53f8b22f8fSZachary Turner TEST_F(PythonDataObjectsTest, TestOwnedReferences) 54f8b22f8fSZachary Turner { 5518426935SZachary Turner // After creating a new object, the refcount should be >= 1 56f8b22f8fSZachary Turner PyObject *obj = PyLong_FromLong(3); 5718426935SZachary Turner Py_ssize_t original_refcnt = obj->ob_refcnt; 5818426935SZachary Turner EXPECT_LE(1, original_refcnt); 59f8b22f8fSZachary Turner 6018426935SZachary Turner // If we take an owned reference, the refcount should be the same 61f8b22f8fSZachary Turner PythonObject owned_long(PyRefType::Owned, obj); 6218426935SZachary Turner EXPECT_EQ(original_refcnt, owned_long.get()->ob_refcnt); 63f8b22f8fSZachary Turner 6418426935SZachary Turner // Take another reference and verify that the refcount increases by 1 65f8b22f8fSZachary Turner PythonObject strong_ref(owned_long); 6618426935SZachary Turner EXPECT_EQ(original_refcnt + 1, strong_ref.get()->ob_refcnt); 67f8b22f8fSZachary Turner 6818426935SZachary Turner // If we reset the first one, the refcount should be the original value. 69f8b22f8fSZachary Turner owned_long.Reset(); 7018426935SZachary Turner EXPECT_EQ(original_refcnt, strong_ref.get()->ob_refcnt); 71f8b22f8fSZachary Turner } 72f8b22f8fSZachary Turner 73f8b22f8fSZachary Turner TEST_F(PythonDataObjectsTest, TestResetting) 74f8b22f8fSZachary Turner { 75f8b22f8fSZachary Turner PythonDictionary dict(PyInitialValue::Empty); 76f8b22f8fSZachary Turner 77f8b22f8fSZachary Turner PyObject *new_dict = PyDict_New(); 78f8b22f8fSZachary Turner dict.Reset(PyRefType::Owned, new_dict); 79f8b22f8fSZachary Turner EXPECT_EQ(new_dict, dict.get()); 80f8b22f8fSZachary Turner 81f8b22f8fSZachary Turner dict.Reset(PyRefType::Owned, nullptr); 82f8b22f8fSZachary Turner EXPECT_EQ(nullptr, dict.get()); 83f8b22f8fSZachary Turner 84f8b22f8fSZachary Turner dict.Reset(PyRefType::Owned, PyDict_New()); 85f8b22f8fSZachary Turner EXPECT_NE(nullptr, dict.get()); 86f8b22f8fSZachary Turner dict.Reset(); 87f8b22f8fSZachary Turner EXPECT_EQ(nullptr, dict.get()); 88f8b22f8fSZachary Turner } 89f8b22f8fSZachary Turner 90f8b22f8fSZachary Turner TEST_F(PythonDataObjectsTest, TestBorrowedReferences) 91f8b22f8fSZachary Turner { 92f8b22f8fSZachary Turner PythonInteger long_value(PyRefType::Owned, PyLong_FromLong(3)); 9318426935SZachary Turner Py_ssize_t original_refcnt = long_value.get()->ob_refcnt; 9418426935SZachary Turner EXPECT_LE(1, original_refcnt); 95f8b22f8fSZachary Turner 96f8b22f8fSZachary Turner PythonInteger borrowed_long(PyRefType::Borrowed, long_value.get()); 9718426935SZachary Turner EXPECT_EQ(original_refcnt + 1, borrowed_long.get()->ob_refcnt); 98f8b22f8fSZachary Turner } 99f8b22f8fSZachary Turner 1007841efbbSZachary Turner TEST_F(PythonDataObjectsTest, TestGlobalNameResolutionNoDot) 1017841efbbSZachary Turner { 102a1405147SZachary Turner PythonObject sys_module = m_main_module.ResolveName("sys"); 103a1405147SZachary Turner EXPECT_EQ(m_sys_module.get(), sys_module.get()); 1047841efbbSZachary Turner EXPECT_TRUE(sys_module.IsAllocated()); 1057841efbbSZachary Turner EXPECT_TRUE(PythonModule::Check(sys_module.get())); 1067841efbbSZachary Turner } 1077841efbbSZachary Turner 1087841efbbSZachary Turner TEST_F(PythonDataObjectsTest, TestModuleNameResolutionNoDot) 1097841efbbSZachary Turner { 110a1405147SZachary Turner PythonObject sys_path = m_sys_module.ResolveName("path"); 111a1405147SZachary Turner PythonObject sys_version_info = m_sys_module.ResolveName("version_info"); 1127841efbbSZachary Turner EXPECT_TRUE(sys_path.IsAllocated()); 1137841efbbSZachary Turner EXPECT_TRUE(sys_version_info.IsAllocated()); 1147841efbbSZachary Turner 1157841efbbSZachary Turner EXPECT_TRUE(PythonList::Check(sys_path.get())); 1167841efbbSZachary Turner } 1177841efbbSZachary Turner 1187841efbbSZachary Turner TEST_F(PythonDataObjectsTest, TestTypeNameResolutionNoDot) 1197841efbbSZachary Turner { 120a1405147SZachary Turner PythonObject sys_version_info = m_sys_module.ResolveName("version_info"); 1217841efbbSZachary Turner 1227841efbbSZachary Turner PythonObject version_info_type(PyRefType::Owned, PyObject_Type(sys_version_info.get())); 1237841efbbSZachary Turner EXPECT_TRUE(version_info_type.IsAllocated()); 1247841efbbSZachary Turner PythonObject major_version_field = version_info_type.ResolveName("major"); 1257841efbbSZachary Turner EXPECT_TRUE(major_version_field.IsAllocated()); 1267841efbbSZachary Turner } 1277841efbbSZachary Turner 1287841efbbSZachary Turner TEST_F(PythonDataObjectsTest, TestInstanceNameResolutionNoDot) 1297841efbbSZachary Turner { 130a1405147SZachary Turner PythonObject sys_version_info = m_sys_module.ResolveName("version_info"); 1317841efbbSZachary Turner PythonObject major_version_field = sys_version_info.ResolveName("major"); 1327841efbbSZachary Turner PythonObject minor_version_field = sys_version_info.ResolveName("minor"); 1337841efbbSZachary Turner 1347841efbbSZachary Turner EXPECT_TRUE(major_version_field.IsAllocated()); 1357841efbbSZachary Turner EXPECT_TRUE(minor_version_field.IsAllocated()); 1367841efbbSZachary Turner 1377841efbbSZachary Turner PythonInteger major_version_value = major_version_field.AsType<PythonInteger>(); 1387841efbbSZachary Turner PythonInteger minor_version_value = minor_version_field.AsType<PythonInteger>(); 1397841efbbSZachary Turner 1407841efbbSZachary Turner EXPECT_EQ(PY_MAJOR_VERSION, major_version_value.GetInteger()); 1417841efbbSZachary Turner EXPECT_EQ(PY_MINOR_VERSION, minor_version_value.GetInteger()); 1427841efbbSZachary Turner } 1437841efbbSZachary Turner 1447841efbbSZachary Turner TEST_F(PythonDataObjectsTest, TestGlobalNameResolutionWithDot) 1457841efbbSZachary Turner { 146a1405147SZachary Turner PythonObject sys_path = m_main_module.ResolveName("sys.path"); 1477841efbbSZachary Turner EXPECT_TRUE(sys_path.IsAllocated()); 1487841efbbSZachary Turner EXPECT_TRUE(PythonList::Check(sys_path.get())); 1497841efbbSZachary Turner 150a1405147SZachary Turner PythonInteger version_major = m_main_module.ResolveName( 151a1405147SZachary Turner "sys.version_info.major").AsType<PythonInteger>(); 152a1405147SZachary Turner PythonInteger version_minor = m_main_module.ResolveName( 153a1405147SZachary Turner "sys.version_info.minor").AsType<PythonInteger>(); 1547841efbbSZachary Turner EXPECT_TRUE(version_major.IsAllocated()); 1557841efbbSZachary Turner EXPECT_TRUE(version_minor.IsAllocated()); 1567841efbbSZachary Turner EXPECT_EQ(PY_MAJOR_VERSION, version_major.GetInteger()); 1577841efbbSZachary Turner EXPECT_EQ(PY_MINOR_VERSION, version_minor.GetInteger()); 1587841efbbSZachary Turner } 1597841efbbSZachary Turner 160a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestDictionaryResolutionWithDot) 161a1405147SZachary Turner { 162a1405147SZachary Turner // Make up a custom dictionary with "sys" pointing to the `sys` module. 163a1405147SZachary Turner PythonDictionary dict(PyInitialValue::Empty); 164a1405147SZachary Turner dict.SetItemForKey(PythonString("sys"), m_sys_module); 165a1405147SZachary Turner 166a1405147SZachary Turner // Now use that dictionary to resolve `sys.version_info.major` 167a1405147SZachary Turner PythonInteger version_major = PythonObject::ResolveNameWithDictionary( 168a1405147SZachary Turner "sys.version_info.major", dict).AsType<PythonInteger>(); 169a1405147SZachary Turner PythonInteger version_minor = PythonObject::ResolveNameWithDictionary( 170a1405147SZachary Turner "sys.version_info.minor", dict).AsType<PythonInteger>(); 171a1405147SZachary Turner EXPECT_EQ(PY_MAJOR_VERSION, version_major.GetInteger()); 172a1405147SZachary Turner EXPECT_EQ(PY_MINOR_VERSION, version_minor.GetInteger()); 173a1405147SZachary Turner } 174a1405147SZachary Turner 17522c8efcdSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonInteger) 17622c8efcdSZachary Turner { 17722c8efcdSZachary Turner // Test that integers behave correctly when wrapped by a PythonInteger. 17822c8efcdSZachary Turner 17922c8efcdSZachary Turner #if PY_MAJOR_VERSION < 3 18022c8efcdSZachary Turner // Verify that `PythonInt` works correctly when given a PyInt object. 18122c8efcdSZachary Turner // Note that PyInt doesn't exist in Python 3.x, so this is only for 2.x 18222c8efcdSZachary Turner PyObject *py_int = PyInt_FromLong(12); 18322c8efcdSZachary Turner EXPECT_TRUE(PythonInteger::Check(py_int)); 184f8b22f8fSZachary Turner PythonInteger python_int(PyRefType::Owned, py_int); 18522c8efcdSZachary Turner 18622c8efcdSZachary Turner EXPECT_EQ(PyObjectType::Integer, python_int.GetObjectType()); 18722c8efcdSZachary Turner EXPECT_EQ(12, python_int.GetInteger()); 18822c8efcdSZachary Turner #endif 18922c8efcdSZachary Turner 19018426935SZachary Turner // Verify that `PythonInteger` works correctly when given a PyLong object. 19122c8efcdSZachary Turner PyObject *py_long = PyLong_FromLong(12); 19222c8efcdSZachary Turner EXPECT_TRUE(PythonInteger::Check(py_long)); 193f8b22f8fSZachary Turner PythonInteger python_long(PyRefType::Owned, py_long); 19422c8efcdSZachary Turner EXPECT_EQ(PyObjectType::Integer, python_long.GetObjectType()); 19522c8efcdSZachary Turner 19622c8efcdSZachary Turner // Verify that you can reset the value and that it is reflected properly. 19722c8efcdSZachary Turner python_long.SetInteger(40); 19822c8efcdSZachary Turner EXPECT_EQ(40, python_long.GetInteger()); 19918426935SZachary Turner 20018426935SZachary Turner // Test that creating a `PythonInteger` object works correctly with the 20118426935SZachary Turner // int constructor. 20218426935SZachary Turner PythonInteger constructed_int(7); 20318426935SZachary Turner EXPECT_EQ(7, constructed_int.GetInteger()); 20422c8efcdSZachary Turner } 20522c8efcdSZachary Turner 2065a72c02bSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonBytes) 2075a72c02bSZachary Turner { 2085a72c02bSZachary Turner static const char *test_bytes = "PythonDataObjectsTest::TestPythonBytes"; 2095a72c02bSZachary Turner PyObject *py_bytes = PyBytes_FromString(test_bytes); 2105a72c02bSZachary Turner EXPECT_TRUE(PythonBytes::Check(py_bytes)); 2115a72c02bSZachary Turner PythonBytes python_bytes(PyRefType::Owned, py_bytes); 2125a72c02bSZachary Turner EXPECT_EQ(PyObjectType::Bytes, python_bytes.GetObjectType()); 2135a72c02bSZachary Turner 2145a72c02bSZachary Turner #if PY_MAJOR_VERSION < 3 2155a72c02bSZachary Turner EXPECT_TRUE(PythonString::Check(py_bytes)); 2165a72c02bSZachary Turner EXPECT_EQ(PyObjectType::String, python_bytes.GetObjectType()); 2175a72c02bSZachary Turner #else 2185a72c02bSZachary Turner EXPECT_FALSE(PythonString::Check(py_bytes)); 2195a72c02bSZachary Turner EXPECT_NE(PyObjectType::String, python_bytes.GetObjectType()); 2205a72c02bSZachary Turner #endif 2215a72c02bSZachary Turner 2225a72c02bSZachary Turner llvm::ArrayRef<uint8_t> bytes = python_bytes.GetBytes(); 2235a72c02bSZachary Turner EXPECT_EQ(bytes.size(), strlen(test_bytes)); 2245a72c02bSZachary Turner EXPECT_EQ(0, ::memcmp(bytes.data(), test_bytes, bytes.size())); 2255a72c02bSZachary Turner } 2265a72c02bSZachary Turner 227f9d6d204SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonByteArray) 228f9d6d204SZachary Turner { 229f9d6d204SZachary Turner static const char *test_bytes = "PythonDataObjectsTest::TestPythonByteArray"; 230f9d6d204SZachary Turner llvm::StringRef orig_bytes(test_bytes); 231f9d6d204SZachary Turner PyObject *py_bytes = PyByteArray_FromStringAndSize(test_bytes, orig_bytes.size()); 232f9d6d204SZachary Turner EXPECT_TRUE(PythonByteArray::Check(py_bytes)); 233f9d6d204SZachary Turner PythonByteArray python_bytes(PyRefType::Owned, py_bytes); 234f9d6d204SZachary Turner EXPECT_EQ(PyObjectType::ByteArray, python_bytes.GetObjectType()); 235f9d6d204SZachary Turner 236f9d6d204SZachary Turner llvm::ArrayRef<uint8_t> after_bytes = python_bytes.GetBytes(); 237f9d6d204SZachary Turner EXPECT_EQ(after_bytes.size(), orig_bytes.size()); 238f9d6d204SZachary Turner EXPECT_EQ(0, ::memcmp(orig_bytes.data(), test_bytes, orig_bytes.size())); 239f9d6d204SZachary Turner } 240f9d6d204SZachary Turner 24122c8efcdSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonString) 24222c8efcdSZachary Turner { 24322c8efcdSZachary Turner // Test that strings behave correctly when wrapped by a PythonString. 24422c8efcdSZachary Turner 24518426935SZachary Turner static const char *test_string = "PythonDataObjectsTest::TestPythonString1"; 24618426935SZachary Turner static const char *test_string2 = "PythonDataObjectsTest::TestPythonString2"; 24722c8efcdSZachary Turner 24822c8efcdSZachary Turner #if PY_MAJOR_VERSION < 3 24922c8efcdSZachary Turner // Verify that `PythonString` works correctly when given a PyString object. 25022c8efcdSZachary Turner // Note that PyString doesn't exist in Python 3.x, so this is only for 2.x 25122c8efcdSZachary Turner PyObject *py_string = PyString_FromString(test_string); 25222c8efcdSZachary Turner EXPECT_TRUE(PythonString::Check(py_string)); 253f8b22f8fSZachary Turner PythonString python_string(PyRefType::Owned, py_string); 25422c8efcdSZachary Turner 25522c8efcdSZachary Turner EXPECT_EQ(PyObjectType::String, python_string.GetObjectType()); 25622c8efcdSZachary Turner EXPECT_STREQ(test_string, python_string.GetString().data()); 25718426935SZachary Turner #else 25822c8efcdSZachary Turner // Verify that `PythonString` works correctly when given a PyUnicode object. 25922c8efcdSZachary Turner PyObject *py_unicode = PyUnicode_FromString(test_string); 26022c8efcdSZachary Turner EXPECT_TRUE(PythonString::Check(py_unicode)); 261f8b22f8fSZachary Turner PythonString python_unicode(PyRefType::Owned, py_unicode); 26222c8efcdSZachary Turner EXPECT_EQ(PyObjectType::String, python_unicode.GetObjectType()); 26322c8efcdSZachary Turner EXPECT_STREQ(test_string, python_unicode.GetString().data()); 26418426935SZachary Turner #endif 26522c8efcdSZachary Turner 26618426935SZachary Turner // Test that creating a `PythonString` object works correctly with the 26718426935SZachary Turner // string constructor 268*de9dfe78SReid Kleckner PythonString constructed_string(test_string2); 269*de9dfe78SReid Kleckner EXPECT_STREQ(test_string2, constructed_string.GetString().str().c_str()); 27022c8efcdSZachary Turner } 27122c8efcdSZachary Turner 27218426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonStringToStr) 27318426935SZachary Turner { 27418426935SZachary Turner const char *c_str = "PythonDataObjectsTest::TestPythonStringToStr"; 27518426935SZachary Turner 27618426935SZachary Turner PythonString str(c_str); 27718426935SZachary Turner EXPECT_STREQ(c_str, str.GetString().str().c_str()); 27818426935SZachary Turner 27918426935SZachary Turner PythonString str_str = str.Str(); 28018426935SZachary Turner EXPECT_STREQ(c_str, str_str.GetString().str().c_str()); 28118426935SZachary Turner } 28218426935SZachary Turner 28318426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonIntegerToStr) 28418426935SZachary Turner { 28518426935SZachary Turner } 28618426935SZachary Turner 28718426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonIntegerToStructuredInteger) 28818426935SZachary Turner { 28918426935SZachary Turner PythonInteger integer(7); 29018426935SZachary Turner auto int_sp = integer.CreateStructuredInteger(); 291*de9dfe78SReid Kleckner EXPECT_EQ(7U, int_sp->GetValue()); 29218426935SZachary Turner } 29318426935SZachary Turner 29418426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonStringToStructuredString) 29518426935SZachary Turner { 29618426935SZachary Turner static const char *test_string = "PythonDataObjectsTest::TestPythonStringToStructuredString"; 29718426935SZachary Turner PythonString constructed_string(test_string); 29818426935SZachary Turner auto string_sp = constructed_string.CreateStructuredString(); 29918426935SZachary Turner EXPECT_STREQ(test_string, string_sp->GetStringValue().c_str()); 30018426935SZachary Turner } 30118426935SZachary Turner 30218426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonListValueEquality) 30322c8efcdSZachary Turner { 30422c8efcdSZachary Turner // Test that a list which is built through the native 30522c8efcdSZachary Turner // Python API behaves correctly when wrapped by a PythonList. 306*de9dfe78SReid Kleckner static const unsigned list_size = 2; 30718426935SZachary Turner static const long long_value0 = 5; 30818426935SZachary Turner static const char *const string_value1 = "String Index 1"; 30922c8efcdSZachary Turner 31022c8efcdSZachary Turner PyObject *py_list = PyList_New(2); 311f8b22f8fSZachary Turner EXPECT_TRUE(PythonList::Check(py_list)); 312f8b22f8fSZachary Turner PythonList list(PyRefType::Owned, py_list); 313f8b22f8fSZachary Turner 314f8b22f8fSZachary Turner PythonObject list_items[list_size]; 31518426935SZachary Turner list_items[0].Reset(PythonInteger(long_value0)); 31618426935SZachary Turner list_items[1].Reset(PythonString(string_value1)); 31722c8efcdSZachary Turner 318*de9dfe78SReid Kleckner for (unsigned i = 0; i < list_size; ++i) 319f8b22f8fSZachary Turner list.SetItemAtIndex(i, list_items[i]); 32022c8efcdSZachary Turner 32122c8efcdSZachary Turner EXPECT_EQ(list_size, list.GetSize()); 32222c8efcdSZachary Turner EXPECT_EQ(PyObjectType::List, list.GetObjectType()); 32322c8efcdSZachary Turner 32418426935SZachary Turner // Verify that the values match 32518426935SZachary Turner PythonObject chk_value1 = list.GetItemAtIndex(0); 32618426935SZachary Turner PythonObject chk_value2 = list.GetItemAtIndex(1); 32718426935SZachary Turner EXPECT_TRUE(PythonInteger::Check(chk_value1.get())); 32818426935SZachary Turner EXPECT_TRUE(PythonString::Check(chk_value2.get())); 32922c8efcdSZachary Turner 33018426935SZachary Turner PythonInteger chk_int(PyRefType::Borrowed, chk_value1.get()); 33118426935SZachary Turner PythonString chk_str(PyRefType::Borrowed, chk_value2.get()); 33222c8efcdSZachary Turner 33318426935SZachary Turner EXPECT_EQ(long_value0, chk_int.GetInteger()); 33418426935SZachary Turner EXPECT_STREQ(string_value1, chk_str.GetString().str().c_str()); 33522c8efcdSZachary Turner } 33622c8efcdSZachary Turner 33722c8efcdSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonListManipulation) 33822c8efcdSZachary Turner { 33922c8efcdSZachary Turner // Test that manipulation of a PythonList behaves correctly when 34022c8efcdSZachary Turner // wrapped by a PythonDictionary. 34122c8efcdSZachary Turner 34218426935SZachary Turner static const long long_value0 = 5; 34318426935SZachary Turner static const char *const string_value1 = "String Index 1"; 34422c8efcdSZachary Turner 345f8b22f8fSZachary Turner PythonList list(PyInitialValue::Empty); 34618426935SZachary Turner PythonInteger integer(long_value0); 34718426935SZachary Turner PythonString string(string_value1); 34822c8efcdSZachary Turner 34922c8efcdSZachary Turner list.AppendItem(integer); 35022c8efcdSZachary Turner list.AppendItem(string); 351*de9dfe78SReid Kleckner EXPECT_EQ(2U, list.GetSize()); 35222c8efcdSZachary Turner 35318426935SZachary Turner // Verify that the values match 35418426935SZachary Turner PythonObject chk_value1 = list.GetItemAtIndex(0); 35518426935SZachary Turner PythonObject chk_value2 = list.GetItemAtIndex(1); 35618426935SZachary Turner EXPECT_TRUE(PythonInteger::Check(chk_value1.get())); 35718426935SZachary Turner EXPECT_TRUE(PythonString::Check(chk_value2.get())); 35818426935SZachary Turner 35918426935SZachary Turner PythonInteger chk_int(PyRefType::Borrowed, chk_value1.get()); 36018426935SZachary Turner PythonString chk_str(PyRefType::Borrowed, chk_value2.get()); 36118426935SZachary Turner 36218426935SZachary Turner EXPECT_EQ(long_value0, chk_int.GetInteger()); 36318426935SZachary Turner EXPECT_STREQ(string_value1, chk_str.GetString().str().c_str()); 36418426935SZachary Turner } 36518426935SZachary Turner 36618426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonListToStructuredList) 36718426935SZachary Turner { 36818426935SZachary Turner static const long long_value0 = 5; 36918426935SZachary Turner static const char *const string_value1 = "String Index 1"; 37018426935SZachary Turner 37118426935SZachary Turner PythonList list(PyInitialValue::Empty); 37218426935SZachary Turner list.AppendItem(PythonInteger(long_value0)); 37318426935SZachary Turner list.AppendItem(PythonString(string_value1)); 37418426935SZachary Turner 37518426935SZachary Turner auto array_sp = list.CreateStructuredArray(); 37618426935SZachary Turner EXPECT_EQ(StructuredData::Type::eTypeInteger, array_sp->GetItemAtIndex(0)->GetType()); 37718426935SZachary Turner EXPECT_EQ(StructuredData::Type::eTypeString, array_sp->GetItemAtIndex(1)->GetType()); 37818426935SZachary Turner 37918426935SZachary Turner auto int_sp = array_sp->GetItemAtIndex(0)->GetAsInteger(); 38018426935SZachary Turner auto string_sp = array_sp->GetItemAtIndex(1)->GetAsString(); 38118426935SZachary Turner 382*de9dfe78SReid Kleckner EXPECT_EQ(long_value0, long(int_sp->GetValue())); 38318426935SZachary Turner EXPECT_STREQ(string_value1, string_sp->GetValue().c_str()); 38418426935SZachary Turner } 38518426935SZachary Turner 386a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonTupleSize) 387a1405147SZachary Turner { 388a1405147SZachary Turner PythonTuple tuple(PyInitialValue::Empty); 389*de9dfe78SReid Kleckner EXPECT_EQ(0U, tuple.GetSize()); 390a1405147SZachary Turner 391a1405147SZachary Turner tuple = PythonTuple(3); 392*de9dfe78SReid Kleckner EXPECT_EQ(3U, tuple.GetSize()); 393a1405147SZachary Turner } 394a1405147SZachary Turner 395a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonTupleValues) 396a1405147SZachary Turner { 397a1405147SZachary Turner PythonTuple tuple(3); 398a1405147SZachary Turner 399a1405147SZachary Turner PythonInteger int_value(1); 400a1405147SZachary Turner PythonString string_value("Test"); 401a1405147SZachary Turner PythonObject none_value(PyRefType::Borrowed, Py_None); 402a1405147SZachary Turner 403a1405147SZachary Turner tuple.SetItemAtIndex(0, int_value); 404a1405147SZachary Turner tuple.SetItemAtIndex(1, string_value); 405a1405147SZachary Turner tuple.SetItemAtIndex(2, none_value); 406a1405147SZachary Turner 407a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(0).get(), int_value.get()); 408a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(1).get(), string_value.get()); 409a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(2).get(), none_value.get()); 410a1405147SZachary Turner } 411a1405147SZachary Turner 412a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonTupleInitializerList) 413a1405147SZachary Turner { 414a1405147SZachary Turner PythonInteger int_value(1); 415a1405147SZachary Turner PythonString string_value("Test"); 416a1405147SZachary Turner PythonObject none_value(PyRefType::Borrowed, Py_None); 417a1405147SZachary Turner PythonTuple tuple{ int_value, string_value, none_value }; 418*de9dfe78SReid Kleckner EXPECT_EQ(3U, tuple.GetSize()); 419a1405147SZachary Turner 420a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(0).get(), int_value.get()); 421a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(1).get(), string_value.get()); 422a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(2).get(), none_value.get()); 423a1405147SZachary Turner } 424a1405147SZachary Turner 425a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonTupleInitializerList2) 426a1405147SZachary Turner { 427a1405147SZachary Turner PythonInteger int_value(1); 428a1405147SZachary Turner PythonString string_value("Test"); 429a1405147SZachary Turner PythonObject none_value(PyRefType::Borrowed, Py_None); 430a1405147SZachary Turner 431a1405147SZachary Turner PythonTuple tuple{ int_value.get(), string_value.get(), none_value.get() }; 432*de9dfe78SReid Kleckner EXPECT_EQ(3U, tuple.GetSize()); 433a1405147SZachary Turner 434a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(0).get(), int_value.get()); 435a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(1).get(), string_value.get()); 436a1405147SZachary Turner EXPECT_EQ(tuple.GetItemAtIndex(2).get(), none_value.get()); 437a1405147SZachary Turner } 438a1405147SZachary Turner 439a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonTupleToStructuredList) 440a1405147SZachary Turner { 441a1405147SZachary Turner PythonInteger int_value(1); 442a1405147SZachary Turner PythonString string_value("Test"); 443a1405147SZachary Turner 444a1405147SZachary Turner PythonTuple tuple{ int_value.get(), string_value.get() }; 445a1405147SZachary Turner 446a1405147SZachary Turner auto array_sp = tuple.CreateStructuredArray(); 447a1405147SZachary Turner EXPECT_EQ(tuple.GetSize(), array_sp->GetSize()); 448a1405147SZachary Turner EXPECT_EQ(StructuredData::Type::eTypeInteger, array_sp->GetItemAtIndex(0)->GetType()); 449a1405147SZachary Turner EXPECT_EQ(StructuredData::Type::eTypeString, array_sp->GetItemAtIndex(1)->GetType()); 450a1405147SZachary Turner } 451a1405147SZachary Turner 45218426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonDictionaryValueEquality) 45318426935SZachary Turner { 45418426935SZachary Turner // Test that a dictionary which is built through the native 45518426935SZachary Turner // Python API behaves correctly when wrapped by a PythonDictionary. 456*de9dfe78SReid Kleckner static const unsigned dict_entries = 2; 45718426935SZachary Turner const char *key_0 = "Key 0"; 45818426935SZachary Turner int key_1 = 1; 45918426935SZachary Turner const int value_0 = 0; 46018426935SZachary Turner const char *value_1 = "Value 1"; 46118426935SZachary Turner 46218426935SZachary Turner PythonObject py_keys[dict_entries]; 46318426935SZachary Turner PythonObject py_values[dict_entries]; 46418426935SZachary Turner 46518426935SZachary Turner py_keys[0].Reset(PythonString(key_0)); 46618426935SZachary Turner py_keys[1].Reset(PythonInteger(key_1)); 46718426935SZachary Turner py_values[0].Reset(PythonInteger(value_0)); 46818426935SZachary Turner py_values[1].Reset(PythonString(value_1)); 46918426935SZachary Turner 47018426935SZachary Turner PyObject *py_dict = PyDict_New(); 47118426935SZachary Turner EXPECT_TRUE(PythonDictionary::Check(py_dict)); 47218426935SZachary Turner PythonDictionary dict(PyRefType::Owned, py_dict); 47318426935SZachary Turner 474*de9dfe78SReid Kleckner for (unsigned i = 0; i < dict_entries; ++i) 47518426935SZachary Turner PyDict_SetItem(py_dict, py_keys[i].get(), py_values[i].get()); 47618426935SZachary Turner EXPECT_EQ(dict.GetSize(), dict_entries); 47718426935SZachary Turner EXPECT_EQ(PyObjectType::Dictionary, dict.GetObjectType()); 47818426935SZachary Turner 47918426935SZachary Turner // Verify that the values match 48018426935SZachary Turner PythonObject chk_value1 = dict.GetItemForKey(py_keys[0]); 48118426935SZachary Turner PythonObject chk_value2 = dict.GetItemForKey(py_keys[1]); 48218426935SZachary Turner EXPECT_TRUE(PythonInteger::Check(chk_value1.get())); 48318426935SZachary Turner EXPECT_TRUE(PythonString::Check(chk_value2.get())); 48418426935SZachary Turner 48518426935SZachary Turner PythonInteger chk_int(PyRefType::Borrowed, chk_value1.get()); 48618426935SZachary Turner PythonString chk_str(PyRefType::Borrowed, chk_value2.get()); 48718426935SZachary Turner 48818426935SZachary Turner EXPECT_EQ(value_0, chk_int.GetInteger()); 48918426935SZachary Turner EXPECT_STREQ(value_1, chk_str.GetString().str().c_str()); 49022c8efcdSZachary Turner } 49122c8efcdSZachary Turner 49222c8efcdSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonDictionaryManipulation) 49322c8efcdSZachary Turner { 49422c8efcdSZachary Turner // Test that manipulation of a dictionary behaves correctly when wrapped 49522c8efcdSZachary Turner // by a PythonDictionary. 496*de9dfe78SReid Kleckner static const unsigned dict_entries = 2; 49722c8efcdSZachary Turner 49818426935SZachary Turner const char *const key_0 = "Key 0"; 49918426935SZachary Turner const char *const key_1 = "Key 1"; 50018426935SZachary Turner const long value_0 = 1; 50118426935SZachary Turner const char *const value_1 = "Value 1"; 50218426935SZachary Turner 503f8b22f8fSZachary Turner PythonString keys[dict_entries]; 504f8b22f8fSZachary Turner PythonObject values[dict_entries]; 50522c8efcdSZachary Turner 50618426935SZachary Turner keys[0].Reset(PythonString(key_0)); 50718426935SZachary Turner keys[1].Reset(PythonString(key_1)); 50818426935SZachary Turner values[0].Reset(PythonInteger(value_0)); 50918426935SZachary Turner values[1].Reset(PythonString(value_1)); 51022c8efcdSZachary Turner 511f8b22f8fSZachary Turner PythonDictionary dict(PyInitialValue::Empty); 51222c8efcdSZachary Turner for (int i = 0; i < 2; ++i) 513f8b22f8fSZachary Turner dict.SetItemForKey(keys[i], values[i]); 51422c8efcdSZachary Turner 51522c8efcdSZachary Turner EXPECT_EQ(dict_entries, dict.GetSize()); 51622c8efcdSZachary Turner 51718426935SZachary Turner // Verify that the keys and values match 51818426935SZachary Turner PythonObject chk_value1 = dict.GetItemForKey(keys[0]); 51918426935SZachary Turner PythonObject chk_value2 = dict.GetItemForKey(keys[1]); 52018426935SZachary Turner EXPECT_TRUE(PythonInteger::Check(chk_value1.get())); 52118426935SZachary Turner EXPECT_TRUE(PythonString::Check(chk_value2.get())); 52218426935SZachary Turner 52318426935SZachary Turner PythonInteger chk_int(PyRefType::Borrowed, chk_value1.get()); 52418426935SZachary Turner PythonString chk_str(PyRefType::Borrowed, chk_value2.get()); 52518426935SZachary Turner 52618426935SZachary Turner EXPECT_EQ(value_0, chk_int.GetInteger()); 52718426935SZachary Turner EXPECT_STREQ(value_1, chk_str.GetString().str().c_str()); 52822c8efcdSZachary Turner } 52922c8efcdSZachary Turner 53018426935SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonDictionaryToStructuredDictionary) 53122c8efcdSZachary Turner { 53218426935SZachary Turner static const char *const string_key0 = "String Key 0"; 53318426935SZachary Turner static const char *const string_key1 = "String Key 1"; 53422c8efcdSZachary Turner 53518426935SZachary Turner static const char *const string_value0 = "String Value 0"; 53618426935SZachary Turner static const long int_value1 = 7; 53722c8efcdSZachary Turner 538f8b22f8fSZachary Turner PythonDictionary dict(PyInitialValue::Empty); 53918426935SZachary Turner dict.SetItemForKey(PythonString(string_key0), PythonString(string_value0)); 54018426935SZachary Turner dict.SetItemForKey(PythonString(string_key1), PythonInteger(int_value1)); 54122c8efcdSZachary Turner 54218426935SZachary Turner auto dict_sp = dict.CreateStructuredDictionary(); 543*de9dfe78SReid Kleckner EXPECT_EQ(2U, dict_sp->GetSize()); 54422c8efcdSZachary Turner 54518426935SZachary Turner EXPECT_TRUE(dict_sp->HasKey(string_key0)); 54618426935SZachary Turner EXPECT_TRUE(dict_sp->HasKey(string_key1)); 54722c8efcdSZachary Turner 54818426935SZachary Turner auto string_sp = dict_sp->GetValueForKey(string_key0)->GetAsString(); 54918426935SZachary Turner auto int_sp = dict_sp->GetValueForKey(string_key1)->GetAsInteger(); 55022c8efcdSZachary Turner 55118426935SZachary Turner EXPECT_STREQ(string_value0, string_sp->GetValue().c_str()); 552*de9dfe78SReid Kleckner EXPECT_EQ(int_value1, long(int_sp->GetValue())); 55322c8efcdSZachary Turner } 5549c40264fSZachary Turner 555a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonCallableCheck) 556a1405147SZachary Turner { 557a1405147SZachary Turner PythonObject sys_exc_info = m_sys_module.ResolveName("exc_info"); 558a1405147SZachary Turner PythonObject none(PyRefType::Borrowed, Py_None); 559a1405147SZachary Turner 560a1405147SZachary Turner EXPECT_TRUE(PythonCallable::Check(sys_exc_info.get())); 561a1405147SZachary Turner EXPECT_FALSE(PythonCallable::Check(none.get())); 562a1405147SZachary Turner } 563a1405147SZachary Turner 564a1405147SZachary Turner TEST_F(PythonDataObjectsTest, TestPythonCallableInvoke) 565a1405147SZachary Turner { 566a1405147SZachary Turner auto list = m_builtins_module.ResolveName("list").AsType<PythonCallable>(); 567a1405147SZachary Turner PythonInteger one(1); 568a1405147SZachary Turner PythonString two("two"); 569a1405147SZachary Turner PythonTuple three = { one, two }; 570a1405147SZachary Turner 571a1405147SZachary Turner PythonTuple tuple_to_convert = { one, two, three }; 572a1405147SZachary Turner PythonObject result = list({ tuple_to_convert }); 573a1405147SZachary Turner 574a1405147SZachary Turner EXPECT_TRUE(PythonList::Check(result.get())); 575a1405147SZachary Turner auto list_result = result.AsType<PythonList>(); 576*de9dfe78SReid Kleckner EXPECT_EQ(3U, list_result.GetSize()); 577a1405147SZachary Turner EXPECT_EQ(one.get(), list_result.GetItemAtIndex(0).get()); 578a1405147SZachary Turner EXPECT_EQ(two.get(), list_result.GetItemAtIndex(1).get()); 579a1405147SZachary Turner EXPECT_EQ(three.get(), list_result.GetItemAtIndex(2).get()); 580a1405147SZachary Turner } 581a1405147SZachary Turner 5829c40264fSZachary Turner TEST_F(PythonDataObjectsTest, TestPythonFile) 5839c40264fSZachary Turner { 5849c40264fSZachary Turner File file(FileSystem::DEV_NULL, File::eOpenOptionRead); 5859c40264fSZachary Turner PythonFile py_file(file, "r"); 5869c40264fSZachary Turner EXPECT_TRUE(PythonFile::Check(py_file.get())); 5879c40264fSZachary Turner } 5887d6d218eSZachary Turner 5897d6d218eSZachary Turner TEST_F(PythonDataObjectsTest, TestObjectAttributes) 5907d6d218eSZachary Turner { 5917d6d218eSZachary Turner PythonInteger py_int(42); 5927d6d218eSZachary Turner EXPECT_TRUE(py_int.HasAttribute("numerator")); 5937d6d218eSZachary Turner EXPECT_FALSE(py_int.HasAttribute("this_should_not_exist")); 5947d6d218eSZachary Turner 5957d6d218eSZachary Turner PythonInteger numerator_attr = py_int.GetAttributeValue("numerator").AsType<PythonInteger>(); 5967d6d218eSZachary Turner EXPECT_TRUE(numerator_attr.IsAllocated()); 5977d6d218eSZachary Turner EXPECT_EQ(42, numerator_attr.GetInteger()); 5987d6d218eSZachary Turner } 599