public string Write() { //find first table with no dependencies (foreign keys) and a .Net name. var entity = _schema.Tables .FirstOrDefault(t => t.ForeignKeys.Count == 0 && !string.IsNullOrEmpty(t.NetName)); if (entity == null) { return(null); } //we'll also run a sproc if we find one var sproc = _schema.StoredProcedures.FirstOrDefault(p => p.ResultSets.Count > 0); ClassName = entity.NetName + "Test"; WriteNamespaces(sproc != null); using (_cb.BeginNest("namespace " + _codeWriterSettings.Namespace + ".Tests")) { _cb.AppendLine("[TestClass]"); using (_cb.BeginNest("public class " + ClassName)) { WriteStaticConstructor(entity); WriteOpenSession(); WriteGenerateString(); WriteCreateEntity(entity); WriteCrudTest(entity); WriteSproc(sproc); } } return(_cb.ToString()); }
/// <summary> /// Writes the C# code of the table /// </summary> /// <returns></returns> public string Write() { var codeTarget = _codeWriterSettings.CodeTarget; _dataAnnotationWriter = new DataAnnotationWriter(IsEntityFramework(), _codeWriterSettings); var className = _table.NetName; if (string.IsNullOrEmpty(className) && _table.DatabaseSchema != null) { PrepareSchemaNames.Prepare(_table.DatabaseSchema, _codeWriterSettings.Namer); className = _table.NetName; } _dataTypeWriter.CodeTarget = codeTarget; _inheritanceTable = _table.FindInheritanceTable(); WriteNamespaces(); if (!string.IsNullOrEmpty(_codeWriterSettings.Namespace)) { _cb.BeginNest("namespace " + _codeWriterSettings.Namespace); } if (codeTarget == CodeTarget.PocoRiaServices) { WriteRiaClass(className); } else { var tableOrView = _table is DatabaseView ? "view" : "table"; var comment = "Class representing " + _table.Name + " " + tableOrView; var classDefinition = "public class " + className; if (_inheritanceTable != null) { classDefinition += " : " + _inheritanceTable.NetName; } using (_cb.BeginNest(classDefinition, comment)) { WriteClassMembers(className); } } if (_table.HasCompositeKey && _inheritanceTable == null) { WriteCompositeKeyClass(className); } if (!string.IsNullOrEmpty(_codeWriterSettings.Namespace)) { _cb.EndNest(); } return(_cb.ToString()); }