-
Notifications
You must be signed in to change notification settings - Fork 8
281 lines (236 loc) · 9.07 KB
/
check-sdk-tests.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
name: Bittensor SDK Test
permissions:
pull-requests: write
contents: read
concurrency:
group: e2e-sdk-${{ github.ref }}
cancel-in-progress: true
on:
pull_request:
branches:
- main
- staging
types: [opened, synchronize, reopened, labeled, unlabeled]
env:
CARGO_TERM_COLOR: always
VERBOSE: ${{ github.event.inputs.verbose }}
jobs:
apply-label-to-new-pr:
runs-on: ubuntu-latest
if: ${{ github.event.pull_request.draft == false }}
outputs:
should_continue_sdk: ${{ steps.check.outputs.should_continue_sdk }}
steps:
- name: Check
id: check
run: |
ACTION="${{ github.event.action }}"
if [[ "$ACTION" == "opened" || "$ACTION" == "reopened" ]]; then
echo "should_continue_sdk=true" >> $GITHUB_OUTPUT
else
echo "should_continue_sdk=false" >> $GITHUB_OUTPUT
fi
shell: bash
- name: Add label
if: steps.check.outputs.should_continue_sdk == 'true'
uses: actions-ecosystem/action-add-labels@v1
with:
github_token: ${{ secrets.GITHUB_TOKEN }}
labels: run-bittensor-sdk-tests
check-labels:
needs: apply-label-to-new-pr
runs-on: ubuntu-latest
if: always()
outputs:
run-sdk-tests: ${{ steps.get-labels.outputs.run-sdk-tests }}
steps:
- name: Check out repository
uses: actions/checkout@v4
- name: Get labels from PR
id: get-labels
run: |
sleep 5
LABELS=$(gh api repos/${{ github.repository }}/issues/${{ github.event.pull_request.number }}/labels --jq '.[].name')
echo "Current labels: $LABELS"
if echo "$LABELS" | grep -q "run-bittensor-sdk-tests"; then
echo "run-sdk-tests=true" >> $GITHUB_OUTPUT
else
echo "run-sdk-tests=false" >> $GITHUB_OUTPUT
fi
env:
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
find-e2e-tests:
needs: check-labels
if: always() && needs.check-labels.outputs.run-sdk-tests == 'true'
runs-on: ubuntu-latest
outputs:
test-files: ${{ steps.get-tests.outputs.test-files }}
steps:
- name: Research preparation
working-directory: ${{ github.workspace }}
run: git clone https://github.com/opentensor/bittensor.git
- name: Checkout
working-directory: ${{ github.workspace }}/bittensor
run: git checkout staging
- name: Install dependencies
run: sudo apt-get install -y jq
- name: Find e2e test files
id: get-tests
run: |
test_files=$(find ${{ github.workspace }}/bittensor/tests/e2e_tests -name "test*.py" | jq -R -s -c 'split("\n") | map(select(. != ""))')
echo "::set-output name=test-files::$test_files"
shell: bash
pull-docker-image:
needs: check-labels
runs-on: ubuntu-latest
if: always() && needs.check-labels.outputs.run-sdk-tests == 'true'
steps:
- name: Log in to GitHub Container Registry
run: echo "${{ secrets.GITHUB_TOKEN }}" | docker login ghcr.io -u $GITHUB_ACTOR --password-stdin
- name: Pull Docker Image
run: docker pull ghcr.io/opentensor/subtensor-localnet:devnet-ready
- name: Save Docker Image to Cache
run: docker save -o subtensor-localnet.tar ghcr.io/opentensor/subtensor-localnet:devnet-ready
- name: Upload Docker Image as Artifact
uses: actions/upload-artifact@v4
with:
name: subtensor-localnet
path: subtensor-localnet.tar
run-e2e-tests:
needs:
- check-labels
- find-e2e-tests
- pull-docker-image
if: always() && needs.check-labels.outputs.run-sdk-tests == 'true'
runs-on: ubuntu-latest
strategy:
fail-fast: false
max-parallel: 16
matrix:
rust-branch:
- stable
rust-target:
- x86_64-unknown-linux-gnu
os:
- ubuntu-latest
test-file: ${{ fromJson(needs.find-e2e-tests.outputs.test-files) }}
env:
RELEASE_NAME: development
RUSTV: ${{ matrix.rust-branch }}
RUST_BACKTRACE: full
RUST_BIN_DIR: target/${{ matrix.rust-target }}
TARGET: ${{ matrix.rust-target }}
timeout-minutes: 60
name: "e2e tests: ${{ matrix.test-file }}"
steps:
- name: Check-out repository
uses: actions/checkout@v4
- name: Install Rust stable
uses: actions-rs/toolchain@v1.0.6
with:
toolchain: ${{ matrix.rust-branch }}
components: rustfmt
profile: minimal
- name: Install rust/cargo dependencies
run: |
sudo apt-get update &&
sudo apt-get install -y clang curl libssl-dev llvm libudev-dev protobuf-compiler jq
- name: Add wasm32-unknown-unknown target
run: |
rustup target add wasm32-unknown-unknown --toolchain stable-x86_64-unknown-linux-gnu
rustup component add rust-src --toolchain stable-x86_64-unknown-linux-gnu
- name: Install uv
uses: astral-sh/setup-uv@v5
- name: Create Python virtual environment
working-directory: ${{ github.workspace }}
run: uv venv ${{ github.workspace }}/venv
- name: Clone Bittensor SDK repo
working-directory: ${{ github.workspace }}
run: git clone https://github.com/opentensor/bittensor.git
- name: Setup Bittensor SDK from cloned repo
working-directory: ${{ github.workspace }}/bittensor
run: |
source ${{ github.workspace }}/venv/bin/activate
git checkout staging
git fetch origin staging
uv run --active pip install --upgrade pip
uv run --active pip install '.[dev]'
uv run --active pip install maturin
- name: Install uv dependencies
working-directory: ${{ github.workspace }}
run: uv sync --all-extras --dev
- name: Clone Bittensor Wallet repo
run: git clone https://github.com/opentensor/btwallet.git
- name: Checkout PR branch in Wallet repo
working-directory: ${{ github.workspace }}/btwallet
run: |
git fetch origin ${{ github.event.pull_request.head.ref }}
git checkout ${{ github.event.pull_request.head.ref }}
echo "Current branch: $(git rev-parse --abbrev-ref HEAD)"
- name: Build and install Rust Bittensor Wallet package
working-directory: ${{ github.workspace }}
run: |
source ${{ github.workspace }}/venv/bin/activate
uv run --active maturin develop
- name: Download Cached Docker Image
uses: actions/download-artifact@v4
with:
name: subtensor-localnet
- name: Load Docker Image
run: docker load -i subtensor-localnet.tar
- name: Run tests
working-directory: ${{ github.workspace }}
run: |
source ${{ github.workspace }}/venv/bin/activate
uv run pytest ${{ matrix.test-file }} -s
run-integration-and-unit-test:
needs:
- check-labels
if: always() && needs.check-labels.outputs.run-sdk-tests == 'true'
runs-on: ubuntu-latest
steps:
- name: Check-out repository
uses: actions/checkout@v4
- name: Install dependencies
run: |
sudo apt-get update &&
sudo apt-get install -y clang curl libssl-dev llvm libudev-dev protobuf-compiler
- name: Create Python virtual environment
working-directory: ${{ github.workspace }}
run: python3 -m venv venv
- name: Clone Bittensor SDK repo
working-directory: ${{ github.workspace }}
run: git clone https://github.com/opentensor/bittensor.git
- name: Setup Bittensor SDK from cloned repo
working-directory: ${{ github.workspace }}/bittensor
run: |
source ${{ github.workspace }}/venv/bin/activate
git checkout staging
git fetch origin staging
python3 -m pip install --upgrade pip
python3 -m pip install '.[dev]'
python3 -m pip install -r requirements/torch.txt
python3 -m pip install maturin
- name: Clone Bittensor Wallet repo
run: git clone https://github.com/opentensor/btwallet.git
- name: Checkout PR branch in Wallet repo
working-directory: ${{ github.workspace }}/btwallet
run: |
git fetch origin ${{ github.event.pull_request.head.ref }}
git checkout ${{ github.event.pull_request.head.ref }}
echo "Current branch: $(git rev-parse --abbrev-ref HEAD)"
- name: Build and install Rust Bittensor Wallet package
working-directory: ${{ github.workspace }}
run: |
source ${{ github.workspace }}/venv/bin/activate
maturin develop
- name: Run SDK integration tests
working-directory: ${{ github.workspace }}
run: |
source ${{ github.workspace }}/venv/bin/activate
pytest ${{ github.workspace }}/bittensor/tests/integration_tests
- name: Run bittensor-sdk unit tests
working-directory: ${{ github.workspace }}
run: |
source ${{ github.workspace }}/venv/bin/activate
pytest ${{ github.workspace }}/bittensor/tests/unit_tests