Biztalk 2010 X12 EDI Parties

Question

I am trying to convert some BizTalk 2006 R2 helper code to BizTalk 2010 and I am running into a peculiar issue. I am trying to unit test a method that had a breaking API change from 2006 R2 -> 2010 and I keep getting the following exception when I try to access the batches of the party:

System.Data.SqlClient.SqlException: Could not find stored procedure 'edi_PartnerBatchScheduleSelect'.

Code:

[TestMethod()]
public void GetPartyBatchStatusTest()
{
    Assert.IsTrue(GetPartyBatchStatus("Party1"));
}

public bool GetPartyBatchStatus(string PartyName)
{
    if (string.IsNullOrEmpty(PartyName))
    {
        // Throw Exception
        throw new System.ArgumentException("Parameter PartyName cannot be null or empty in the GetPartyBatchStatus method.", "PartyName");
    }

    bool RetVal = false;

    Partner objPartner = new Partner(PartyName);

    if (objPartner.PartyId != -1)
    {
        foreach (IPartnerBatch batch in objPartner.Batches.Batches)
        {
            RetVal = batch.BatchingActivated;
        }
    }

    return RetVal;
}

For this test case, I have set up a Party1 and a Party2 and started a batch between them.