/src/ServiceManagement/Services/Commands.Test/CloudService/Development/SetAzureInstancesTests.cs

// ----------------------------------------------------------------------------------

//

// Copyright Microsoft 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.

// ----------------------------------------------------------------------------------



using System;

using System.IO;

using System.Management.Automation;

using Microsoft.WindowsAzure.Commands.CloudService.Development;

using Microsoft.WindowsAzure.Commands.Common.Test.Mocks;

using Microsoft.WindowsAzure.Commands.Test.Utilities.Common;

using Microsoft.WindowsAzure.Commands.Utilities.CloudService;

using Microsoft.WindowsAzure.Commands.Utilities.Common.XmlSchema.ServiceConfigurationSchema;

using Microsoft.WindowsAzure.Commands.Utilities.Properties;

using Microsoft.WindowsAzure.Commands.ScenarioTest;

using Xunit;

using Microsoft.WindowsAzure.Commands.Utilities.Common;



namespace Microsoft.WindowsAzure.Commands.Test.CloudService.Development.Tests.Cmdlet

{

    public class SetAzureInstancesTests : SMTestBase

    {

        private const string serviceName = "AzureService";



        private MockCommandRuntime mockCommandRuntime;



        private SetAzureServiceProjectRoleCommand cmdlet;



        public SetAzureInstancesTests()

