rusttests/test_local.rs -rw-r--r-- 4.4 KiB
1// Copyright (c) 2021 Jason White
2//
3// Permission is hereby granted, free of charge, to any person obtaining a copy
4// of this software and associated documentation files (the "Software"), to deal
5// in the Software without restriction, including without limitation the rights
6// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7// copies of the Software, and to permit persons to whom the Software is
8// furnished to do so, subject to the following conditions:
9//
10// The above copyright notice and this permission notice shall be included in
11// all copies or substantial portions of the Software.
12//
13// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
19// SOFTWARE.
20mod common;
21
22use std::io;
23use std::net::SocketAddr;
24use std::path::Path;
25
26use futures::future::Either;
27use rand::rngs::StdRng;
28use rand::Rng;
29use rand::SeedableRng;
30use rudolfs::LocalServerBuilder;
31use tokio::sync::oneshot;
32
33use common::{init_logger, GitRepo, SERVER_ADDR};
34
35#[tokio::test(flavor = "multi_thread")]
36async fn local_smoke_test_encrypted() -> Result<(), Box<dyn std::error::Error>>
37{
38 init_logger();
39
40 // Make sure our seed is deterministic. This makes it easier to reproduce
41 // the same repo every time.
42 let mut rng = StdRng::seed_from_u64(42);
43
44 let data = tempfile::TempDir::new()?;
45 let key = rng.gen();
46
47 let mut server = LocalServerBuilder::new(data.path().into());
48 server.key(key);
49 let server = server.spawn(SERVER_ADDR).await?;
50 let addr = server.addr();
51
52 let (shutdown_tx, shutdown_rx) = oneshot::channel();
53
54 let server = tokio::spawn(futures::future::select(shutdown_rx, server));
55
56 exercise_server(addr, &mut rng)?;
57
58 shutdown_tx.send(()).expect("server died too soon");
59
60 if let Either::Right((result, _)) = server.await? {
61 // If the server exited first, then propagate the error.
62 result?;
63 }
64
65 Ok(())
66}
67
68#[tokio::test(flavor = "multi_thread")]
69async fn local_smoke_test_unencrypted() -> Result<(), Box<dyn std::error::Error>>
70{
71 init_logger();
72
73 // Make sure our seed is deterministic. This makes it easier to reproduce
74 // the same repo every time.
75 let mut rng = StdRng::seed_from_u64(42);
76
77 let data = tempfile::TempDir::new()?;
78
79 let server = LocalServerBuilder::new(data.path().into());
80 let server = server.spawn(SERVER_ADDR).await?;
81 let addr = server.addr();
82
83 let (shutdown_tx, shutdown_rx) = oneshot::channel();
84
85 let server = tokio::spawn(futures::future::select(shutdown_rx, server));
86
87 exercise_server(addr, &mut rng)?;
88
89 shutdown_tx.send(()).expect("server died too soon");
90
91 if let Either::Right((result, _)) = server.await? {
92 // If the server exited first, then propagate the error.
93 result?;
94 }
95
96 Ok(())
97}
98
99/// Creates a repository with a few LFS files in it to exercise the LFS server.
100fn exercise_server(addr: SocketAddr, rng: &mut impl Rng) -> io::Result<()> {
101 let repo = GitRepo::init(addr)?;
102 repo.add_random(Path::new("4mb.bin"), 4 * 1024 * 1024, rng)?;
103 repo.add_random(Path::new("8mb.bin"), 8 * 1024 * 1024, rng)?;
104 repo.add_random(Path::new("16mb.bin"), 16 * 1024 * 1024, rng)?;
105 repo.commit("Add LFS objects")?;
106
107 // Make sure we can push LFS objects to the server.
108 repo.lfs_push()?;
109
110 // Push again. This should be super fast.
111 repo.lfs_push()?;
112
113 // This should be fast since we already have the data
114 repo.lfs_pull()?;
115
116 // Make sure we can re-download the same objects in another repo
117 let repo_clone = repo.clone_repo().expect("unable to clone");
118
119 // This should be fast since the lfs data should come along properly with
120 // the clone
121 repo_clone.lfs_pull()?;
122
123 // Add some more files and make sure you can pull those into the clone
124 repo.add_random(Path::new("4mb_2.bin"), 4 * 1024 * 1024, rng)?;
125 repo.add_random(Path::new("8mb_2.bin"), 8 * 1024 * 1024, rng)?;
126 repo.add_random(Path::new("16mb_2.bin"), 16 * 1024 * 1024, rng)?;
127 repo.commit("Add LFS objects 2")?;
128
129 repo_clone.pull()?;
130
131 Ok(())
132}