public void GetResultTest() { // arrange var model = new EfDaigramModel { Entities = new List <Entity> { new Entity { Name = "e01", Columns = new List <Column> { new Column { Name = "id", Type = "int" }, new Column { Name = "name", Type = "int" } }, }, new Entity { Name = "e02", Columns = new List <Column> { new Column { Name = "id", Type = "int" } } } }, }; // act var result = this._target.GetResult(model); Assert.Contains("@startum", result.Content); Assert.Contains("@enduml", result.Content); }
DiagramResult IDiagramGenerator.GetResult(EfDaigramModel model) { if (model.Entities?.Any() is not true) { return(default(DiagramResult)); } var result = new PumlSyntaxModel { Entities = new List <string>() }; foreach (var e in model.Entities) { var columns = this.GetColumns(e.Columns); var entity = this.GetEntity(e.Name, columns); result.Entities.Add(entity); } if (model.RelationShips?.Any() is true) { result.RelationShips = this.GetRelationShips(model.RelationShips); } return(new DiagramResult { DatabaseName = model.DatabaseName, Content = result.ToString(), FileType = FileType.puml }); }