Improve examples and tests

Author: ghostdogpr

entities/src/test/scala/com/devsisters/shardcake/ShardingSpec.scala

@@ -22,7 +22,7 @@ object ShardingSpec extends DefaultRunnableSpec {
   def spec: ZSpec[TestEnvironment, Throwable] =
     suite("ShardingSpec")(
       testM("Send message to entities") {
-        (Sharding.registerManaged *> Sharding.registerEntity(Counter, behavior)).use { _ =>
+        (Sharding.registerEntity(Counter, behavior) *> Sharding.registerManaged).use { _ =>
           for {
             counter <- Sharding.messenger(Counter)
             _       <- counter.sendDiscard("c1")(IncrementCounter)

examples/src/main/scala/example/complex/GuildApp.scala

@@ -16,7 +16,7 @@ object GuildApp extends zio.App {
       .toLayer
 
   val program =
-    Sharding.registerEntity(Guild, behavior).use { _ =>
+    (Sharding.registerEntity(Guild, behavior) *> Sharding.registerManaged).use { _ =>
       Sharding.messenger(Guild).map { guild =>
         for {
           user1 <- random.nextUUID.map(_.toString)
@@ -41,5 +41,5 @@ object GuildApp extends zio.App {
   val layer    = sharding ++ service ++ redis
 
   def run(args: List[String]): URIO[ZEnv, ExitCode] =
-    Sharding.registerManaged.use(_ => program).provideLayer(layer).exitCode
+    program.provideLayer(layer).exitCode
 }

examples/src/main/scala/example/simple/GuildApp.scala

@@ -10,7 +10,7 @@ import zio.random.Random
 
 object GuildApp extends zio.App {
   val program =
-    Sharding.registerEntity(Guild, behavior).use { _ =>
+    (Sharding.registerEntity(Guild, behavior) *> Sharding.registerManaged).use { _ =>
       Sharding.messenger(Guild).map { guild =>
         for {
           _ <- guild.send("guild1")(Join("user1", _)).debug
@@ -34,5 +34,5 @@ object GuildApp extends zio.App {
   val layer    = sharding ++ service
 
   def run(args: List[String]): URIO[ZEnv, ExitCode] =
-    Sharding.registerManaged.use(_ => program).provideLayer(layer).exitCode
+    program.provideLayer(layer).exitCode
 }

examples/src/test/scala/example/EndToEndSpec.scala

@@ -83,7 +83,7 @@ object EndToEndSpec extends DefaultRunnableSpec {
   def spec =
     suite("EndToEndSpec")(
       testM("Send message to entities") {
-        (Sharding.registerManaged *> Sharding.registerEntity(Guild, GuildBehavior.behavior)).use { _ =>
+        (Sharding.registerEntity(Guild, GuildBehavior.behavior) *> Sharding.registerManaged).use { _ =>
           for {
             guild   <- Sharding.messenger(Guild)
             _       <- guild.send("guild1")(Join("user1", _))