Synnefo » Pithos MS Client

using System;

using System.Collections.Generic;

using System.IO;

using System.Linq;

using System.Text;

using NUnit.Framework;

using Pithos.Core.Agents;

using Pithos.Network;

namespace Pithos.Core.Test

{

    [TestFixture]

    class NetworkAgentTest

    {

        [Test]

        public void TestUpload()

        {

            var agent = new NetworkAgent

            {

                PithosContainer = "pithos",

                BlockSize = 4194304,

                BlockHash="sha256",

                CloudClient = new CloudFilesClient

                                {

                                    AuthenticationUrl = @"https://pithos.dev.grnet.gr",

                                    UsePithos = true

                                }

            };

            agent.CloudClient.Authenticate("890329@vho.grnet.gr", "24989dce4e0fcb072f8cb60c8922be19");

            var fileName = "012345.dump";

            var filePath = Path.Combine(@"e:\pithos\", fileName);

            var random = new Random();

            var buffer = new byte[4096];

            random.NextBytes(buffer);

            if (File.Exists(filePath))

                File.Delete(filePath);

            using (var stream = File.OpenWrite(filePath))

            {

                for (var i = 0; i < 2* 1024;i++ )

                    stream.Write(buffer, 0, buffer.Length);

            }

            agent.CloudClient.DeleteObject(agent.PithosContainer, fileName);

            var task = Signature.CalculateTreeHashAsync(filePath, agent.BlockSize, agent.BlockHash);

            agent.UploadWithHashMap(new FileInfo(filePath),fileName,task);

            var newHash = agent.CloudClient.GetHashMap(agent.PithosContainer, fileName).Result;

            var treeHash = task.Result;

            Assert.AreEqual(treeHash.TopHash, newHash.TopHash);

            Assert.AreEqual(treeHash.Hashes, newHash.Hashes);

        }

        [Test]

        public void TestDownload()

        {

            var fileAgent = new FileAgent {FragmentsPath = @"e:\pithos\Fragments"};

            var agent = new NetworkAgent

            {

                PithosContainer = "pithos",

                BlockSize = 4194304,

                BlockHash="sha256",

                CloudClient = new CloudFilesClient

                                {

                                    AuthenticationUrl = @"https://pithos.dev.grnet.gr",

                                    UsePithos = true

                                },

                FileAgent=fileAgent

            };

            agent.CloudClient.Authenticate("890329@vho.grnet.gr", "24989dce4e0fcb072f8cb60c8922be19");

            var fileName = @"vlc-1.1.11-win32.exe";

            var filePath = Path.Combine(@"e:\pithos\", fileName);

            if (File.Exists(filePath))

                File.Delete(filePath);            

            var newHash = agent.CloudClient.GetHashMap(agent.PithosContainer, fileName).Result;

            agent.DownloadWithBlocks(agent.PithosContainer,new Uri(fileName,UriKind.Relative),filePath,newHash)

                .Wait();

            Assert.IsTrue(File.Exists(filePath));

            var treeHash = Signature.CalculateTreeHashAsync(filePath, agent.BlockSize, agent.BlockHash).Result;

            Assert.AreEqual(treeHash.TopHash, newHash.TopHash);

            Assert.AreEqual(treeHash.Hashes, newHash.Hashes);

        }

    }

}