[Prover Service] Cleanups and improvements to the existing tests (#106)

* [Prover Service] Remove playground.rs tests.

* [Prover Service] Pass over training_wheels.rs and add failure test.

* [Prover Service] Reshuffle common test code.

* [Prover Service] Clean ups to smoke tests.

* [Prover Service] Rename smoke.rs -> prover_handler.rs

* [Prover Service] Clean up to test utils.

* [Prover Service] Clean up to test types.rs

Author: JoshLind

prover-service/Cargo.toml

@@ -53,3 +53,6 @@ tokio = { workspace = true }
 
 [dev-dependencies]
 aptos-time-service = { workspace = true, features = ["testing"] }
+
+[package.metadata.cargo-machete]
+ignored = ["hex"]

prover-service/src/tests/common/mod.rs

@@ -4,8 +4,8 @@ use crate::external_resources::jwk_fetcher::{
     get_federated_jwk, AUTH0_ISSUER_NAME, AUTH0_REGEX_STR, COGNITO_ISSUER_NAME, COGNITO_REGEX_STR,
 };
 use crate::external_resources::jwk_types::{FederatedJWKIssuerInterface, FederatedJWKs, KeyID};
-use crate::tests::common::gen_test_jwk_keypair_with_kid_override;
-use crate::tests::common::types::{ProofTestCase, TestJWTPayload};
+use crate::tests::types::{ProofTestCase, TestJWTPayload};
+use crate::tests::utils;
 use aptos_keyless_common::input_processing::jwt::DecodedJWT;
 use aptos_types::jwks::rsa::{INSECURE_TEST_RSA_JWK, RSA_JWK};
 use regex::Regex;
@@ -189,7 +189,7 @@ async fn test_federated_jwk_fetch_fails_for_missing_kid() {
 fn get_decoded_jwt(kid: &str, jwt_payload: TestJWTPayload) -> DecodedJWT {
     // Create the prover request input
     let testcase = ProofTestCase::default_with_payload(jwt_payload).compute_nonce();
-    let jwk_keypair = gen_test_jwk_keypair_with_kid_override(kid);
+    let jwk_keypair = utils::generate_test_jwk_keypair_with_kid(kid);
     let prover_request_input = testcase.convert_to_prover_request(&jwk_keypair);
 
     // Return the decoded JWT

prover-service/src/tests/common/rsa.rs

@@ -1,9 +1,8 @@
 // Copyright (c) Aptos Foundation
-
-pub mod common;
 pub mod federated_jwk;
 pub mod jwk_fetcher;
-pub mod playground;
+pub mod prover_handler;
 pub mod request_handler;
-pub mod smoke;
 pub mod training_wheels;
+pub mod types;
+pub mod utils;