internal/rosa: pass stage alongside state

This cleans up many function signatures.

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 5 insertions, 5 deletions

diff --git a/internal/rosa/state_test.go b/internal/rosa/state_test.go
index 2b951a24..8b01f7d3 100644
--- a/internal/rosa/state_test.go
+++ b/internal/rosa/state_test.go
@@ -13,21 +13,21 @@ func TestLoad(t *testing.T) {
 		t.Run(rosa.Native().MustGet(p).Name, func(t *testing.T) {
 			t.Parallel()
 
-			rosa.Native().Load(rosa.Std, p)
+			rosa.Native().Std().MustLoad(p)
 		})
 	}
 }
 
 func BenchmarkAll(b *testing.B) {
-	s := rosa.Native().Clone()
+	t := rosa.Native().Clone().Std()
 
 	for b.Loop() {
-		for _, p := range s.Collect() {
-			s.Load(rosa.Std, p)
+		for _, p := range t.Collect() {
+			t.MustLoad(p)
 		}
 
 		b.StopTimer()
-		s.DropCaches("", 0)
+		t.DropCaches("", 0)
 		b.StartTimer()
 	}
 }