From b81bddb0851d2cfcfc74fdcd89b31913d88ff02c Mon Sep 17 00:00:00 2001
From: Voronsky <roflocopter911@gmail.com>
Date: Wed, 28 Aug 2024 23:48:54 -0400
Subject: [PATCH] Refactored Unit tests based on rework

---
 .../java/com/voronsky/unifi4j/UnifiTests.java | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

diff --git a/src/test/java/com/voronsky/unifi4j/UnifiTests.java b/src/test/java/com/voronsky/unifi4j/UnifiTests.java
index a1410a1..6a195da 100644
--- a/src/test/java/com/voronsky/unifi4j/UnifiTests.java
+++ b/src/test/java/com/voronsky/unifi4j/UnifiTests.java
@@ -13,32 +13,21 @@ public class UnifiTests {
 
     @Test
     public void GetHostTest(){
-        ResponseEntity<String> response;
+        String response;
         System.out.println("UNIFI_KEY: "+System.getenv("UNIFI_KEY"));
         Unifi unifi = new Unifi("https://api.ui.com/ea", System.getenv("UNIFI_KEY"));
         response = unifi.getHosts();
-        System.out.println("Status Code:" + response.getStatusCode());
-        System.out.println("Content: "+response.getBody());
-
-        assert response.getStatusCode().is2xxSuccessful();
-        assert !response.getBody().isEmpty();
-
-        try{
-            JSONObject jsonObject = new JSONObject(response.getBody());
-            System.out.println(jsonObject);
-        } catch (JSONException err){
-            System.out.println(err.toString());
-        }
+        System.out.println(response);
 
     }
 
     @Test
-    public void GetHardwaresTest(){
+    public void GetHardwareListTest(){
         List<String> response = new ArrayList<>();
         System.out.println("UNIFI_KEY: "+System.getenv("UNIFI_KEY"));
         Unifi unifi = new Unifi("https://api.ui.com/ea", System.getenv("UNIFI_KEY"));
         try {
-            response = unifi.getHardware();
+            response = unifi.getHardwareList();
         }
         catch(Exception e){
             System.out.println("Exception: "+e);