def testIncompatibleRows(self):
row0 = collections.OrderedDict()
row0["foo"] = "quux"
row0["bar"] = "thud"
row1 = collections.OrderedDict()
row1["baz"] = "thud"
row1["bar"] = "blargh"
with self.assertRaises(ValueError):
osquery.ParseHeader([row0, row1])
def testSingleRow(self):
row = collections.OrderedDict()
row["foo"] = "quux"
row["bar"] = "thud"
row["baz"] = "norf"
header = osquery.ParseHeader([row])
self.assertLen(header.columns, 3)
self.assertEqual(header.columns[0].name, "foo")
self.assertEqual(header.columns[1].name, "bar")
self.assertEqual(header.columns[2].name, "baz")
def testMultiRow(self):
row0 = collections.OrderedDict()
row0["foo"] = "quux"
row0["bar"] = "thud"
row0["baz"] = "norf"
row1 = collections.OrderedDict()
row1["foo"] = "blargh"
row1["bar"] = "plugh"
row1["baz"] = "ztesch"
header = osquery.ParseHeader([row0, row1])
self.assertLen(header.columns, 3)
self.assertEqual(header.columns[0].name, "foo")
self.assertEqual(header.columns[1].name, "bar")
self.assertEqual(header.columns[2].name, "baz")
def testEmpty(self):
header = osquery.ParseHeader([])
self.assertEmpty(header.columns, 0)
