143 lines
2.2 KiB
C++
143 lines
2.2 KiB
C++
|
// Suites.cpp: implementation of the CSuites class.
|
||
|
|
//
|
||
|
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
#include "stdafx.h"
|
||
|
|
#include "TSDiag.h"
|
||
|
|
#include "Suites.h"
|
||
|
|
|
||
|
|
#include "suite.h"
|
||
|
|
#include "testdata.h"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
// Construction/Destruction
|
||
|
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
CSuites::CSuites()
|
||
|
|
{
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
CSuites::~CSuites()
|
||
|
|
{
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CSuites::get_Count(long *pVal)
|
||
|
|
{
|
||
|
|
if (!pVal)
|
||
|
|
return E_POINTER;
|
||
|
|
|
||
|
|
*pVal = GlobalTestData.GetSuiteCount();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CSuites::get__NewEnum(LPUNKNOWN *pVal)
|
||
|
|
{
|
||
|
|
if (!pVal)
|
||
|
|
return E_POINTER;
|
||
|
|
|
||
|
|
return E_NOTIMPL;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CSuites::get_Item(VARIANT Index, VARIANT *pVal)
|
||
|
|
{
|
||
|
|
if (!pVal)
|
||
|
|
{
|
||
|
|
return E_POINTER;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
// BUGBUG find right index value.
|
||
|
|
DWORD dwIndex;
|
||
|
|
if (!SuiteIndexFromVarient(Index, &dwIndex ))
|
||
|
|
{
|
||
|
|
return E_INVALIDARG;
|
||
|
|
}
|
||
|
|
|
||
|
|
ASSERT(dwIndex < GlobalTestData.GetSuiteCount());
|
||
|
|
|
||
|
|
CComObject<CSuite> *pSuite;
|
||
|
|
HRESULT hr = CComObject<CSuite>::CreateInstance(&pSuite);
|
||
|
|
if (FAILED(hr))
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
// if we need to initialize our Suite object, it should be done here.
|
||
|
|
pSuite->SetSuiteIndex(dwIndex);
|
||
|
|
|
||
|
|
IDispatch* pDisp = NULL;
|
||
|
|
|
||
|
|
hr = pSuite->QueryInterface(&pDisp);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
VariantInit(pVal);
|
||
|
|
pVal->vt = VT_DISPATCH;
|
||
|
|
pVal->pdispVal = pDisp;
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
delete pSuite;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
bool CSuites::SuiteIndexFromVarient(const VARIANT &Index, DWORD *pdwIndex)
|
||
|
|
{
|
||
|
|
ASSERT(pdwIndex);
|
||
|
|
*pdwIndex = 0xffffffff;
|
||
|
|
|
||
|
|
switch(Index.vt)
|
||
|
|
{
|
||
|
|
case VT_I4:
|
||
|
|
case VT_UI2:
|
||
|
|
case VT_UINT:
|
||
|
|
case VT_INT:
|
||
|
|
case VT_I2:
|
||
|
|
{
|
||
|
|
if ((Index.iVal >= 0) && (DWORD(Index.iVal) < GlobalTestData.GetSuiteCount()))
|
||
|
|
{
|
||
|
|
*pdwIndex = Index.iVal;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case VT_BSTR :
|
||
|
|
{
|
||
|
|
if (!Index.bstrVal)
|
||
|
|
{
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
for (DWORD dw = 0; dw < GlobalTestData.GetSuiteCount(); dw++)
|
||
|
|
{
|
||
|
|
if (_tcscmp(GlobalTestData.GetSuiteName(dw), Index.bstrVal) == 0)
|
||
|
|
{
|
||
|
|
// ok we got the index
|
||
|
|
*pdwIndex = dw;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
return false;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|