#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for create_pkg_info."""

from __future__ import print_function

import io
import unittest

import create_pkg_info


class CreatePkgInfoTest(unittest.TestCase):
  """All the tests."""

  INSTALLED_PKG_INPUT_LINES = [
      'app-arch/gzip-1.9',
      'dev-libs/popt-1.16-r2',
      'app-emulation/docker-credential-helpers-0.6.3-r1',
      '_not.real-category1+/_not-real_package1-12.34.56.78',
      ' _not.real-category1+/_not-real_package2-12.34.56.78-r26 ',
      '  _not.real-category1+/_not-real_package3-12.34.56.78_rc3  ',
      '   _not.real-category1+/_not-real_package4-12.34.56.78_rc3-r26   ',
      '    _not.real-category1+/_not-real_package5-12.34.56.78_pre2_rc3-r26   ',
      '_not.real-category2+/_not-real_package1-12.34.56.78q',
      '_not.real-category2+/_not-real_package2-12.34.56.78q-r26',
      '_not.real-category2+/_not-real_package3-12.34.56.78q_rc3',
      '_not.real-category2+/_not-real_package4-12.34.56.78q_rc3-r26',
      '_not.real-category2+/_not-real_package5-12.34.56.78q_pre2_rc3-r26'
  ]

  BUILDTIME_PKG_INPUT_LINES = [
      'virtual/pkgconfig-0-r1',
      'dev-go/protobuf-1.3.2-r1',
      'app-emulation/docker-credential-helpers-0.6.3-r1',
      '_not.real-category1+/_not-real_package1-12.34.56.78',
      ' _not.real-category1+/_not-real_package2-12.34.56.78-r26 ',
      '  _not.real-category1+/_not-real_package3-12.34.56.78_rc3  ',
      '   _not.real-category1+/_not-real_package4-12.34.56.78_rc3-r26   ',
      '    _not.real-category1+/_not-real_package5-12.34.56.78_pre2_rc3-r26   ',
      '_not.real-category2+/_not-real_package1-12.34.56.78q',
      '_not.real-category2+/_not-real_package2-12.34.56.78q-r26',
      '_not.real-category2+/_not-real_package3-12.34.56.78q_rc3',
      '_not.real-category2+/_not-real_package4-12.34.56.78q_rc3-r26',
      '_not.real-category2+/_not-real_package5-12.34.56.78q_pre2_rc3-r26'
  ]

  ALL_PKG_INPUT_LINES = [
      'app-arch/gzip-1.9',
      'dev-libs/popt-1.16-r2 overlay:portage-stable',
      'virtual/pkgconfig-0-r1',
      'dev-go/protobuf-1.3.2-r1 overlay:portage-stable',
      'app-emulation/docker-credential-helpers-0.6.3-r1 use_flags:+audit,+crypt,+multiarch,+ssp,+static-libs',
      'sys-apps/systemd-248.6-r2 overlay:project-lakitu use_flags:+apparmor,+audit,+cgroup-hybrid',
  ]

  INSTALLED_PKG_EXPECTED_LIST = [
      {'category': 'app-arch', 'package': 'gzip', 'version_no_rev': '1.9',
       'revision': None},
      {'category': 'dev-libs', 'package': 'popt', 'version_no_rev': '1.16',
       'revision': '2'},
      {'category': 'app-emulation', 'package': 'docker-credential-helpers',
       'version_no_rev': '0.6.3', 'revision': '1'},
      {'category': '_not.real-category1+', 'package': '_not-real_package1',
       'version_no_rev': '12.34.56.78', 'revision': None},
      {'category': '_not.real-category1+', 'package': '_not-real_package2',
       'version_no_rev': '12.34.56.78', 'revision': '26'},
      {'category': '_not.real-category1+', 'package': '_not-real_package3',
       'version_no_rev': '12.34.56.78_rc3', 'revision': None},
      {'category': '_not.real-category1+', 'package': '_not-real_package4',
       'version_no_rev': '12.34.56.78_rc3', 'revision': '26'},
      {'category': '_not.real-category1+', 'package': '_not-real_package5',
       'version_no_rev': '12.34.56.78_pre2_rc3', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package1',
       'version_no_rev': '12.34.56.78q', 'revision': None},
      {'category': '_not.real-category2+', 'package': '_not-real_package2',
       'version_no_rev': '12.34.56.78q', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package3',
       'version_no_rev': '12.34.56.78q_rc3', 'revision': None},
      {'category': '_not.real-category2+', 'package': '_not-real_package4',
       'version_no_rev': '12.34.56.78q_rc3', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package5',
       'version_no_rev': '12.34.56.78q_pre2_rc3', 'revision': '26'}
  ]

  BUILDTIME_PKG_EXPECTED_LIST = [
      {'category': 'virtual', 'package': 'pkgconfig', 'version_no_rev': '0',
       'revision': '1'},
      {'category': 'dev-go', 'package': 'protobuf', 'version_no_rev': '1.3.2',
       'revision': '1'},
      {'category': 'app-emulation', 'package': 'docker-credential-helpers',
       'version_no_rev': '0.6.3', 'revision': '1'},
      {'category': '_not.real-category1+', 'package': '_not-real_package1',
       'version_no_rev': '12.34.56.78', 'revision': None},
      {'category': '_not.real-category1+', 'package': '_not-real_package2',
       'version_no_rev': '12.34.56.78', 'revision': '26'},
      {'category': '_not.real-category1+', 'package': '_not-real_package3',
       'version_no_rev': '12.34.56.78_rc3', 'revision': None},
      {'category': '_not.real-category1+', 'package': '_not-real_package4',
       'version_no_rev': '12.34.56.78_rc3', 'revision': '26'},
      {'category': '_not.real-category1+', 'package': '_not-real_package5',
       'version_no_rev': '12.34.56.78_pre2_rc3', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package1',
       'version_no_rev': '12.34.56.78q', 'revision': None},
      {'category': '_not.real-category2+', 'package': '_not-real_package2',
       'version_no_rev': '12.34.56.78q', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package3',
       'version_no_rev': '12.34.56.78q_rc3', 'revision': None},
      {'category': '_not.real-category2+', 'package': '_not-real_package4',
       'version_no_rev': '12.34.56.78q_rc3', 'revision': '26'},
      {'category': '_not.real-category2+', 'package': '_not-real_package5',
       'version_no_rev': '12.34.56.78q_pre2_rc3', 'revision': '26'}
  ]

  ALL_PKG_EXPECTED_LIST = [
      {'category': 'app-arch', 'package': 'gzip', 'version_no_rev': '1.9',
       'revision': None, 'overlay': None, 'use_flags': None},
      {'category': 'dev-libs', 'package': 'popt', 'version_no_rev': '1.16',
       'revision': '2', 'overlay': 'overlay:portage-stable', 'use_flags': None},
      {'category': 'virtual', 'package': 'pkgconfig', 'version_no_rev': '0',
       'revision': '1', 'overlay': None, 'use_flags': None},
      {'category': 'dev-go', 'package': 'protobuf', 'version_no_rev': '1.3.2',
       'revision': '1', 'overlay': 'overlay:portage-stable', 'use_flags': None},
      {'category': 'app-emulation', 'package': 'docker-credential-helpers',
       'version_no_rev': '0.6.3', 'revision': '1', 'overlay': None,
       'use_flags': 'use_flags:+audit,+crypt,+multiarch,+ssp,+static-libs'},
      {'category': 'sys-apps', 'package': 'systemd', 'version_no_rev': '248.6',
       'revision': '2', 'overlay': 'overlay:project-lakitu',
       'use_flags': 'use_flags:+apparmor,+audit,+cgroup-hybrid'}
  ]

  EXPECTED_JSON = """{
    "installedPackages": [
        {
            "category": "app-arch",
            "name": "gzip",
            "version": "123.0.1",
            "ebuild_version": "1.9"
        },
        {
            "category": "dev-libs",
            "name": "popt",
            "version": "123.0.1",
            "ebuild_version": "1.16-r2"
        },
        {
            "category": "app-emulation",
            "name": "docker-credential-helpers",
            "version": "123.0.1",
            "ebuild_version": "0.6.3-r1"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package1",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package2",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78-r26"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package3",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_rc3"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package4",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_rc3-r26"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package5",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_pre2_rc3-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package1",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package2",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package3",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_rc3"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package4",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_rc3-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package5",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_pre2_rc3-r26"
        }
    ],
    "buildTimePackages": [
        {
            "category": "virtual",
            "name": "pkgconfig",
            "version": "123.0.1",
            "ebuild_version": "0-r1"
        },
        {
            "category": "dev-go",
            "name": "protobuf",
            "version": "123.0.1",
            "ebuild_version": "1.3.2-r1"
        },
        {
            "category": "app-emulation",
            "name": "docker-credential-helpers",
            "version": "123.0.1",
            "ebuild_version": "0.6.3-r1"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package1",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package2",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78-r26"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package3",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_rc3"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package4",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_rc3-r26"
        },
        {
            "category": "_not.real-category1+",
            "name": "_not-real_package5",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78_pre2_rc3-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package1",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package2",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package3",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_rc3"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package4",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_rc3-r26"
        },
        {
            "category": "_not.real-category2+",
            "name": "_not-real_package5",
            "version": "123.0.1",
            "ebuild_version": "12.34.56.78q_pre2_rc3-r26"
        }
    ]
}"""

  DEBUG_EXPECTED_JSON = """{
    "allPackages": [
        {
            "category": "app-arch",
            "name": "gzip",
            "version": "123.0.1",
            "ebuild_version": "1.9"
        },
        {
            "category": "dev-libs",
            "name": "popt",
            "version": "123.0.1",
            "ebuild_version": "1.16-r2",
            "overlay": "portage-stable"
        },
        {
            "category": "virtual",
            "name": "pkgconfig",
            "version": "123.0.1",
            "ebuild_version": "0-r1"
        },
        {
            "category": "dev-go",
            "name": "protobuf",
            "version": "123.0.1",
            "ebuild_version": "1.3.2-r1",
            "overlay": "portage-stable"
        },
        {
            "category": "app-emulation",
            "name": "docker-credential-helpers",
            "version": "123.0.1",
            "ebuild_version": "0.6.3-r1",
            "use_flags": "+audit,+crypt,+multiarch,+ssp,+static-libs"
        },
        {
            "category": "sys-apps",
            "name": "systemd",
            "version": "123.0.1",
            "ebuild_version": "248.6-r2",
            "overlay": "project-lakitu",
            "use_flags": "+apparmor,+audit,+cgroup-hybrid"
        }
    ]
}"""

  def __init__(self, *args, **kwargs):
    unittest.TestCase.__init__(self, *args, **kwargs)
    self.maxDiff = None

  def testCreateList(self):
    self.assertEqual(
        create_pkg_info.CreateList(self.INSTALLED_PKG_INPUT_LINES, False),
        self.INSTALLED_PKG_EXPECTED_LIST)
    self.assertEqual(
        create_pkg_info.CreateList(self.BUILDTIME_PKG_INPUT_LINES, False),
        self.BUILDTIME_PKG_EXPECTED_LIST)

  def testCreateDebugList(self):
    self.assertEqual(
        create_pkg_info.CreateList(self.ALL_PKG_INPUT_LINES, True),
        self.ALL_PKG_EXPECTED_LIST)

  def testWriteJson(self):
    mock_file = io.StringIO()
    pkg_info = {
        'installedPackages': self.INSTALLED_PKG_EXPECTED_LIST,
        'buildTimePackages': self.BUILDTIME_PKG_EXPECTED_LIST,
    }
    create_pkg_info.WriteJson(pkg_info, '123.0.1', mock_file, False)
    print(mock_file.getvalue())
    self.assertEqual(mock_file.getvalue(), self.EXPECTED_JSON)
    mock_file.close()

  def testWriteDebugJson(self):
    mock_file = io.StringIO()
    pkg_info = {
        'allPackages': self.ALL_PKG_EXPECTED_LIST,
    }
    create_pkg_info.WriteJson(pkg_info, '123.0.1', mock_file, True)
    print(mock_file.getvalue())
    self.assertEqual(mock_file.getvalue(), self.DEBUG_EXPECTED_JSON)
    mock_file.close()

if __name__ == '__main__':
  unittest.main()