def test_primary_file_creation(self):

        path = os.path.join(self.metadata_file_dir, REPO_DATA_DIR_NAME, PRIMARY_XML_FILE_NAME)

        context = PrimaryXMLFileContext(self.metadata_file_dir, 0)

        context._open_metadata_file_handle()
        context._close_metadata_file_handle()

        self.assertTrue(os.path.exists(path))
    def test_primary_unit_metadata(self):

        path = os.path.join(self.metadata_file_dir, REPO_DATA_DIR_NAME, PRIMARY_XML_FILE_NAME)

        unit = self._generate_rpm("seems-legit")

        context = PrimaryXMLFileContext(self.metadata_file_dir, 1)

        context._open_metadata_file_handle()
        context.add_unit_metadata(unit)
        context._close_metadata_file_handle()

        handle = gzip.open(path, "r")
        content = handle.read()
        handle.close()

        self.assertEqual(content, "PRIMARY")
    def test_primary_opening_tag(self):

        path = os.path.join(self.metadata_file_dir, REPO_DATA_DIR_NAME, PRIMARY_XML_FILE_NAME)

        context = PrimaryXMLFileContext(self.metadata_file_dir, 0)

        context._open_metadata_file_handle()
        context._write_root_tag_open()
        context._close_metadata_file_handle()

        self.assertNotEqual(os.path.getsize(path), 0)

        primary_handle = gzip.open(path, "r")
        content = primary_handle.read()
        primary_handle.close()

        self.assertTrue(content.startswith("<metadata"))
        self.assertEqual(content.count('xmlns="%s"' % COMMON_NAMESPACE), 1)
        self.assertEqual(content.count('xmlns:rpm="%s"' % RPM_NAMESPACE), 1)
        self.assertEqual(content.count('packages="0"'), 1)