def test_get_relpath_from_unit(self): distributor = YumDistributor() test_unit = Unit("rpm", "unit_key", {}, "") test_unit.unit_key = {"fileName" : "test_1"} rel_path = util.get_relpath_from_unit(test_unit) self.assertEqual(rel_path, "test_1") test_unit.unit_key = {} test_unit.storage_path = "test_0" rel_path = util.get_relpath_from_unit(test_unit) self.assertEqual(rel_path, "test_0") test_unit.metadata["filename"] = "test_2" rel_path = util.get_relpath_from_unit(test_unit) self.assertEqual(rel_path, "test_2") test_unit.metadata["relativepath"] = "test_3" rel_path = util.get_relpath_from_unit(test_unit) self.assertEqual(rel_path, "test_3")