test_fragments.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831
  1. #!/usr/bin/env python
  2. #
  3. # SPDX-FileCopyrightText: 2021-2022 Espressif Systems (Shanghai) CO LTD
  4. # SPDX-License-Identifier: Apache-2.0
  5. #
  6. import os
  7. import sys
  8. import tempfile
  9. import unittest
  10. from io import StringIO
  11. try:
  12. from ldgen.fragments import Align, Flag, Keep, Sort, Surround, parse_fragment_file
  13. from ldgen.sdkconfig import SDKConfig
  14. except ImportError:
  15. sys.path.append(os.path.dirname(os.path.dirname(__file__)))
  16. from ldgen.fragments import Align, Flag, Keep, Sort, Surround, parse_fragment_file
  17. from ldgen.sdkconfig import SDKConfig
  18. from pyparsing import ParseException, ParseFatalException
  19. class FragmentTest(unittest.TestCase):
  20. def setUp(self):
  21. with tempfile.NamedTemporaryFile(delete=False) as f:
  22. self.kconfigs_source_file = os.path.join(tempfile.gettempdir(), f.name)
  23. with tempfile.NamedTemporaryFile(delete=False) as f:
  24. self.kconfig_projbuilds_source_file = os.path.join(tempfile.gettempdir(), f.name)
  25. os.environ['COMPONENT_KCONFIGS_SOURCE_FILE'] = self.kconfigs_source_file
  26. os.environ['COMPONENT_KCONFIGS_PROJBUILD_SOURCE_FILE'] = self.kconfig_projbuilds_source_file
  27. os.environ['COMPONENT_KCONFIGS'] = ''
  28. os.environ['COMPONENT_KCONFIGS_PROJBUILD'] = ''
  29. # prepare_kconfig_files.py doesn't have to be called because COMPONENT_KCONFIGS and
  30. # COMPONENT_KCONFIGS_PROJBUILD are empty
  31. self.sdkconfig = SDKConfig('data/Kconfig', 'data/sdkconfig')
  32. def tearDown(self):
  33. try:
  34. os.remove(self.kconfigs_source_file)
  35. os.remove(self.kconfig_projbuilds_source_file)
  36. except Exception:
  37. pass
  38. @staticmethod
  39. def create_fragment_file(contents, name='test_fragment.lf'):
  40. f = StringIO(contents)
  41. f.name = name
  42. return f
  43. def test_basic(self):
  44. test_fragment = self.create_fragment_file(u"""
  45. [sections:test]
  46. entries:
  47. value_1
  48. value_2 # comments should be ignored
  49. value_3
  50. # this is a comment as well
  51. value_a
  52. # this is the last comment
  53. """)
  54. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  55. self.assertEqual(fragment_file.fragments[0].name, 'test')
  56. self.assertEqual(fragment_file.fragments[0].entries, {'value_1', 'value_2', 'value_3', 'value_a'})
  57. def test_conditional(self):
  58. test_fragment = self.create_fragment_file(u"""
  59. [sections:test]
  60. entries:
  61. value_1
  62. if A = y:
  63. value_2
  64. value_3
  65. if A = n:
  66. value_4
  67. if B = n:
  68. value_5
  69. """)
  70. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  71. self.assertEqual(fragment_file.fragments[0].name, 'test')
  72. self.assertEqual(fragment_file.fragments[0].entries, {'value_1', 'value_2', 'value_3', 'value_5'})
  73. test_fragment = self.create_fragment_file(u"""
  74. [sections:test]
  75. entries:
  76. value_1
  77. if B = y:
  78. value_2
  79. elif C = y:
  80. value_3
  81. elif A = y:
  82. value_4
  83. else:
  84. value_5
  85. value_6
  86. """)
  87. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  88. self.assertEqual(fragment_file.fragments[0].name, 'test')
  89. self.assertEqual(fragment_file.fragments[0].entries, {'value_1', 'value_3', 'value_6'})
  90. test_fragment = self.create_fragment_file(u"""
  91. [sections:test]
  92. entries:
  93. value_1
  94. if A = y:
  95. value_2
  96. if B = y:
  97. value_3
  98. else:
  99. value_4
  100. if C = y:
  101. value_5
  102. value_6
  103. value_7
  104. """)
  105. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  106. self.assertEqual(fragment_file.fragments[0].name, 'test')
  107. self.assertEqual(fragment_file.fragments[0].entries,
  108. {'value_1', 'value_2', 'value_4', 'value_5', 'value_6', 'value_7'})
  109. test_fragment = self.create_fragment_file(u"""
  110. [sections:test]
  111. entries:
  112. if A = n:
  113. value_2
  114. """)
  115. with self.assertRaises(ParseFatalException):
  116. parse_fragment_file(test_fragment, self.sdkconfig)
  117. def test_empty_file(self):
  118. test_fragment = self.create_fragment_file(u"""
  119. """)
  120. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  121. self.assertEqual(len(fragment_file.fragments), 0)
  122. def test_setting_indent(self):
  123. test_fragment = self.create_fragment_file(u"""
  124. [sections:test]
  125. entries:
  126. value_1
  127. value_2
  128. value_3
  129. """)
  130. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  131. self.assertEqual(fragment_file.fragments[0].name, 'test')
  132. self.assertEqual(fragment_file.fragments[0].entries, {'value_1', 'value_2', 'value_3'})
  133. def test_settings_unmatch_indent(self):
  134. test_fragment = self.create_fragment_file(u"""
  135. [sections:test]
  136. entries:
  137. value_1
  138. value_2 # first element dictates indent
  139. value_3
  140. """)
  141. with self.assertRaises(ParseException):
  142. parse_fragment_file(test_fragment, self.sdkconfig)
  143. def test_unsupported_key(self):
  144. test_fragment = self.create_fragment_file(u"""
  145. [sections:test]
  146. key_1:
  147. value_a
  148. """)
  149. with self.assertRaises(ParseException):
  150. parse_fragment_file(test_fragment, self.sdkconfig)
  151. def test_empty_fragment(self):
  152. test_fragment = self.create_fragment_file(u"""
  153. [sections:test]
  154. """)
  155. with self.assertRaises(ParseException):
  156. parse_fragment_file(test_fragment, self.sdkconfig)
  157. def test_empty_conditional(self):
  158. test_fragment = self.create_fragment_file(u"""
  159. [sections:test]
  160. entries:
  161. if B = y:
  162. else:
  163. value_1
  164. """)
  165. with self.assertRaises(ParseException):
  166. parse_fragment_file(test_fragment, self.sdkconfig)
  167. test_fragment = self.create_fragment_file(u"""
  168. [sections:test]
  169. entries:
  170. if B = y:
  171. value_1
  172. else B = y:
  173. """)
  174. with self.assertRaises(ParseFatalException):
  175. parse_fragment_file(test_fragment, self.sdkconfig)
  176. test_fragment = self.create_fragment_file(u"""
  177. [sections:test]
  178. entries:
  179. if B = y:
  180. value_1
  181. elif B = y:
  182. else:
  183. value_2
  184. """)
  185. with self.assertRaises(ParseFatalException):
  186. parse_fragment_file(test_fragment, self.sdkconfig)
  187. def test_out_of_order_conditional(self):
  188. test_fragment = self.create_fragment_file(u"""
  189. [sections:test]
  190. entries:
  191. elif B = y:
  192. value_1
  193. else:
  194. value_2
  195. """)
  196. with self.assertRaises(ParseException):
  197. parse_fragment_file(test_fragment, self.sdkconfig)
  198. test_fragment = self.create_fragment_file(u"""
  199. [sections:test]
  200. entries:
  201. else:
  202. value_2
  203. """)
  204. with self.assertRaises(ParseException):
  205. parse_fragment_file(test_fragment, self.sdkconfig)
  206. def test_multiple_fragments(self):
  207. test_fragment = self.create_fragment_file(u"""
  208. [sections:test1]
  209. entries:
  210. value_1
  211. [scheme:test2]
  212. entries:
  213. section -> target
  214. """)
  215. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  216. self.assertEqual(fragment_file.fragments[0].name, 'test1')
  217. self.assertEqual(fragment_file.fragments[0].entries, {'value_1'})
  218. self.assertEqual(fragment_file.fragments[1].name, 'test2')
  219. self.assertEqual(fragment_file.fragments[1].entries, {('section', 'target')})
  220. def test_whole_conditional_fragment(self):
  221. test_fragment = self.create_fragment_file(u"""
  222. if B = y:
  223. [sections:test1]
  224. entries:
  225. value_1
  226. else:
  227. [sections:test2]
  228. entries:
  229. value_2
  230. if A = y:
  231. [sections:test3]
  232. entries:
  233. value_3
  234. if C = y:
  235. value_6
  236. [sections:test4]
  237. entries:
  238. value_4
  239. [sections:test5]
  240. entries:
  241. value_5
  242. """)
  243. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  244. self.assertEqual(len(fragment_file.fragments), 4)
  245. self.assertEqual(fragment_file.fragments[0].name, 'test2')
  246. self.assertEqual(fragment_file.fragments[0].entries, {'value_2'})
  247. self.assertEqual(fragment_file.fragments[1].name, 'test3')
  248. self.assertEqual(fragment_file.fragments[1].entries, {'value_3', 'value_6'})
  249. self.assertEqual(fragment_file.fragments[2].name, 'test4')
  250. self.assertEqual(fragment_file.fragments[2].entries, {'value_4'})
  251. self.assertEqual(fragment_file.fragments[3].name, 'test5')
  252. self.assertEqual(fragment_file.fragments[3].entries, {'value_5'})
  253. def test_equivalent_conditional_fragment(self):
  254. test_fragment1 = self.create_fragment_file(u"""
  255. if A = y:
  256. [sections:test1]
  257. entries:
  258. value_1
  259. else:
  260. [sections:test2]
  261. entries:
  262. value_2
  263. """)
  264. fragment_file1 = parse_fragment_file(test_fragment1, self.sdkconfig)
  265. self.assertEqual(fragment_file1.fragments[0].name, 'test1')
  266. self.assertEqual(fragment_file1.fragments[0].entries, {'value_1'})
  267. test_fragment2 = self.create_fragment_file(u"""
  268. [sections:test1]
  269. entries:
  270. if A = y:
  271. value_1
  272. else:
  273. value_2
  274. """)
  275. fragment_file2 = parse_fragment_file(test_fragment2, self.sdkconfig)
  276. self.assertEqual(fragment_file2.fragments[0].name, 'test1')
  277. self.assertEqual(fragment_file2.fragments[0].entries, {'value_1'})
  278. class SectionsTest(FragmentTest):
  279. def test_basic(self):
  280. test_fragment = self.create_fragment_file(u"""
  281. [sections:test]
  282. entries:
  283. .section1
  284. .section2
  285. """)
  286. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  287. self.assertEqual(fragment_file.fragments[0].entries, {'.section1', '.section2'})
  288. def test_duplicate_entries(self):
  289. test_fragment = self.create_fragment_file(u"""
  290. [sections:test]
  291. entries:
  292. .section1
  293. .section2
  294. .section3
  295. .section2
  296. """)
  297. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  298. self.assertEqual(fragment_file.fragments[0].entries, {'.section1', '.section2', '.section3'})
  299. def test_empty_entries(self):
  300. test_fragment = self.create_fragment_file(u"""
  301. [sections:test]
  302. entries:
  303. """)
  304. with self.assertRaises(ParseException):
  305. parse_fragment_file(test_fragment, self.sdkconfig)
  306. test_fragment = self.create_fragment_file(u"""
  307. [sections:test]
  308. entries:
  309. if B = y:
  310. .section1
  311. """)
  312. with self.assertRaises(ParseFatalException):
  313. parse_fragment_file(test_fragment, self.sdkconfig)
  314. def test_entries_grammar(self):
  315. test_fragment = self.create_fragment_file(u"""
  316. [sections:test]
  317. entries:
  318. _valid1
  319. valid2.
  320. .valid3_-
  321. """)
  322. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  323. self.assertEqual(fragment_file.fragments[0].entries,
  324. {'_valid1', 'valid2.', '.valid3_-'})
  325. # invalid starting char
  326. test_fragment = self.create_fragment_file(u"""
  327. [sections:test]
  328. entries:
  329. 1invalid
  330. """)
  331. with self.assertRaises(ParseException):
  332. parse_fragment_file(test_fragment, self.sdkconfig)
  333. test_fragment = self.create_fragment_file(u"""
  334. [sections:test]
  335. entries:
  336. -invalid
  337. """)
  338. with self.assertRaises(ParseException):
  339. parse_fragment_file(test_fragment, self.sdkconfig)
  340. # + notation
  341. test_fragment = self.create_fragment_file(u"""
  342. [sections:test]
  343. entries:
  344. valid+
  345. """)
  346. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  347. self.assertEqual(fragment_file.fragments[0].entries,
  348. {'valid+'})
  349. test_fragment = self.create_fragment_file(u"""
  350. [sections:test]
  351. entries:
  352. inva+lid+
  353. """)
  354. with self.assertRaises(ParseException):
  355. parse_fragment_file(test_fragment, self.sdkconfig)
  356. class SchemeTest(FragmentTest):
  357. def test_basic(self):
  358. test_fragment = self.create_fragment_file(u"""
  359. [scheme:test]
  360. entries:
  361. sections1 -> target1
  362. sections2 -> target2
  363. """)
  364. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  365. self.assertEqual(fragment_file.fragments[0].entries,
  366. {('sections1', 'target1'),
  367. ('sections2', 'target2')})
  368. def test_duplicate_entries(self):
  369. test_fragment = self.create_fragment_file(u"""
  370. [scheme:test]
  371. entries:
  372. sections1 -> target1
  373. sections2 -> target2
  374. sections2 -> target2
  375. """)
  376. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  377. self.assertEqual(fragment_file.fragments[0].entries,
  378. {('sections1', 'target1'),
  379. ('sections2', 'target2')})
  380. def test_empty_entries(self):
  381. test_fragment = self.create_fragment_file(u"""
  382. [scheme:test]
  383. entries:
  384. """)
  385. with self.assertRaises(ParseException):
  386. parse_fragment_file(test_fragment, self.sdkconfig)
  387. test_fragment = self.create_fragment_file(u"""
  388. [scheme:test]
  389. entries:
  390. if B = y:
  391. sections1 -> target1
  392. """)
  393. with self.assertRaises(ParseFatalException):
  394. parse_fragment_file(test_fragment, self.sdkconfig)
  395. def test_improper_grammar(self):
  396. test_fragment = self.create_fragment_file(u"""
  397. [scheme:test]
  398. entries:
  399. sections1, target1 # improper separator
  400. """)
  401. with self.assertRaises(ParseException):
  402. parse_fragment_file(test_fragment, self.sdkconfig)
  403. class MappingTest(FragmentTest):
  404. def test_basic(self):
  405. test_fragment = self.create_fragment_file(u"""
  406. [mapping:test]
  407. archive: lib.a
  408. entries:
  409. obj:symbol (noflash)
  410. obj (noflash)
  411. obj:symbol_2 (noflash)
  412. obj_2 (noflash)
  413. * (noflash)
  414. """)
  415. expected = {('obj', 'symbol', 'noflash'),
  416. ('obj', None, 'noflash'),
  417. ('obj', 'symbol_2', 'noflash'),
  418. ('obj_2', None, 'noflash'),
  419. ('*', None, 'noflash')}
  420. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  421. self.assertEqual(expected, fragment_file.fragments[0].entries)
  422. def test_archive(self):
  423. test_fragment = self.create_fragment_file(u"""
  424. [mapping:test]
  425. archive:
  426. entries:
  427. * (default)
  428. """)
  429. with self.assertRaises(ParseException):
  430. parse_fragment_file(test_fragment, self.sdkconfig)
  431. test_fragment = self.create_fragment_file(u"""
  432. [mapping:test]
  433. archive:
  434. lib1.a
  435. lib2.a
  436. entries:
  437. * (default)
  438. """)
  439. with self.assertRaises(ParseFatalException):
  440. parse_fragment_file(test_fragment, self.sdkconfig)
  441. def test_archive_allowed_names(self):
  442. test_fragment = self.create_fragment_file(u"""
  443. [mapping:test]
  444. archive:
  445. libstdc++.a
  446. entries:
  447. * (default)
  448. """)
  449. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  450. self.assertEqual('libstdc++.a', fragment_file.fragments[0].archive)
  451. def test_empty_entries(self):
  452. test_fragment = self.create_fragment_file(u"""
  453. [mapping:test]
  454. archive:
  455. lib.a
  456. entries:
  457. if B = y:
  458. * (noflash) # if condition is false, then no 'entries' key value
  459. """)
  460. expected = set()
  461. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  462. self.assertEqual(expected, fragment_file.fragments[0].entries)
  463. test_fragment = self.create_fragment_file(u"""
  464. [mapping:test]
  465. archive:
  466. lib.a
  467. entries:
  468. """)
  469. with self.assertRaises(ParseException):
  470. parse_fragment_file(test_fragment, self.sdkconfig)
  471. def test_duplicate_entries(self):
  472. test_fragment = self.create_fragment_file(u"""
  473. [mapping:test]
  474. archive:
  475. lib.a
  476. entries:
  477. obj:symbol (noflash)
  478. obj:symbol (noflash)
  479. """)
  480. expected = {('obj', 'symbol', 'noflash')}
  481. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  482. self.assertEqual(expected, fragment_file.fragments[0].entries)
  483. def test_invalid_grammar(self):
  484. test_fragment = self.create_fragment_file(u"""
  485. [mapping:test]
  486. archive:
  487. lib.a
  488. """)
  489. with self.assertRaises(ParseException):
  490. parse_fragment_file(test_fragment, self.sdkconfig)
  491. test_fragment = self.create_fragment_file(u"""
  492. [mapping:test]
  493. entries:
  494. * (default)
  495. """)
  496. with self.assertRaises(ParseException):
  497. parse_fragment_file(test_fragment, self.sdkconfig)
  498. test_fragment = self.create_fragment_file(u"""
  499. [mapping:test]
  500. archive: lib.a
  501. entries:
  502. obj: (noflash)
  503. """)
  504. with self.assertRaises(ParseException):
  505. parse_fragment_file(test_fragment, self.sdkconfig)
  506. test_fragment = self.create_fragment_file(u"""
  507. [mapping:test]
  508. archive: lib.a
  509. entries:
  510. obj: ()
  511. """)
  512. with self.assertRaises(ParseException):
  513. parse_fragment_file(test_fragment, self.sdkconfig)
  514. test_fragment = self.create_fragment_file(u"""
  515. [mapping:test]
  516. archive: lib.a
  517. entries:
  518. obj:symbol
  519. """)
  520. with self.assertRaises(ParseException):
  521. parse_fragment_file(test_fragment, self.sdkconfig)
  522. test_fragment = self.create_fragment_file(u"""
  523. [mapping:test]
  524. archive: lib.a
  525. entries:
  526. (noflash)
  527. """)
  528. with self.assertRaises(ParseException):
  529. parse_fragment_file(test_fragment, self.sdkconfig)
  530. test_fragment = self.create_fragment_file(u"""
  531. [mapping:test]
  532. archive: lib.a
  533. entries:
  534. obj:* (noflash)
  535. """)
  536. with self.assertRaises(ParseException):
  537. parse_fragment_file(test_fragment, self.sdkconfig)
  538. test_fragment = self.create_fragment_file(u"""
  539. [mapping:test]
  540. archive: lib.a
  541. entries:
  542. :symbol (noflash)
  543. """)
  544. with self.assertRaises(ParseException):
  545. parse_fragment_file(test_fragment, self.sdkconfig)
  546. test_fragment = self.create_fragment_file(u"""
  547. [mapping:test]
  548. archive: lib.a
  549. entries:
  550. *:symbol (noflash)
  551. """)
  552. with self.assertRaises(ParseException):
  553. parse_fragment_file(test_fragment, self.sdkconfig)
  554. def test_keep_flag(self):
  555. # Test parsing combinations and orders of flags
  556. test_fragment = self.create_fragment_file(u"""
  557. [mapping:map]
  558. archive: libmain.a
  559. entries:
  560. obj1 (default);
  561. text->flash_text KEEP(),
  562. rodata->flash_rodata KEEP() KEEP()
  563. """)
  564. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  565. fragment = fragment_file.fragments[0]
  566. expected = [Flag('text', 'flash_text', [Keep()]),
  567. Flag('rodata', 'flash_rodata', [Keep(), Keep()])]
  568. actual = fragment.flags[('obj1', None, 'default')]
  569. self.assertEqual(expected, actual)
  570. def test_align_flag(self):
  571. # Test parsing combinations and orders of flags
  572. test_fragment = self.create_fragment_file(u"""
  573. [mapping:map]
  574. archive: libmain.a
  575. entries:
  576. obj1 (default);
  577. text->flash_text ALIGN(8),
  578. rodata->flash_rodata ALIGN(8, pre),
  579. data->dram0_data ALIGN(8, pre, post),
  580. bss->dram0_bss ALIGN(8, post),
  581. common->dram0_bss ALIGN(8, pre, post) ALIGN(8)
  582. """)
  583. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  584. fragment = fragment_file.fragments[0]
  585. expected = [Flag('text', 'flash_text', [Align(8, True, False)]),
  586. Flag('rodata', 'flash_rodata', [Align(8, True, False)]),
  587. Flag('data', 'dram0_data', [Align(8, True, True)]),
  588. Flag('bss', 'dram0_bss', [Align(8, False, True)]),
  589. Flag('common', 'dram0_bss', [Align(8, True, True), Align(8, True, False)])]
  590. actual = fragment.flags[('obj1', None, 'default')]
  591. self.assertEqual(expected, actual)
  592. # Wrong post, pre order
  593. test_fragment = self.create_fragment_file(u"""
  594. [mapping:map]
  595. archive: libmain.a
  596. entries:
  597. obj1 (noflash)
  598. text->iram0_text ALIGN(8, post, pre)
  599. """)
  600. with self.assertRaises(ParseException):
  601. parse_fragment_file(test_fragment, self.sdkconfig)
  602. def test_sort_flag(self):
  603. # Test parsing combinations and orders of flags
  604. test_fragment = self.create_fragment_file(u"""
  605. [mapping:map]
  606. archive: libmain.a
  607. entries:
  608. obj1 (default);
  609. text->flash_text SORT(name),
  610. rodata->flash_rodata SORT(alignment),
  611. data->dram0_data SORT(init_priority),
  612. bss->dram0_bss SORT(name, alignment),
  613. common->dram0_bss SORT(alignment, name),
  614. iram->iram0_text SORT(name, name),
  615. dram->dram0_data SORT(alignment, alignment)
  616. """)
  617. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  618. fragment = fragment_file.fragments[0]
  619. expected = [Flag('text', 'flash_text', [Sort('name')]),
  620. Flag('rodata', 'flash_rodata', [Sort('alignment')]),
  621. Flag('data', 'dram0_data', [Sort('init_priority')]),
  622. Flag('bss', 'dram0_bss', [Sort('name', 'alignment')]),
  623. Flag('common', 'dram0_bss', [Sort('alignment', 'name')]),
  624. Flag('iram', 'iram0_text', [Sort('name', 'name')]),
  625. Flag('dram', 'dram0_data', [Sort('alignment', 'alignment')])]
  626. actual = fragment.flags[('obj1', None, 'default')]
  627. self.assertEqual(expected, actual)
  628. def test_surround_flag(self):
  629. # Test parsing combinations and orders of flags
  630. test_fragment = self.create_fragment_file(u"""
  631. [mapping:map]
  632. archive: libmain.a
  633. entries:
  634. obj1 (default);
  635. text->flash_text SURROUND(sym1)
  636. """)
  637. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  638. fragment = fragment_file.fragments[0]
  639. expected = [Flag('text', 'flash_text', [Surround('sym1')])]
  640. actual = fragment.flags[('obj1', None, 'default')]
  641. self.assertEqual(expected, actual)
  642. def test_flag_order(self):
  643. # Test that the order in which the flags are specified is retained
  644. test_fragment = self.create_fragment_file(u"""
  645. [mapping:map]
  646. archive: libmain.a
  647. entries:
  648. obj1 (default);
  649. text->flash_text ALIGN(4) KEEP() SURROUND(sym1) ALIGN(8) SORT(name),
  650. rodata->flash_rodata KEEP() ALIGN(4) KEEP() SURROUND(sym1) ALIGN(8) ALIGN(4) SORT(name)
  651. """)
  652. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  653. fragment = fragment_file.fragments[0]
  654. expected = [Flag('text', 'flash_text', [Align(4, True, False),
  655. Keep(),
  656. Surround('sym1'),
  657. Align(8, True, False),
  658. Sort('name')]),
  659. Flag('rodata', 'flash_rodata', [Keep(),
  660. Align(4, True, False),
  661. Keep(),
  662. Surround('sym1'),
  663. Align(8, True, False),
  664. Align(4, True, False),
  665. Sort('name')])]
  666. actual = fragment.flags[('obj1', None, 'default')]
  667. self.assertEqual(expected, actual)
  668. def test_flags_entries_multiple_flags(self):
  669. # Not an error, generation step handles this, since
  670. # it that step has a more complete information
  671. # about all mappings.
  672. test_fragment = self.create_fragment_file(u"""
  673. [mapping:map]
  674. archive: libmain.a
  675. entries:
  676. obj1 (default);
  677. text->flash_text ALIGN(4) KEEP() SURROUND(sym1) SORT(name),
  678. text->flash_text ALIGN(4) KEEP() SURROUND(sym1) SORT(name)
  679. """)
  680. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  681. fragment = fragment_file.fragments[0]
  682. expected = [Flag('text', 'flash_text', [Align(4, True, False),
  683. Keep(),
  684. Surround('sym1'),
  685. Sort('name')]),
  686. Flag('text', 'flash_text', [Align(4, True, False),
  687. Keep(),
  688. Surround('sym1'),
  689. Sort('name')])]
  690. actual = fragment.flags[('obj1', None, 'default')]
  691. self.assertEqual(expected, actual)
  692. def test_flags_entries_multiple_flags_and_entries(self):
  693. # Not an error, generation step handles this, since
  694. # it that step has a more complete information
  695. # about all mappings. This can happen across multiple
  696. # mapping fragments.
  697. test_fragment = self.create_fragment_file(u"""
  698. [mapping:map]
  699. archive: libmain.a
  700. entries:
  701. obj1 (default);
  702. text->flash_text ALIGN(4) KEEP() SURROUND(sym1) SORT(name)
  703. obj1 (default);
  704. text->flash_text ALIGN(4) KEEP() SURROUND(sym1) SORT(name)
  705. """)
  706. fragment_file = parse_fragment_file(test_fragment, self.sdkconfig)
  707. fragment = fragment_file.fragments[0]
  708. expected = [Flag('text', 'flash_text', [Align(4, True, False),
  709. Keep(),
  710. Surround('sym1'),
  711. Sort('name')]),
  712. Flag('text', 'flash_text', [Align(4, True, False),
  713. Keep(),
  714. Surround('sym1'),
  715. Sort('name')])]
  716. actual = fragment.flags[('obj1', None, 'default')]
  717. self.assertEqual(expected, actual)
  718. if __name__ == '__main__':
  719. unittest.main()