Add AA9 tests

1 files changed, 45 insertions, 2 deletions

diff --git a/linaro_image_tools/media_create/tests/test_media_create.py b/linaro_image_tools/media_create/tests/test_media_create.py
index 71beb56..82048cf 100644
--- a/linaro_image_tools/media_create/tests/test_media_create.py
+++ b/linaro_image_tools/media_create/tests/test_media_create.py
@@ -1440,6 +1440,12 @@ class TestBootSteps(TestCaseWithFixtures):
             'make_dtb', 'make_boot_script', 'make_boot_ini']
         self.assertEqual(expected, self.funcs_calls)
 
+    def test_aa9_steps(self):
+        board_conf = boards.Aa9Config()
+        board_conf.hwpack_format = HardwarepackHandler.FORMAT_1
+        expected = []
+        self.assertEqual(expected, self.funcs_calls)
+
 
 class TestPopulateRawPartition(TestCaseWithFixtures):
 
@@ -1571,6 +1577,11 @@ class TestPopulateRawPartition(TestCaseWithFixtures):
         expected = []
         self.assertEqual(expected, self.funcs_calls)
 
+    def test_aa9_raw(self):
+        self.populate_raw_partition(boards.Aa9Config())
+        expected = []
+        self.assertEqual(expected, self.funcs_calls)
+
 
 class TestPopulateRawPartitionAndroid(TestCaseWithFixtures):
 
@@ -1878,10 +1889,17 @@ class TestGetSfdiskCmd(TestCase):
             board_conf.get_sfdisk_cmd())
 
     def test_beaglebone(self):
-        board_conf = get_board_config('highbank')
+        board_conf = get_board_config('beaglebone')
         self.set_up_config(board_conf)
         self.assertEquals(
-            '63,106432,0x83,*\n106496,,,-',
+            '63,106432,0x0C,*\n106496,,,-',
+            board_conf.get_sfdisk_cmd())
+
+    def test_aa9(self):
+        board_conf = get_board_config('aa9')
+        self.set_up_config(board_conf)
+        self.assertEquals(
+            '63,106432,0x0C,*\n106496,,,-',
             board_conf.get_sfdisk_cmd())
 
     def test_panda_android(self):
@@ -2013,6 +2031,13 @@ class TestGetSfdiskCmdV2(TestCase):
             '63,106432,0x0C,*\n106496,,,-',
             board_conf.get_sfdisk_cmd())
 
+    def test_aa9(self):
+        board_conf = get_board_config('aa9')
+        board_conf.partition_layout = 'bootfs_rootfs'
+        self.assertEquals(
+            '63,106432,0x0C,*\n106496,,,-',
+            board_conf.get_sfdisk_cmd())
+
 
 class TestGetBootCmd(TestCase):
 
@@ -2289,6 +2314,24 @@ class TestGetBootCmd(TestCase):
             'initrd_high': '0xffffffff'}
         self.assertEqual(expected, boot_commands)
 
+    def test_aa9(self):
+        config = get_board_config('aa9')
+        config.serial_tty = config._serial_tty
+        boot_commands = config._get_boot_env(
+            is_live=False, is_lowmem=False, consoles=[],
+            rootfs_id="UUID=deadbeef", i_img_data="initrd",
+            d_img_data="board.dtb")
+        expected = {
+            'bootargs': 'console=ttyS0,115200n8  '
+                        'root=UUID=deadbeef rootwait ro',
+            'bootcmd': 'fatload mmc 0:1 0x40000000 uImage; '
+                       'fatload mmc 0:1 0x41100000 uInitrd; '
+                       'fatload mmc 0:1 0x41000000 board.dtb; '
+                       'bootm 0x40000000 0x41100000 0x41000000',
+            'fdt_high': '0xffffffff',
+            'initrd_high': '0xffffffff'}
+        self.assertEqual(expected, boot_commands)
+
 
 class TestExtraBootCmd(TestCaseWithFixtures):