public void SetupTestClass() { GenerationContext.TestClass.IsPartial = true; GenerationContext.TestClass.TypeAttributes |= TypeAttributes.Public; _codeDomHelper.AddLinePragmaInitial(GenerationContext.TestClass, GenerationContext.Document.SourceFilePath, _specFlowConfiguration); _testGeneratorProvider.SetTestClass(GenerationContext, GenerationContext.Feature.Name, GenerationContext.Feature.Description); _decoratorRegistry.DecorateTestClass(GenerationContext, out List <string> unprocessedTags); if (!unprocessedTags.Any()) { return; } _testGeneratorProvider.SetTestClassCategories(GenerationContext, unprocessedTags); }