        {

            mockCommandRuntime = new MockCommandRuntime();

            cmdlet = new SetAzureServiceProjectRoleCommand();

            cmdlet.CommandRuntime = mockCommandRuntime;

            cmdlet.PassThru = true;

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsNode()

        {

            int newRoleInstances = 10;



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                string roleName = "WebRole1";

                service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath);

                cmdlet.PassThru = false;

                RoleSettings roleSettings = cmdlet.SetAzureInstancesProcess("WebRole1", newRoleInstances, service.Paths.RootPath);

                service = new CloudServiceProject(service.Paths.RootPath, null);



                Assert.Equal<int>(newRoleInstances, service.Components.CloudConfig.Role[0].Instances.count);

                Assert.Equal<int>(newRoleInstances, service.Components.LocalConfig.Role[0].Instances.count);

                Assert.Equal<int>(0, mockCommandRuntime.OutputPipeline.Count);

                Assert.Equal<int>(newRoleInstances, roleSettings.Instances.count);

                Assert.Equal<string>(roleName, roleSettings.name);



            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsPHP()

        {

            int newRoleInstances = 10;



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                string roleName = "WebRole1";

                service.AddWebRole(Test.Utilities.Common.Data.PHPWebRoleScaffoldingPath);

                RoleSettings roleSettings = cmdlet.SetAzureInstancesProcess("WebRole1", newRoleInstances, service.Paths.RootPath);

                service = new CloudServiceProject(service.Paths.RootPath, null);



                Assert.Equal<int>(newRoleInstances, service.Components.CloudConfig.Role[0].Instances.count);

                Assert.Equal<int>(newRoleInstances, service.Components.LocalConfig.Role[0].Instances.count);

                Assert.Equal<string>(roleName, ((PSObject)mockCommandRuntime.OutputPipeline[0]).Members[Parameters.RoleName].Value.ToString());

                Assert.True(((PSObject)mockCommandRuntime.OutputPipeline[0]).TypeNames.Contains(typeof(RoleSettings).FullName));

                Assert.Equal<int>(newRoleInstances, roleSettings.Instances.count);

                Assert.Equal<string>(roleName, roleSettings.name);

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsRoleNameDoesNotExistFail()

        {

            string roleName = "WebRole1";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, roleName, 10), string.Format(Resources.RoleNotFoundMessage, roleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsNodeRoleNameDoesNotExistServiceContainsWebRoleFail()

        {

            string roleName = "WebRole1";

            string invalidRoleName = "foo";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath, roleName, 1);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, invalidRoleName, 10), string.Format(Resources.RoleNotFoundMessage, invalidRoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsPHPRoleNameDoesNotExistServiceContainsWebRoleFail()

        {

            string roleName = "WebRole1";

            string invalidRoleName = "foo";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                service.AddWebRole(Test.Utilities.Common.Data.PHPWebRoleScaffoldingPath, roleName, 1);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, invalidRoleName, 10), string.Format(Resources.RoleNotFoundMessage, invalidRoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsNodeRoleNameDoesNotExistServiceContainsWorkerRoleFail()

        {

            string roleName = "WorkerRole1";

            string invalidRoleName = "foo";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                service.AddWorkerRole(Test.Utilities.Common.Data.NodeWorkerRoleScaffoldingPath, roleName, 1);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, invalidRoleName, 10), string.Format(Resources.RoleNotFoundMessage, invalidRoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsPHPRoleNameDoesNotExistServiceContainsWorkerRoleFail()

        {

            string roleName = "WorkerRole1";

            string invalidRoleName = "foo";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                service.AddWorkerRole(Test.Utilities.Common.Data.PHPWorkerRoleScaffoldingPath, roleName, 1);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, invalidRoleName, 10), string.Format(Resources.RoleNotFoundMessage, invalidRoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsEmptyRoleNameFail()

        {

            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, string.Empty, 10), string.Format(Resources.InvalidOrEmptyArgumentMessage, Resources.RoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsNullRoleNameFail()

        {

            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, null, 10), string.Format(Resources.InvalidOrEmptyArgumentMessage, Resources.RoleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsLargeRoleInstanceFail()

        {

            string roleName = "WebRole1";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, roleName, 2000), string.Format(Resources.InvalidInstancesCount, roleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessNegativeRoleInstanceFail()

        {

            string roleName = "WebRole1";



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                Testing.AssertThrows<ArgumentException>(() => service.SetRoleInstances(service.Paths, roleName, -1), string.Format(Resources.InvalidInstancesCount, roleName));

            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureInstancesProcessTestsCaseInsensitive()

        {

            int newRoleInstances = 10;



            using (FileSystemHelper files = new FileSystemHelper(this))

            {

                CloudServiceProject service = new CloudServiceProject(files.RootPath, serviceName, null);

                string roleName = "WebRole1";

                service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath);

                cmdlet.PassThru = false;

                RoleSettings roleSettings = cmdlet.SetAzureInstancesProcess("WeBrolE1", newRoleInstances, service.Paths.RootPath);

                service = new CloudServiceProject(service.Paths.RootPath, null);



                Assert.Equal<int>(newRoleInstances, service.Components.CloudConfig.Role[0].Instances.count);

                Assert.Equal<int>(newRoleInstances, service.Components.LocalConfig.Role[0].Instances.count);

                Assert.Equal<int>(0, mockCommandRuntime.OutputPipeline.Count);

                Assert.Equal<int>(newRoleInstances, roleSettings.Instances.count);

                Assert.Equal<string>(roleName, roleSettings.name);



            }

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureServiceProjectRoleWithoutPassingRoleName()

        {

            TestMockSupport.TestExecutionFolder = AppDomain.CurrentDomain.BaseDirectory;

            string serviceName = "AzureService1";

            if (Directory.Exists(Path.Combine(TestMockSupport.TestExecutionFolder,serviceName)))

            {

                Directory.Delete(Path.Combine(TestMockSupport.TestExecutionFolder, serviceName), true);

            }

            CloudServiceProject service = new CloudServiceProject(TestMockSupport.TestExecutionFolder, serviceName, null);

            service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath);

            TestMockSupport.TestExecutionFolder = Path.Combine(service.Paths.RootPath, "WebRole1");

            cmdlet.RoleName = string.Empty;

            cmdlet.ExecuteCmdlet();

            service = new CloudServiceProject(service.Paths.RootPath, null);



            Assert.Equal<string>("WebRole1", cmdlet.RoleName);

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureServiceProjectRoleInDeepDirectory()

        {

            TestMockSupport.TestExecutionFolder = AppDomain.CurrentDomain.BaseDirectory;

            string serviceName = "AzureService2";

            if (Directory.Exists(Path.Combine(TestMockSupport.TestExecutionFolder, serviceName)))

            {

                Directory.Delete(Path.Combine(TestMockSupport.TestExecutionFolder, serviceName), true);

            }

            CloudServiceProject service = new CloudServiceProject(TestMockSupport.TestExecutionFolder, serviceName, null);

            service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath);

            TestMockSupport.TestExecutionFolder = Path.Combine(service.Paths.RootPath, "WebRole1", "bin");

            cmdlet.RoleName = string.Empty;

            cmdlet.ExecuteCmdlet();

            service = new CloudServiceProject(service.Paths.RootPath, null);



            Assert.Equal<string>("WebRole1", cmdlet.RoleName);

        }



        [Fact]

        [Trait(Category.AcceptanceType, Category.CheckIn)]

        public void SetAzureServiceProjectRoleInServiecRootDirectoryFail()

        {

            TestMockSupport.TestExecutionFolder = AppDomain.CurrentDomain.BaseDirectory;

            string serviceName = "AzureService3";

            if (Directory.Exists(Path.Combine(TestMockSupport.TestExecutionFolder, serviceName)))

            {

                Directory.Delete(Path.Combine(TestMockSupport.TestExecutionFolder, serviceName), true);

            }

            CloudServiceProject service = new CloudServiceProject(TestMockSupport.TestExecutionFolder, serviceName, null);

            service.AddWebRole(Test.Utilities.Common.Data.NodeWebRoleScaffoldingPath);

            cmdlet.RoleName = string.Empty;

            Testing.AssertThrows<InvalidOperationException>(() => cmdlet.ExecuteCmdlet(), Resources.CannotFindServiceRoot);

        }

    }

}