Fix for permission set + tests

model/Models/Assembly.cs

@@ -12,9 +12,12 @@ public SqlAssembly(string permissionSet, string name) {
  PermissionSet = permissionSet;
  Name = name;
 
- if (PermissionSet == "SAFE_ACCESS")
- PermissionSet = "SAFE";
- }
+ if (PermissionSet == "SAFE_ACCESS")
+ PermissionSet = "SAFE";
+
+ if (PermissionSet == "UNSAFE_ACCESS")
+ PermissionSet = "UNSAFE";
+ }
 
  public string ScriptCreate() {
  var commands =

test/AssemblyTester.cs

@@ -0,0 +1,23 @@
+using System.Collections.Generic;
+using NUnit.Framework;
+using SchemaZen.Library.Models;
+
+namespace SchemaZen.Tests {
+ [TestFixture]
+ public class AssemblyTester {
+
+ [Test]
+ [TestCase("SAFE_ACCESS", "SAFE")]
+ [TestCase("UNSAFE_ACCESS", "UNSAFE")]
+ [TestCase("EXTERNAL_ACCESS", "EXTERNAL_ACCESS")]
+ public void Assembly_WithPermissionSetCases(string permissionSet, string scriptedPermissionSet) {
+ var assembly = new SqlAssembly(permissionSet, "SchemazenAssembly");
+ assembly.Files.Add(new KeyValuePair<string, byte[]>("mydll", new byte[0]));
+
+ var expected = @"CREATE ASSEMBLY [SchemazenAssembly]
+FROM 0x
+WITH PERMISSION_SET = " + scriptedPermissionSet;
+ Assert.AreEqual(expected, assembly.ScriptCreate());
+ }
+ }
+}

test/BatchSqlParserTester.cs

@@ -1,7 +1,6 @@
 using System;
 using NUnit.Framework;
 using SchemaZen.Library;
-using SchemaZen.Library.Models;
 
 namespace SchemaZen.Tests {
  [TestFixture]

test/Tests.csproj

@@ -74,6 +74,7 @@
  <Reference Include="System.Xml" />
  </ItemGroup>
  <ItemGroup>
+ <Compile Include="AssemblyTester.cs" />
  <Compile Include="Properties\AssemblyInfo.cs" />
  <Compile Include="UserTester.cs" />
  </ItemGroup>