User: add 2x unit test for notGraphCilentSourced

mypricehealth · Aug 30, 2021 · 405ecf0 · 405ecf0
1 parent 80c7018
commit 405ecf0
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/User_test.go b/User_test.go
@@ -53,6 +53,13 @@ func TestUser_ListCalendars(t *testing.T) {
 	if skipCalendarTests {
 		t.Skip("Skipping due to missing 'MSGraphExistingCalendarsOfUser' value")
 	}
+	// testing for ErrNotGraphClientSourced
+	notGraphClientSourcedUser := User{ID: "none"}
+	_, err := notGraphClientSourcedUser.ListCalendars()
+	if err != ErrNotGraphClientSourced {
+		t.Errorf("Expected error \"ErrNotGraphClientSourced\", but got: %v", err)
+	}
+	// continue with normal tests
 	userToTest := GetTestUser(t)
 
 	var wantedCalendars []Calendar
@@ -160,6 +167,14 @@ func TestUser_String(t *testing.T) {
 }
 
 func TestUser_UpdateUser(t *testing.T) {
+	// testing for ErrNotGraphClientSourced
+	notGraphClientSourcedUser := User{ID: "none"}
+	err := notGraphClientSourcedUser.UpdateUser(User{})
+	if err != ErrNotGraphClientSourced {
+		t.Errorf("Expected error \"ErrNotGraphClientSourced\", but got: %v", err)
+	}
+
+	// continue with normal tests
 	testuser := createUnitTestUser(t)
 
 	targetedCompanyName := "go-msgraph unit test suite UpdateUser" + randomString(25)