#! /usr/bin/env python

# Copyright (C) 2006, Stefan Schwarzer
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# - Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
#
# - Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
#
# - Neither the name of the above author nor the names of the
#   contributors to the software may be used to endorse or promote
#   products derived from this software without specific prior written
#   permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

# $Id: upload_download_test.py 633 2006-11-22 22:15:32Z schwa $

# Test script for ticket #13 (reported by Pete Schott)

import getpass
import random
import sys

import ftputil


def login_data():
    """Get host, user, password and return them as a triple."""
    print "Please enter login data:"
    remote_host = raw_input("Remote host: ")
    user = raw_input("User name: ")
    password = getpass.getpass("Password: ")
    print
    #return "localhost", 'ftptest', 'd605581757de5eb56d568a4419f4126e'
    return remote_host, user, password

def test_data():
    """Return a pseudo-random string of length between 0 and 5120."""
    length = random.randint(0, 5120)
    data = [chr(random.randint(0, 255)) for i in range(length)]
    return "".join(data)
    
# open connection and read local data
login_data_ = login_data()
host = ftputil.FTPHost(*login_data_)

# download and test several times
passed = failed = 0
for i in range(500):
    # save and upload random data and try to test remote integrity
    data = test_data()
    local_data_file = open("local_data", "wb")
    local_data_file.write(data)
    local_data_file.close()
    host.upload("local_data", "remote_data", "b")
    # download file and compare it with the expected data
    host.download("remote_data", "local_data", "b")
    local_data_file = open("local_data", "rb")
    local_data = local_data_file.read()
    local_data_file.close()
    print "File %3d (length %4d) is" % ((i+1), len(data)),
    if data == local_data:
        print "ok"
        passed += 1
    else:
        print "NOT OK"
        failed += 1

host.remove("remote_data")
host.close()

print "\n%d tests passed, %d tests failed" % (passed, failed)