require 'rgen/ecore/ecore'

# This "light" version of the ECore model checker is used to check the
# model produced by the XMLInstantiatorTest only. 
# 
module SimpleECoreModelChecker			
  include RGen::ECore
  
  def checkECoreModel(env)
    
    # check main package
    mainPackage = env.elements.select {|e| e.is_a? EPackage and e.name == "HouseMetamodel"}.first
    assert mainPackage != nil
    
    # check Rooms package
    assert mainPackage.eSubpackages.is_a?(Array)
    assert_equal 1, mainPackage.eSubpackages.size
    assert mainPackage.eSubpackages[0].is_a?(EPackage)
    roomsPackage = mainPackage.eSubpackages[0]
    assert_equal "Rooms", roomsPackage.name
    
    # check main package classes
    assert mainPackage.eClassifiers.is_a?(Array)
    assert_equal 3, mainPackage.eClassifiers.size
    assert mainPackage.eClassifiers.all?{|c| c.is_a?(EClass)}
    houseClass = mainPackage.eClassifiers.select{|c| c.name == "House"}.first
    personClass = mainPackage.eClassifiers.select{|c| c.name == "Person"}.first
    meetingPlaceClass = mainPackage.eClassifiers.select{|c| c.name == "MeetingPlace"}.first
    assert houseClass != nil
    assert personClass != nil
    assert meetingPlaceClass != nil
    
    # check Rooms package classes
    assert roomsPackage.eClassifiers.is_a?(Array)
    assert_equal 3, roomsPackage.eClassifiers.size
    assert roomsPackage.eClassifiers.all?{|c| c.is_a?(EClass)}
    roomClass = roomsPackage.eClassifiers.select{|c| c.name == "Room"}.first
    kitchenClass = roomsPackage.eClassifiers.select{|c| c.name == "Kitchen"}.first
    bathroomClass = roomsPackage.eClassifiers.select{|c| c.name == "Bathroom"}.first
    assert roomClass != nil
    assert kitchenClass != nil
    assert bathroomClass != nil
    
    # check Room inheritance
    assert kitchenClass.eSuperTypes.is_a?(Array)
    assert_equal 2, kitchenClass.eSuperTypes.size
    assert_equal roomClass.object_id, kitchenClass.eSuperTypes.select{|c| c.name == "Room"}.first.object_id
    assert_equal meetingPlaceClass.object_id, kitchenClass.eSuperTypes.select{|c| c.name == "MeetingPlace"}.first.object_id
    assert bathroomClass.eSuperTypes.is_a?(Array)
    assert_equal 1, bathroomClass.eSuperTypes.size
    assert_equal roomClass.object_id, bathroomClass.eSuperTypes[0].object_id
    
    # check House-Room "part of" association
    assert houseClass.eAllContainments.eType.is_a?(Array)
    assert_equal 1, houseClass.eAllContainments.eType.size
    roomRef = houseClass.eAllContainments.first
    assert_equal roomClass.object_id, roomRef.eType.object_id
    assert_equal "room", roomRef.name
    assert_equal 1, roomRef.lowerBound
    assert_equal(-1, roomRef.upperBound)
    assert roomRef.eOpposite != nil
    assert_equal houseClass.object_id, roomRef.eOpposite.eType.object_id
    
    partOfRefs = roomClass.eReferences.select{|r| r.eOpposite && r.eOpposite.containment}
    assert_equal 1, partOfRefs.size
    assert_equal houseClass.object_id, partOfRefs.first.eType.object_id
    assert_equal "house", partOfRefs.first.name
    assert_equal roomRef.object_id, partOfRefs.first.eOpposite.object_id
    
    # check House OUT associations
    assert houseClass.eReferences.is_a?(Array)
    assert_equal 3, houseClass.eReferences.size
    bathRef = houseClass.eReferences.find {|e| e.name == "bathroom"}
    kitchenRef = houseClass.eReferences.find {|e| e.name == "kitchen"}
    roomRef = houseClass.eReferences.find {|e| e.name == "room"}
    assert bathRef != nil
    assert_nil bathRef.eOpposite
    assert kitchenRef != nil
    assert roomRef != nil
    assert_equal 1, kitchenRef.lowerBound
    assert_equal 1, kitchenRef.upperBound
    assert_equal 1, roomRef.lowerBound
    assert_equal(-1, roomRef.upperBound)
    
    # check House IN associations
    houseInRefs = env.find(:class => EReference, :eType => houseClass)
    assert_equal 3, houseInRefs.size
    homeEnd = houseInRefs.find{|e| e.name == "home"}
    assert homeEnd != nil
    assert_equal 0, homeEnd.lowerBound
    assert_equal(-1, homeEnd.upperBound)
    
  end
end