'''
Copyright 2016, EMC, Inc.

Author(s):
George Paulos

'''

import os
import sys
import subprocess
# set path to common libraries
sys.path.append(subprocess.check_output("git rev-parse --show-toplevel", shell=True).rstrip("\n") + "/test/common")
import fit_common

# Select test group here using @attr
from nose.plugins.attrib import attr
@attr(all=True, regression=True, smoke=True)
class rackhd11_api_files(fit_common.unittest.TestCase):
    def setUp(self):
        # create test file
        TESTFILE = open(fit_common.TEST_PATH + 'testfile','w+')
        TESTFILE.write("1234567890ABCDEF")
        TESTFILE.close()
        # delete any instance of testfile on host
        api_data = fit_common.rackhdapi('/api/1.1/files/list/all')
        for item in api_data['json']:
            if item['filename'] == 'testfile':
                fit_common.rackhdapi('/api/1.1/files/' + item['uuid'], action="delete")
    def tearDown(self):
        os.remove(fit_common.TEST_PATH + 'testfile')
    def test_api_11_files_put_get_delete(self):
        # put file fia files API, then check data
        api_data = fit_common.rackhdapi('/api/1.1/files/testfile', action="binary-put", payload = file(fit_common.TEST_PATH + 'testfile').read())
        self.assertEqual(api_data['status'], 201, 'Incorrect HTTP return code, expected 201, got:' + str(api_data['status']))
        fileid = api_data['text']
        # Retrieve file
        api_data = fit_common.rackhdapi('/api/1.1/files/' + fileid)
        self.assertEqual(api_data['status'], 200, 'Incorrect HTTP return code, expected 200, got:' + str(api_data['status']))
        self.assertEqual(api_data['text'], open(fit_common.TEST_PATH + 'testfile').read(), 'File corrupted, ID: ' + fileid)
        # list all
        api_data = fit_common.rackhdapi('/api/1.1/files/list/all')
        self.assertIn(fileid, fit_common.json.dumps(api_data['json']), 'File ID missing in file list.')
        # delete file
        api_data = fit_common.rackhdapi('/api/1.1/files/' + fileid, action='delete')
        self.assertEqual(api_data['status'], 204, 'Incorrect HTTP return code, expected 204, got:' + str(api_data['status']))

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