test_that("Diverted ungauged nodes have correct color", {
  nodes_div <- loadSevernNodes()
  nodes_div$model[nodes_div$id == "54029"] <- "Ungauged"
  nodes_div <- rbind(nodes_div, data.frame(id = "54029",
                                           down = "54002",
                                           length = 20,
                                           model = "Diversion",
                                           area = NA))
  griwrm_div <- CreateGRiwrm(nodes_div)
  mmd <- plot(griwrm_div, display = FALSE)
  expect_true(any(grepl("id_54029 UpstreamUngaugedDiversion", strsplit(mmd, "\n\n")[[1]])))
})

test_that("Ungauged nodes and donors are in a box!", {
  nds <- loadSevernNodes()
  nds$donor <- as.character(NA)
  nds$model[nds$id %in% c("54001", "54032", "54029")] <- "Ungauged"
  nds$donor[nds$id %in% c("54001", "54029")] <- "54095"
  g <- CreateGRiwrm(nds)
  expect_s3_class(sortGRiwrm4plot(g), "GRiwrm")
  expect_equal(sortGRiwrm4plot(g)$id,
               c("54095", "54001", "54029", "54002", "54032", "54057"))
  s <- plot(g, display = FALSE)
  expect_equal(strsplit(s, "\n")[[1]][c(5,13)], c("subgraph donor_54095 [54095]", "end"))
})

test_that("Single node plot does not crash", {
  sgl_node <- loadSevernNodes()[1,]
  g <- CreateGRiwrm(sgl_node)
  mmd <- plot(g, display = FALSE)
  expect_false(grepl("-->", mmd, fixed = TRUE))
})