RoleControllerTest

src/main/java/application/controllers/RoleController.java

@@ -19,13 +19,19 @@ import org.springframework.web.bind.annotation.ResponseBody;
 @Controller
 @RequestMapping(value = "/roles")
 public class RoleController {
+
+    public void setRoleRepository(RoleRepository roleRepository) {
+        this.roleRepository = roleRepository;
+    }
+
     @Autowired
     private RoleRepository roleRepository;
 
     @RequestMapping(value = "/add")
-    public void addRole(@RequestParam String rolename,@RequestParam String password) {
+    public Role addRole(@RequestParam String rolename,@RequestParam String password) {
         Role role = new Role(rolename,password);
-        roleRepository.save(role);
+        Role result = roleRepository.save(role);
+        return  result;
     }
 
     @RequestMapping(value = "/addSampleRoles")

src/main/java/application/model/Role.java

@@ -9,6 +9,10 @@ import java.util.Set;
  */
 @Entity (name = "role")
 public class Role {
+    public void setRoleId(long roleId) {
+        this.roleId = roleId;
+    }
+
     @Id
     @Column(name = "ID")
     @GeneratedValue(strategy = GenerationType.AUTO)
@@ -63,5 +67,13 @@ public class Role {
         return true;
     }
 
-
+    @Override
+    public String toString() {
+        return "Role{" +
+                "roleId=" + roleId +
+                ", roleName='" + roleName + '\'' +
+                ", password='" + password + '\'' +
+                ", users=" + users +
+                '}';
+    }
 }