public void TearDown() { if (TestContext.CurrentContext.Test.Name.Contains("1")) { ExtentBase.GetTargetTest().AssignCategory("Category1"); } else if (TestContext.CurrentContext.Test.Name.Contains("2")) { ExtentBase.GetTargetTest().AssignCategory("Category2"); } ExtentBase.FlushExtent(); }