EntityLoader_Test.cs 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. using System.ComponentModel.DataAnnotations.Schema;
  2. using Microsoft.VisualStudio.TestTools.UnitTesting;
  3. namespace Vitorm.MsTest.CommonTest
  4. {
  5. [TestClass]
  6. public class EntityLoader_Test
  7. {
  8. [TestMethod]
  9. public void Test_EntityDescriptor()
  10. {
  11. using var dbContext = DataSource.CreateDbContext();
  12. {
  13. var entityDescriptor = dbContext.GetEntityDescriptor<User>();
  14. Assert.AreEqual("id", entityDescriptor.key?.columnName);
  15. Assert.AreEqual("User", entityDescriptor.tableName);
  16. }
  17. {
  18. var entityDescriptor = dbContext.GetEntityDescriptor<User2>();
  19. Assert.IsNull(entityDescriptor.key);
  20. Assert.AreEqual("User", entityDescriptor.tableName);
  21. }
  22. {
  23. var entityDescriptor = dbContext.GetEntityDescriptor<User3>();
  24. Assert.IsNull(entityDescriptor);
  25. }
  26. }
  27. #region Custom Entity
  28. public class User
  29. {
  30. public int id { get; set; }
  31. public string name { get; set; }
  32. }
  33. [Vitorm.Entity.Loader.DataAnnotations.StrictEntityLoader]
  34. [Table("User")]
  35. public class User2 : User { }
  36. [Vitorm.Entity.Loader.DataAnnotations.StrictEntityLoader]
  37. public class User3 : User { }
  38. #endregion
  39. }
  40. }