def testMain(self):
"""Verify that running main method runs expected functons.
Expected: LoadAndMergeTables, WriteTable.
"""
self.mox.StubOutWithMock(mps, 'LoadAndMergeTables')
self.mox.StubOutWithMock(mps, 'WriteTable')
mps.LoadAndMergeTables(mox.IgnoreArg()).AndReturn('csv_table')
mps.WriteTable(mox.Regex(r'csv_table'), 'any-out')
self.mox.ReplayAll()
mps.main(['--out=any-out', 'any-package'])
self.mox.VerifyAll()
def testLoadAndMergeTables(self):
# Create a second table to merge with standard table.
row0_2 = {mps.COL_PACKAGE: 'lib/foo',
mps.COL_SLOT: '1',
mps.COL_OVERLAY: 'portage',
self.COL_VER_arm: '1.2.4',
mps.COL_TARGET: 'virtual/target-os-dev world'}
row1_2 = {mps.COL_PACKAGE: 'dev/bar',
mps.COL_SLOT: '0',
mps.COL_OVERLAY: 'chromiumos-overlay',
self.COL_VER_arm: '1.2.3-r1',
mps.COL_TARGET: 'virtual/target-os-test'}
row2_2 = {mps.COL_PACKAGE: 'dev/newby',
mps.COL_SLOT: '2',
mps.COL_OVERLAY: 'chromiumos-overlay',
self.COL_VER_arm: '3.2.1',
mps.COL_TARGET: 'virtual/target-os virtual/target-sdk'}
cols = [col for col in self.COLUMNS if col != self.COL_VER_x86]
table_2 = self._CreateTableWithRows(cols,
[row0_2, row1_2, row2_2])
# Minor patch to main table for this test.
self._table.GetRowByIndex(2)[self.COL_VER_arm] = '1.2.4'
with self.OutputCapturer():
path1 = self._CreateTmpCsvFile(self._table)
path2 = self._CreateTmpCsvFile(table_2)
combined_table1 = mps.MergeTables([self._table, table_2])
combined_table2 = mps.LoadAndMergeTables([path1, path2])
final_row0 = {mps.COL_PACKAGE: 'dev/bar',
mps.COL_SLOT: '0',
mps.COL_OVERLAY: 'chromiumos-overlay',
self.COL_VER_x86: '1.2.3',
self.COL_VER_arm: '1.2.3-r1',
mps.COL_TARGET: 'virtual/target-os'}
final_row1 = {mps.COL_PACKAGE: 'dev/newby',
mps.COL_SLOT: '2',
mps.COL_OVERLAY: 'chromiumos-overlay',
self.COL_VER_x86: '',
self.COL_VER_arm: '3.2.1',
mps.COL_TARGET: 'virtual/target-os virtual/target-sdk'}
final_row2 = {mps.COL_PACKAGE: 'lib/foo',
mps.COL_SLOT: '0',
mps.COL_OVERLAY: 'portage',
self.COL_VER_x86: '1.2.3',
self.COL_VER_arm: '1.2.3',
mps.COL_TARGET: 'virtual/target-os-dev virtual/target-sdk'}
final_row3 = {mps.COL_PACKAGE: 'lib/foo',
mps.COL_SLOT: '1',
mps.COL_OVERLAY: 'portage',
self.COL_VER_x86: '1.2.3',
self.COL_VER_arm: '1.2.4',
mps.COL_TARGET: 'virtual/target-os-dev world'}
final_rows = (final_row0, final_row1, final_row2, final_row3)
for ix, row_out in enumerate(final_rows):
self.assertRowsEqual(row_out, combined_table1[ix])
self.assertRowsEqual(row_out, combined_table2[ix])
os.unlink(path1)
os.unlink(path2)