# Copyright 2016 Hewlett Packard Enterprise Development Company LP
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

from pyspark.sql import SQLContext

from monasca_transform.transform.transform_utils import RecordStoreUtils
from monasca_transform.transform.transform_utils import TransformSpecsUtils
from monasca_transform.transform import TransformContextUtils

from monasca_transform.component.setter.rollup_quantity \
    import RollupQuantity
from monasca_transform.component.usage.fetch_quantity \
    import FetchQuantity
from tests.unit.spark_context_test import SparkContextTest
from tests.unit.test_resources.mem_total_all.data_provider import DataProvider


class UsageComponentTest(SparkContextTest):

    def setUp(self):
        super(UsageComponentTest, self).setUp()
        self.sql_context = SQLContext(self.spark_context)

    def test_sum_quantity_all_component(self):

        record_store_df = RecordStoreUtils.create_df_from_json(
            self.sql_context, DataProvider.record_store_path)

        transform_spec_df = TransformSpecsUtils.create_df_from_json(
            self.sql_context, DataProvider.transform_spec_path)

        transform_context = \
            TransformContextUtils.get_context(
                transform_spec_df_info=transform_spec_df)

        instance_usage_df = FetchQuantity.usage(
            transform_context, record_store_df)

        instance_usage_df_all = RollupQuantity.setter(
            transform_context, instance_usage_df)

        result_list = [(row.usage_date, row.usage_hour,
                        row.tenant_id, row.host, row.quantity)
                       for row in instance_usage_df_all.rdd.collect()]

        expected_result = [
            ('2016-02-08', '18', 'all', 'all', 12946.0)]

        self.assertItemsEqual(result_list, expected_result)