Lesson 7 added

AfterSuite.java

@@ -0,0 +1,11 @@
+package com.gb;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.METHOD)
+public @interface AfterSuite {
+}

BeforeSuite.java

@@ -0,0 +1,11 @@
+package com.gb;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.METHOD)
+public @interface BeforeSuite {
+}

Main.java

@@ -0,0 +1,7 @@
+package com.gb;
+
+public class Main {
+    public static void main(String[] args) {
+        StartTest.start(MyTest.class);
+    }
+}

MyTest.java

@@ -0,0 +1,34 @@
+package com.gb;
+
+public class MyTest {
+
+    @BeforeSuite
+    public static void test2() {
+        System.out.println("test2 BeforeSuite");
+    }
+
+    @Test
+    public static void test8() {
+        System.out.println("test8");
+    }
+
+    @Test
+    public static void test1() {
+        System.out.println("test1");
+    }
+
+    @Test(priority = 2)
+    public static void test3() {
+        System.out.println("test3 priority = 2 ");
+    }
+
+    @Test(priority = 7)
+    public static void test4() {
+        System.out.println("test4 priority = 7 ");
+    }
+
+    @AfterSuite
+    public static void test5() {
+        System.out.println("test5 AfterSuite");
+    }
+}

StartTest.java

@@ -0,0 +1,54 @@
+package com.gb;
+
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.util.ArrayList;
+import java.util.Comparator;
+import java.util.List;
+
+public class StartTest {
+    public static void start(Class c) {
+        List<Method> methods = new ArrayList<>();
+        Method[] classMethods = c.getDeclaredMethods();
+
+        for (Method m : classMethods) {
+            if (m.isAnnotationPresent(Test.class)) {
+                methods.add(m);
+            }
+        }
+
+        methods.sort(Comparator
+                .comparingInt((Method i) -> i.getAnnotation(Test.class).priority())
+                .reversed());
+
+
+        for (Method m : classMethods) {
+            if (m.isAnnotationPresent(BeforeSuite.class)) {
+                if (methods.size() > 0 && methods.get(0).isAnnotationPresent(BeforeSuite.class)) {
+                    throw new RuntimeException("@BeforeSuite annotation method > 1");
+                }
+                methods.add(0, m);
+            }
+        }
+
+        for (Method m : classMethods) {
+            if (m.isAnnotationPresent(AfterSuite.class)) {
+                if (methods.size() > 0 && methods.get(methods.size() - 1).isAnnotationPresent(AfterSuite.class)) {
+                    throw new RuntimeException("@AfterSuite annotation method > 1");
+                }
+                methods.add(m);
+            }
+        }
+
+        for (Method m : methods) {
+            try {
+                m.invoke(null);
+            } catch (IllegalAccessException e) {
+                e.printStackTrace();
+            } catch (InvocationTargetException e) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+}

Test.java

@@ -0,0 +1,12 @@
+package com.gb;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.METHOD)
+public @interface Test {
+    int priority() default 5;
+}