1313using LTTngDataExtensions . SourceDataCookers . Module ;
1414using LTTngDataExtensions . SourceDataCookers . Syscall ;
1515using LTTngDataExtensions . SourceDataCookers . Thread ;
16+ using LTTngDataExtensions . Tables ;
1617using Microsoft . Performance . SDK . Extensibility ;
1718using Microsoft . Performance . SDK . Processing ;
1819using Microsoft . Performance . Toolkit . Engine ;
1920using Microsoft . VisualStudio . TestTools . UnitTesting ;
20-
21+ using UnitTestCommon ;
22+
2123namespace LTTngDataExtUnitTest
2224{
2325 [ TestClass ]
@@ -74,6 +76,18 @@ public static void ProcessTrace()
7476 LTTngDiskDataCookerPath = lttngDiskDataCooker . Path ;
7577 runtime . EnableCooker ( LTTngDiskDataCookerPath ) ;
7678
79+ // Enable tables used by UI
80+ //runtime.EnableTable(CpuTable.TableDescriptor);
81+ runtime . EnableTable ( DiagnosticMessageTable . TableDescriptor ) ;
82+ runtime . EnableTable ( DiskTable . TableDescriptor ) ;
83+ runtime . EnableTable ( ExecutionEventTable . TableDescriptor ) ;
84+ runtime . EnableTable ( FileEventsTable . TableDescriptor ) ;
85+ runtime . EnableTable ( GenericEventTable . TableDescriptor ) ;
86+ runtime . EnableTable ( ModuleEventsTable . TableDescriptor ) ;
87+ //runtime.EnableTable(ProcessTable.TableDescriptor);
88+ runtime . EnableTable ( SyscallTable . TableDescriptor ) ;
89+ runtime . EnableTable ( ThreadTable . TableDescriptor ) ;
90+
7791 //
7892 // Process our data.
7993 //
@@ -120,7 +134,7 @@ public void ProcessTraceAsFolder()
120134 }
121135
122136 [ TestMethod ]
123- public void DiagnosticMessageTable ( )
137+ public void DiagnosticMessageTableTest ( )
124138 {
125139 ProcessTrace ( ) ;
126140
@@ -130,10 +144,12 @@ public void DiagnosticMessageTable()
130144 nameof ( LTTngDmesgDataCooker . DiagnosticMessages ) ) ) ;
131145
132146 Assert . IsTrue ( eventData . Count == 0 ) ; // TODO - UT - Trace has no DiagMessages
147+
148+ UnitTest . TestTableBuild ( RuntimeExecutionResults , DiagnosticMessageTable . TableDescriptor , 0 , true ) ;
133149 }
134150
135151 [ TestMethod ]
136- public void DiskTable ( )
152+ public void DiskTableTest ( )
137153 {
138154 ProcessTrace ( ) ;
139155
@@ -142,11 +158,13 @@ public void DiskTable()
142158 LTTngDiskDataCookerPath ,
143159 nameof ( LTTngDiskDataCooker . DiskActivity ) ) ) ;
144160
145- Assert . IsTrue ( eventData . Count > 0 ) ;
161+ Assert . IsTrue ( eventData . Count == 6437 ) ;
162+
163+ UnitTest . TestTableBuild ( RuntimeExecutionResults , DiskTable . TableDescriptor , 6437 ) ;
146164 }
147165
148166 [ TestMethod ]
149- public void ExecutionEventTable ( )
167+ public void ExecutionEventTableTest ( )
150168 {
151169 ProcessTrace ( ) ;
152170
@@ -155,11 +173,12 @@ public void ExecutionEventTable()
155173 LTTngThreadDataCookerPath ,
156174 nameof ( LTTngThreadDataCooker . ExecutionEvents ) ) ) ;
157175
158- Assert . IsTrue ( eventData . Count > 0 ) ;
176+ Assert . IsTrue ( eventData . Count == 18780 ) ;
177+ UnitTest . TestTableBuild ( RuntimeExecutionResults , ExecutionEventTable . TableDescriptor , 18780 ) ;
159178 }
160179
161180 [ TestMethod ]
162- public void FileEventsTable ( )
181+ public void FileEventsTableTest ( )
163182 {
164183 ProcessTrace ( ) ;
165184
@@ -168,11 +187,12 @@ public void FileEventsTable()
168187 LTTngDiskDataCookerPath ,
169188 nameof ( LTTngDiskDataCooker . FileEvents ) ) ) ;
170189
171- Assert . IsTrue ( eventData . Count > 0 ) ;
190+ Assert . IsTrue ( eventData . Count == 433143 ) ;
191+ UnitTest . TestTableBuild ( RuntimeExecutionResults , FileEventsTable . TableDescriptor , 433143 ) ;
172192 }
173193
174194 [ TestMethod ]
175- public void GenericEventsTable ( )
195+ public void GenericEventsTableTest ( )
176196 {
177197 ProcessTrace ( ) ;
178198
@@ -181,14 +201,15 @@ public void GenericEventsTable()
181201 LTTngGenericEventDataCookerPath ,
182202 nameof ( LTTngGenericEventDataCooker . Events ) ) ) ;
183203
184- Assert . IsTrue ( eventData . Count > 0 ) ;
204+ Assert . IsTrue ( eventData . Count == 936356 ) ;
205+ UnitTest . TestTableBuild ( RuntimeExecutionResults , GenericEventTable . TableDescriptor , 936356 ) ;
185206
186207 Assert . IsTrue ( ! String . IsNullOrWhiteSpace ( eventData [ 0 ] . FieldNames [ 0 ] ) ) ;
187208 Assert . IsTrue ( ! String . IsNullOrWhiteSpace ( eventData [ 0 ] . FieldValues [ 0 ] ) ) ;
188209 }
189210
190211 [ TestMethod ]
191- public void ModuleEventsTable ( )
212+ public void ModuleEventsTableTest ( )
192213 {
193214 ProcessTrace ( ) ;
194215
@@ -198,10 +219,11 @@ public void ModuleEventsTable()
198219 nameof ( LTTngModuleDataCooker . ModuleEvents ) ) ) ;
199220
200221 Assert . IsTrue ( eventData . Count == 0 ) ; // TODO - UT - Trace has no ModuleEvents
222+ UnitTest . TestTableBuild ( RuntimeExecutionResults , ModuleEventsTable . TableDescriptor , 0 , true ) ;
201223 }
202224
203225 [ TestMethod ]
204- public void SyscallTable ( )
226+ public void SyscallTableTest ( )
205227 {
206228 ProcessTrace ( ) ;
207229
@@ -210,11 +232,12 @@ public void SyscallTable()
210232 LTTngSyscallDataCookerPath ,
211233 nameof ( LTTngSyscallDataCooker . Syscalls ) ) ) ;
212234
213- Assert . IsTrue ( eventData . Count > 0 ) ;
235+ Assert . IsTrue ( eventData . Count == 441037 ) ;
236+ UnitTest . TestTableBuild ( RuntimeExecutionResults , SyscallTable . TableDescriptor , 441037 ) ;
214237 }
215238
216239 [ TestMethod ]
217- public void ThreadTable ( )
240+ public void ThreadTableTest ( )
218241 {
219242 ProcessTrace ( ) ;
220243
@@ -223,7 +246,8 @@ public void ThreadTable()
223246 LTTngThreadDataCookerPath ,
224247 nameof ( LTTngThreadDataCooker . Threads ) ) ) ;
225248
226- Assert . IsTrue ( threads . Count > 0 ) ;
249+ Assert . IsTrue ( threads . Count == 82 ) ;
250+ UnitTest . TestTableBuild ( RuntimeExecutionResults , ThreadTable . TableDescriptor , 82 ) ;
227251 }
228252 }
229253}
0 commit comments