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>