# Copyright (c) 2016 IBM Corporation
#
# 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 test.functional.tests import Base, Utils
from test.functional.swift_test_client import Account, Connection, \
    ResponseError
import test.functional as tf
import time
import logging
import os
import unittest
import xattr


class TestSwiftOnHPSS(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.connection = Connection(tf.config)
        cls.connection.authenticate()
        cls.account = Account(cls.connection,
                              tf.config.get('account',
                                            tf.config['username']))
        cls.container = cls.account.container('swiftonhpss_test')
        cls.container.create(hdrs={'X-Storage-Policy': 'swiftonhpss'})
        cls.hpss_dir = '/srv/hpss'

    @classmethod
    def tearDownClass(cls):
        cls.container.delete()

    def setUp(self):
        self.test_file = self.container.file('testfile')

    def tearDown(self):
        try:
            self.test_file.delete()
        except ResponseError as e:
            if e.message != "Not Found":
                raise

    def test_purge_lock(self):
        resp = self.test_file.write(data='test',
                                    hdrs={'X-Hpss-Purgelock-Status': 'true',
                                          'X-Hpss-Class-Of-Service-Id': '1'},
                                    return_resp=True)

        print resp.status
        print resp.getheaders()
        print resp.read()

        test_file_name = os.path.join(self.hpss_dir,
                                      self.account.name,
                                      self.container.name,
                                      'testfile')

        print test_file_name

        print os.stat(test_file_name)

        print xattr.listxattr(test_file_name)

        self.assertEqual(xattr.get(test_file_name,
                                   'system.hpss.purgelock'),
                         '1')

        self.test_file.post(hdrs={'X-Hpss-Purgelock-Status': 'false'})

        self.assertEqual(xattr.get(test_file_name,
                                   'system.hpss.purgelock'), '0')

    def test_change_cos(self):
        self.test_file.write(data='asdfasdf',
                             hdrs={'X-Hpss-Class-Of-Service-Id': '2'})

        test_file_name = os.path.join(self.hpss_dir,
                                      self.account.name,
                                      self.container.name,
                                      'testfile')

        print test_file_name

        time.sleep(10)  # It takes a long time for HPSS to get around to it.
        self.assertEqual(xattr.get(test_file_name, 'system.hpss.cos'), '2')

        self.test_file.post(hdrs={'X-Hpss-Class-Of-Service-Id': '1'})
        time.sleep(10)
        self.assertEqual(xattr.get(test_file_name, 'system.hpss.cos'), '1')

    def test_hpss_metadata(self):
        # header is X-HPSS-Get-Metadata
        self.test_file.write(data='test')
        self.connection.make_request('HEAD', self.test_file.path,
                                     hdrs={'X-Hpss-Get-Metadata': 'true'})
        md = {t[0]: t[1] for t in self.connection.response.getheaders()}
        print md
        self.assertTrue('x-hpss-account' in md)
        self.assertTrue('x-hpss-bitfile-id' in md)
        self.assertTrue('x-hpss-comment' in md)
        self.assertTrue('x-hpss-class-of-service-id' in md)
        self.assertTrue('x-hpss-data-levels' in md)
        self.assertTrue('x-hpss-family-id' in md)
        self.assertTrue('x-hpss-fileset-id' in md)
        self.assertTrue('x-hpss-optimum-size' in md)
        self.assertTrue('x-hpss-purgelock-status' in md)
        self.assertTrue('x-hpss-reads' in md)
        self.assertTrue('x-hpss-realm-id' in md)
        self.assertTrue('x-hpss-subsys-id' in md)
        self.assertTrue('x-hpss-writes' in md)