From cc5fde41bd62f1623171ba8376128fda68a6aa4e Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Thu, 9 Sep 2021 22:51:39 +0800
Subject: [PATCH] Testing Vec<u8>

---
 tests/common/bakery_chain/metadata.rs |  1 +
 tests/common/setup/schema.rs          |  1 +
 tests/primary_key_tests.rs            | 19 ++++++++++++-------
 3 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/tests/common/bakery_chain/metadata.rs b/tests/common/bakery_chain/metadata.rs
index b9707059..69579492 100644
--- a/tests/common/bakery_chain/metadata.rs
+++ b/tests/common/bakery_chain/metadata.rs
@@ -8,6 +8,7 @@ pub struct Model {
     pub uuid: Uuid,
     pub key: String,
     pub value: String,
+    pub bytes: Vec<u8>,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter)]
diff --git a/tests/common/setup/schema.rs b/tests/common/setup/schema.rs
index 1a0a1c0a..78947c36 100644
--- a/tests/common/setup/schema.rs
+++ b/tests/common/setup/schema.rs
@@ -281,6 +281,7 @@ pub async fn create_metadata_table(db: &DbConn) -> Result<ExecResult, DbErr> {
         )
         .col(ColumnDef::new(metadata::Column::Key).string().not_null())
         .col(ColumnDef::new(metadata::Column::Value).string().not_null())
+        .col(ColumnDef::new(metadata::Column::Bytes).binary().not_null())
         .to_owned();
 
     create_table(db, &stmt, Metadata).await
diff --git a/tests/primary_key_tests.rs b/tests/primary_key_tests.rs
index ea8255e9..8f1acfed 100644
--- a/tests/primary_key_tests.rs
+++ b/tests/primary_key_tests.rs
@@ -1,7 +1,7 @@
 pub mod common;
 
 pub use common::{bakery_chain::*, setup::*, TestContext};
-use sea_orm::{entity::prelude::*, DatabaseConnection, Set};
+use sea_orm::{entity::prelude::*, DatabaseConnection, IntoActiveModel, Set};
 use uuid::Uuid;
 
 #[sea_orm_macros::test]
@@ -21,18 +21,23 @@ async fn main() -> Result<(), DbErr> {
 }
 
 pub async fn create_metadata(db: &DatabaseConnection) -> Result<(), DbErr> {
-    let metadata = metadata::ActiveModel {
-        uuid: Set(Uuid::new_v4()),
-        key: Set("markup".to_owned()),
-        value: Set("1.18".to_owned()),
+    let metadata = metadata::Model {
+        uuid: Uuid::new_v4(),
+        key: "markup".to_owned(),
+        value: "1.18".to_owned(),
+        bytes: vec![1, 2, 3],
     };
 
-    let res = Metadata::insert(metadata.clone()).exec(db).await?;
+    let res = Metadata::insert(metadata.clone().into_active_model())
+        .exec(db)
+        .await?;
+
+    assert_eq!(Metadata::find().one(db).await?, Some(metadata.clone()));
 
     assert_eq!(
         res.last_insert_id,
         if cfg!(feature = "sqlx-postgres") {
-            metadata.uuid.unwrap()
+            metadata.uuid
         } else {
             Default::default()
         }