Experiments/Attic/Generator/Castle.ActiveRecord.Generator.Components.Tests/RelationshipInferenceServiceTestCase.cs

   1 // Copyright 2004-2007 Castle Project - http://www.castleproject.org/
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 namespace Castle.ActiveRecord.Generator.Components.Tests
  16 {
  17         using System;
  18         using System.Collections;
  19         using System.Data.OleDb;
  20
  21         using NUnit.Framework;
  22
  23         using Castle.ActiveRecord.Generator.Components.Database;
  24
  25         /// <summary>
  26         /// TODO: Map HasOne association
  27         /// </summary>
  28         [TestFixture]
  29         public class RelationshipInferenceServiceTestCase : AbstractContainerTestCase
  30         {
  31                 [Test]
  32                 public void BlogHasManyPosts()
  33                 {
  34                         InitKernel();
  35                         IRelationshipInferenceService relService = ObtainService();
  36
  37                         TableDefinition blogTable;
  38                         TableDefinition postTable;
  39
  40                         BuildBlogPostsStructure(out blogTable, out postTable);
  41
  42                         BuildContext context = new BuildContext();
  43
  44                         ActiveRecordDescriptor arDesc = new ActiveRecordDescriptor();
  45
  46                         ActiveRecordPropertyDescriptor[] descs = relService.InferRelations( arDesc, blogTable, context );
  47
  48                         Assert.IsNotNull(descs);
  49                         Assert.AreEqual( 1, descs.Length );
  50
  51                         ActiveRecordHasManyDescriptor desc1 = descs[0] as ActiveRecordHasManyDescriptor;
  52                         Assert.IsNotNull(desc1);
  53                         Assert.IsNotNull(desc1.TargetType);
  54                         Assert.IsNotNull(desc1.PropertyType);
  55
  56                         Assert.AreEqual( "Posts", desc1.PropertyName );
  57                         Assert.AreEqual( "blog_id", desc1.ColumnName );
  58                         Assert.AreEqual( typeof(IList), desc1.PropertyType );
  59
  60                         ActiveRecordDescriptor targetARDescriptor = context.GetNextPendent();
  61                         Assert.AreSame( postTable, targetARDescriptor.Table );
  62                 }
  63
  64                 [Test]
  65                 public void PostBelongsToBlog()
  66                 {
  67                         InitKernel();
  68                         IRelationshipInferenceService relService = ObtainService();
  69
  70                         TableDefinition blogTable;
  71                         TableDefinition postTable;
  72
  73                         BuildBlogPostsStructure(out blogTable, out postTable);
  74
  75                         BuildContext context = new BuildContext();
  76
  77                         ActiveRecordDescriptor arDesc = new ActiveRecordDescriptor();
  78
  79                         ActiveRecordPropertyDescriptor[] descs = relService.InferRelations( arDesc, postTable, context );
  80
  81                         Assert.IsNotNull(descs);
  82                         Assert.AreEqual( 1, descs.Length );
  83
  84                         ActiveRecordBelongsToDescriptor desc1 = descs[0] as ActiveRecordBelongsToDescriptor;
  85                         Assert.IsNotNull(desc1);
  86                         Assert.IsNotNull(desc1.TargetType);
  87                         Assert.IsNull(desc1.PropertyType);
  88
  89                         Assert.AreEqual( "Blog", desc1.PropertyName );
  90                         Assert.AreEqual( "blog_id", desc1.ColumnName );
  91
  92                         ActiveRecordDescriptor targetARDescriptor = context.GetNextPendent();
  93                         Assert.AreSame( blogTable, targetARDescriptor.Table );
  94                 }
  95
  96                 [Test]
  97                 public void SelfReference()
  98                 {
  99                         InitKernel();
 100                         IRelationshipInferenceService relService = ObtainService();
 101
 102                         DatabaseDefinition dbdef = new DatabaseDefinition("alias");
 103
 104                         TableDefinition categoryTable = new TableDefinition("categories", dbdef );
 105                         categoryTable.AddColumn( new ColumnDefinition("id", true, false, true, false, OleDbType.Integer) );
 106                         categoryTable.AddColumn( new ColumnDefinition("name", false, false, false, false, OleDbType.VarChar) );
 107                         categoryTable.AddColumn( new ColumnDefinition("parent_id", false, true, false, false, OleDbType.Integer, categoryTable) );
 108
 109                         categoryTable.AddManyRelation(categoryTable);
 110
 111                         BuildContext context = new BuildContext();
 112
 113                         ActiveRecordDescriptor arDesc = new ActiveRecordDescriptor();
 114
 115                         ActiveRecordPropertyDescriptor[] descs = relService.InferRelations( arDesc, categoryTable, context );
 116
 117                         Assert.IsFalse(context.HasPendents);
 118
 119                         Assert.IsNotNull(descs);
 120                         Assert.AreEqual( 2, descs.Length );
 121
 122                         ActiveRecordHasManyDescriptor desc1 = descs[0] as ActiveRecordHasManyDescriptor;
 123                         Assert.IsNotNull(desc1);
 124                         Assert.IsNotNull(desc1.TargetType);
 125                         Assert.IsNotNull(desc1.PropertyType);
 126
 127                         Assert.AreEqual( "Categories", desc1.PropertyName );
 128                         Assert.AreEqual( "parent_id", desc1.ColumnName );
 129                         Assert.AreEqual( typeof(IList), desc1.PropertyType );
 130
 131                         ActiveRecordBelongsToDescriptor desc2 = descs[1] as ActiveRecordBelongsToDescriptor;
 132                         Assert.IsNotNull(desc2);
 133                         Assert.IsNotNull(desc2.TargetType);
 134                         Assert.IsNull(desc2.PropertyType);
 135                         Assert.AreEqual( "Category", desc2.PropertyName );
 136                         Assert.AreEqual( "parent_id", desc2.ColumnName );
 137                 }
 138
 139                 private void InitKernel()
 140                 {
 141                         Kernel.AddComponent( "relationsService", typeof(IRelationshipInferenceService), typeof(RelationshipInferenceService) );
 142                         Kernel.AddComponent( "nameService", typeof(INamingService), typeof(NamingService) );
 143                         Kernel.AddComponent( "typeinf", typeof(ITypeInferenceService), typeof(TypeInferenceService) );
 144                 }
 145
 146                 private IRelationshipInferenceService ObtainService()
 147                 {
 148                         return Kernel[ typeof(IRelationshipInferenceService) ] as IRelationshipInferenceService;
 149                 }
 150
 151                 private void BuildBlogPostsStructure(out TableDefinition blogTable, out TableDefinition postTable)
 152                 {
 153                         DatabaseDefinition dbdef = new DatabaseDefinition("alias");
 154
 155                         blogTable = new TableDefinition("blogs", dbdef );
 156                         blogTable.AddColumn( new ColumnDefinition("id", true, false, true, false, OleDbType.Integer) );
 157                         blogTable.AddColumn( new ColumnDefinition("name", false, false, false, false, OleDbType.VarChar) );
 158
 159                         postTable = new TableDefinition("posts", dbdef );
 160                         postTable.AddColumn( new ColumnDefinition("id", true, false, true, false, OleDbType.Integer) );
 161                         postTable.AddColumn( new ColumnDefinition("name", false, false, false, false, OleDbType.VarChar) );
 162                         postTable.AddColumn( new ColumnDefinition("blog_id", false, true, false, false, OleDbType.VarChar, blogTable) );
 163
 164                         blogTable.AddManyRelation(postTable);
 165                 }
 166         }
 167 }