e2e-aci_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649
  1. /*
  2. Copyright 2020 Docker, Inc.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package main
  14. import (
  15. "context"
  16. "errors"
  17. "fmt"
  18. "io/ioutil"
  19. "net/http"
  20. "net/url"
  21. "os"
  22. "runtime"
  23. "strconv"
  24. "strings"
  25. "syscall"
  26. "testing"
  27. "time"
  28. "gotest.tools/v3/assert"
  29. is "gotest.tools/v3/assert/cmp"
  30. "gotest.tools/v3/icmd"
  31. "gotest.tools/v3/poll"
  32. "github.com/Azure/azure-sdk-for-go/profiles/2019-03-01/resources/mgmt/resources"
  33. azure_storage "github.com/Azure/azure-sdk-for-go/profiles/2019-03-01/storage/mgmt/storage"
  34. "github.com/Azure/azure-storage-file-go/azfile"
  35. "github.com/Azure/go-autorest/autorest/to"
  36. "github.com/docker/api/aci"
  37. "github.com/docker/api/aci/login"
  38. "github.com/docker/api/containers"
  39. "github.com/docker/api/context/store"
  40. "github.com/docker/api/errdefs"
  41. "github.com/docker/api/tests/aci-e2e/storage"
  42. . "github.com/docker/api/tests/framework"
  43. )
  44. const (
  45. contextName = "aci-test"
  46. location = "eastus2"
  47. )
  48. var binDir string
  49. func TestMain(m *testing.M) {
  50. p, cleanup, err := SetupExistingCLI()
  51. if err != nil {
  52. fmt.Println(err)
  53. os.Exit(1)
  54. }
  55. binDir = p
  56. exitCode := m.Run()
  57. cleanup()
  58. os.Exit(exitCode)
  59. }
  60. // Cannot be parallelized as login/logout is global.
  61. func TestLoginLogout(t *testing.T) {
  62. startTime := strconv.Itoa(int(time.Now().UnixNano()))
  63. c := NewE2eCLI(t, binDir)
  64. rg := "E2E-" + startTime
  65. t.Run("login", func(t *testing.T) {
  66. azureLogin(t, c)
  67. })
  68. t.Run("create context", func(t *testing.T) {
  69. sID := getSubscriptionID(t)
  70. err := createResourceGroup(sID, rg)
  71. assert.Check(t, is.Nil(err))
  72. t.Cleanup(func() {
  73. _ = deleteResourceGroup(rg)
  74. })
  75. c.RunDockerCmd("context", "create", "aci", contextName, "--subscription-id", sID, "--resource-group", rg, "--location", location)
  76. res := c.RunDockerCmd("context", "use", contextName)
  77. res.Assert(t, icmd.Expected{Out: contextName})
  78. res = c.RunDockerCmd("context", "ls")
  79. res.Assert(t, icmd.Expected{Out: contextName + " *"})
  80. })
  81. t.Run("delete context", func(t *testing.T) {
  82. res := c.RunDockerCmd("context", "use", "default")
  83. res.Assert(t, icmd.Expected{Out: "default"})
  84. res = c.RunDockerCmd("context", "rm", contextName)
  85. res.Assert(t, icmd.Expected{Out: contextName})
  86. })
  87. t.Run("logout", func(t *testing.T) {
  88. _, err := os.Stat(login.GetTokenStorePath())
  89. assert.NilError(t, err)
  90. res := c.RunDockerCmd("logout", "azure")
  91. res.Assert(t, icmd.Expected{Out: "Removing login credentials for Azure"})
  92. _, err = os.Stat(login.GetTokenStorePath())
  93. errMsg := "no such file or directory"
  94. if runtime.GOOS == "windows" {
  95. errMsg = "The system cannot find the file specified"
  96. }
  97. assert.ErrorContains(t, err, errMsg)
  98. })
  99. t.Run("create context fail", func(t *testing.T) {
  100. res := c.RunDockerOrExitError("context", "create", "aci", "fail-context")
  101. res.Assert(t, icmd.Expected{
  102. ExitCode: errdefs.ExitCodeLoginRequired,
  103. Err: `not logged in to azure, you need to run "docker login azure" first`,
  104. })
  105. })
  106. }
  107. func TestContainerRun(t *testing.T) {
  108. c := NewParallelE2eCLI(t, binDir)
  109. sID, rg := setupTestResourceGroup(t, c)
  110. const (
  111. testShareName = "dockertestshare"
  112. testFileContent = "Volume mounted successfully!"
  113. testFileName = "index.html"
  114. )
  115. // Bootstrap volume
  116. aciContext := store.AciContext{
  117. SubscriptionID: sID,
  118. Location: location,
  119. ResourceGroup: rg,
  120. }
  121. saName := "e2e" + strconv.Itoa(int(time.Now().UnixNano()))
  122. _, cleanupSa := createStorageAccount(t, aciContext, saName)
  123. t.Cleanup(func() {
  124. if err := cleanupSa(); err != nil {
  125. t.Error(err)
  126. }
  127. })
  128. keys := getStorageKeys(t, aciContext, saName)
  129. assert.Assert(t, len(keys) > 0)
  130. k := *keys[0].Value
  131. cred, u := createFileShare(t, k, testShareName, saName)
  132. uploadFile(t, *cred, u.String(), testFileName, testFileContent)
  133. // Used in subtests
  134. var (
  135. container string
  136. hostIP string
  137. endpoint string
  138. )
  139. t.Run("run", func(t *testing.T) {
  140. mountTarget := "/usr/share/nginx/html"
  141. res := c.RunDockerCmd(
  142. "run", "-d",
  143. "-v", fmt.Sprintf("%s@%s:%s", saName, testShareName, mountTarget),
  144. "-p", "80:80",
  145. "nginx",
  146. )
  147. container = getContainerName(res.Stdout())
  148. })
  149. t.Run("inspect", func(t *testing.T) {
  150. res := c.RunDockerCmd("inspect", container)
  151. containerInspect, err := ParseContainerInspect(res.Stdout())
  152. assert.NilError(t, err)
  153. assert.Equal(t, containerInspect.Platform, "Linux")
  154. assert.Equal(t, containerInspect.CPULimit, 1.0)
  155. assert.Equal(t, containerInspect.RestartPolicyCondition, containers.RestartPolicyNone)
  156. assert.Assert(t, is.Len(containerInspect.Ports, 1))
  157. hostIP = containerInspect.Ports[0].HostIP
  158. endpoint = fmt.Sprintf("http://%s:%d", containerInspect.Ports[0].HostIP, containerInspect.Ports[0].HostPort)
  159. })
  160. t.Run("ps", func(t *testing.T) {
  161. res := c.RunDockerCmd("ps")
  162. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  163. l := out[len(out)-1]
  164. assert.Assert(t, strings.Contains(l, container), "Looking for %q in line: %s", container, l)
  165. assert.Assert(t, strings.Contains(l, "nginx"))
  166. assert.Assert(t, strings.Contains(l, "Running"))
  167. assert.Assert(t, strings.Contains(l, hostIP+":80->80/tcp"))
  168. })
  169. t.Run("http get", func(t *testing.T) {
  170. r, err := http.Get(endpoint)
  171. assert.NilError(t, err)
  172. assert.Equal(t, r.StatusCode, http.StatusOK)
  173. b, err := ioutil.ReadAll(r.Body)
  174. assert.NilError(t, err)
  175. assert.Assert(t, strings.Contains(string(b), testFileContent), "Actual content: "+string(b))
  176. })
  177. t.Run("logs", func(t *testing.T) {
  178. res := c.RunDockerCmd("logs", container)
  179. res.Assert(t, icmd.Expected{Out: "GET"})
  180. })
  181. t.Run("exec", func(t *testing.T) {
  182. res := c.RunDockerCmd("exec", container, "pwd")
  183. res.Assert(t, icmd.Expected{Out: "/"})
  184. res = c.RunDockerOrExitError("exec", container, "echo", "fail_with_argument")
  185. res.Assert(t, icmd.Expected{
  186. ExitCode: 1,
  187. Err: "ACI exec command does not accept arguments to the command. Only the binary should be specified",
  188. })
  189. })
  190. t.Run("logs follow", func(t *testing.T) {
  191. cmd := c.NewDockerCmd("logs", "--follow", container)
  192. res := icmd.StartCmd(cmd)
  193. checkUp := func(t poll.LogT) poll.Result {
  194. r, _ := http.Get(endpoint + "/is_up")
  195. if r != nil && r.StatusCode == http.StatusNotFound {
  196. return poll.Success()
  197. }
  198. return poll.Continue("waiting for container to serve request")
  199. }
  200. poll.WaitOn(t, checkUp, poll.WithDelay(1*time.Second), poll.WithTimeout(60*time.Second))
  201. assert.Assert(t, !strings.Contains(res.Stdout(), "/test"))
  202. checkLogs := func(t poll.LogT) poll.Result {
  203. if strings.Contains(res.Stdout(), "/test") {
  204. return poll.Success()
  205. }
  206. return poll.Continue("waiting for logs to contain /test")
  207. }
  208. // Do request on /test
  209. go func() {
  210. time.Sleep(3 * time.Second)
  211. _, _ = http.Get(endpoint + "/test")
  212. }()
  213. poll.WaitOn(t, checkLogs, poll.WithDelay(3*time.Second), poll.WithTimeout(20*time.Second))
  214. if runtime.GOOS == "windows" {
  215. err := res.Cmd.Process.Kill()
  216. assert.NilError(t, err)
  217. } else {
  218. err := res.Cmd.Process.Signal(syscall.SIGTERM)
  219. assert.NilError(t, err)
  220. }
  221. })
  222. t.Run("rm a running container", func(t *testing.T) {
  223. res := c.RunDockerOrExitError("rm", container)
  224. res.Assert(t, icmd.Expected{
  225. Err: fmt.Sprintf("Error: you cannot remove a running container %s. Stop the container before attempting removal or force remove", container),
  226. ExitCode: 1,
  227. })
  228. })
  229. t.Run("force rm", func(t *testing.T) {
  230. res := c.RunDockerCmd("rm", "-f", container)
  231. res.Assert(t, icmd.Expected{Out: container})
  232. checkStopped := func(t poll.LogT) poll.Result {
  233. res := c.RunDockerOrExitError("inspect", container)
  234. if res.ExitCode == 1 {
  235. return poll.Success()
  236. }
  237. return poll.Continue("waiting for container to stop")
  238. }
  239. poll.WaitOn(t, checkStopped, poll.WithDelay(5*time.Second), poll.WithTimeout(60*time.Second))
  240. })
  241. }
  242. func TestContainerRunAttached(t *testing.T) {
  243. c := NewParallelE2eCLI(t, binDir)
  244. _, _ = setupTestResourceGroup(t, c)
  245. // Used in subtests
  246. var (
  247. container string
  248. endpoint string
  249. )
  250. container = "test-container"
  251. t.Run("run attached limits", func(t *testing.T) {
  252. cmd := c.NewDockerCmd(
  253. "run",
  254. "--name", container,
  255. "--restart", "on-failure",
  256. "--memory", "0.1G", "--cpus", "0.1",
  257. "-p", "80:80",
  258. "nginx",
  259. )
  260. runRes := icmd.StartCmd(cmd)
  261. checkRunning := func(t poll.LogT) poll.Result {
  262. res := c.RunDockerOrExitError("inspect", container)
  263. if res.ExitCode == 0 {
  264. return poll.Success()
  265. }
  266. return poll.Continue("waiting for container to be running")
  267. }
  268. poll.WaitOn(t, checkRunning, poll.WithDelay(5*time.Second), poll.WithTimeout(60*time.Second))
  269. inspectRes := c.RunDockerCmd("inspect", container)
  270. containerInspect, err := ParseContainerInspect(inspectRes.Stdout())
  271. assert.NilError(t, err)
  272. assert.Equal(t, containerInspect.Platform, "Linux")
  273. assert.Equal(t, containerInspect.CPULimit, 0.1)
  274. assert.Equal(t, containerInspect.MemoryLimit, uint64(107374182))
  275. assert.Equal(t, containerInspect.RestartPolicyCondition, containers.RestartPolicyOnFailure)
  276. assert.Assert(t, is.Len(containerInspect.Ports, 1))
  277. port := containerInspect.Ports[0]
  278. assert.Assert(t, len(port.HostIP) > 0)
  279. assert.Equal(t, port.ContainerPort, uint32(80))
  280. assert.Equal(t, port.HostPort, uint32(80))
  281. endpoint = fmt.Sprintf("http://%s:%d", port.HostIP, port.HostPort)
  282. assert.Assert(t, !strings.Contains(runRes.Stdout(), "/test"))
  283. checkRequest := func(t poll.LogT) poll.Result {
  284. r, _ := http.Get(endpoint + "/test")
  285. if r != nil && r.StatusCode == http.StatusNotFound {
  286. return poll.Success()
  287. }
  288. return poll.Continue("waiting for container to serve request")
  289. }
  290. poll.WaitOn(t, checkRequest, poll.WithDelay(1*time.Second), poll.WithTimeout(60*time.Second))
  291. checkLog := func(t poll.LogT) poll.Result {
  292. if strings.Contains(runRes.Stdout(), "/test") {
  293. return poll.Success()
  294. }
  295. return poll.Continue("waiting for logs to contain /test")
  296. }
  297. poll.WaitOn(t, checkLog, poll.WithDelay(1*time.Second), poll.WithTimeout(20*time.Second))
  298. })
  299. t.Run("stop wrong container", func(t *testing.T) {
  300. res := c.RunDockerOrExitError("stop", "unknown-container")
  301. res.Assert(t, icmd.Expected{
  302. Err: "Error: container unknown-container not found",
  303. ExitCode: 1,
  304. })
  305. })
  306. t.Run("stop container", func(t *testing.T) {
  307. res := c.RunDockerCmd("stop", container)
  308. res.Assert(t, icmd.Expected{Out: container})
  309. })
  310. t.Run("ps stopped container with --all", func(t *testing.T) {
  311. res := c.RunDockerCmd("ps", container)
  312. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  313. assert.Assert(t, is.Len(out, 1))
  314. res = c.RunDockerCmd("ps", "--all", container)
  315. out = strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  316. assert.Assert(t, is.Len(out, 2))
  317. })
  318. t.Run("start container", func(t *testing.T) {
  319. res := c.RunDockerCmd("start", container)
  320. res.Assert(t, icmd.Expected{Out: container})
  321. waitForStatus(t, c, container, "Running")
  322. })
  323. t.Run("rm stopped container", func(t *testing.T) {
  324. res := c.RunDockerCmd("stop", container)
  325. res.Assert(t, icmd.Expected{Out: container})
  326. waitForStatus(t, c, container, "Terminated", "Node Stopped")
  327. res = c.RunDockerCmd("rm", container)
  328. res.Assert(t, icmd.Expected{Out: container})
  329. })
  330. }
  331. func TestCompose(t *testing.T) {
  332. c := NewParallelE2eCLI(t, binDir)
  333. _, _ = setupTestResourceGroup(t, c)
  334. const (
  335. composeFile = "../composefiles/aci-demo/aci_demo_port.yaml"
  336. composeFileMultiplePorts = "../composefiles/aci-demo/aci_demo_multi_port.yaml"
  337. composeProjectName = "acidemo"
  338. serverContainer = composeProjectName + "_web"
  339. wordsContainer = composeProjectName + "_words"
  340. )
  341. t.Run("compose up", func(t *testing.T) {
  342. // Name of Compose project is taken from current folder "acie2e"
  343. c.RunDockerCmd("compose", "up", "-f", composeFile)
  344. res := c.RunDockerCmd("ps")
  345. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  346. // Check three containers are running
  347. assert.Assert(t, is.Len(out, 4))
  348. webRunning := false
  349. for _, l := range out {
  350. if strings.Contains(l, serverContainer) {
  351. webRunning = true
  352. strings.Contains(l, ":80->80/tcp")
  353. }
  354. }
  355. assert.Assert(t, webRunning, "web container not running")
  356. res = c.RunDockerCmd("inspect", serverContainer)
  357. containerInspect, err := ParseContainerInspect(res.Stdout())
  358. assert.NilError(t, err)
  359. assert.Assert(t, is.Len(containerInspect.Ports, 1))
  360. endpoint := fmt.Sprintf("http://%s:%d", containerInspect.Ports[0].HostIP, containerInspect.Ports[0].HostPort)
  361. r, err := http.Get(endpoint + "/words/noun")
  362. assert.NilError(t, err)
  363. assert.Equal(t, r.StatusCode, http.StatusOK)
  364. b, err := ioutil.ReadAll(r.Body)
  365. assert.NilError(t, err)
  366. assert.Assert(t, strings.Contains(string(b), `"word":`))
  367. })
  368. t.Run("logs web", func(t *testing.T) {
  369. res := c.RunDockerCmd("logs", serverContainer)
  370. res.Assert(t, icmd.Expected{Out: "Listening on port 80"})
  371. })
  372. t.Run("update", func(t *testing.T) {
  373. c.RunDockerCmd("compose", "up", "-f", composeFileMultiplePorts, "--project-name", composeProjectName)
  374. res := c.RunDockerCmd("ps")
  375. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  376. // Check three containers are running
  377. assert.Assert(t, is.Len(out, 4))
  378. for _, cName := range []string{serverContainer, wordsContainer} {
  379. res = c.RunDockerCmd("inspect", cName)
  380. containerInspect, err := ParseContainerInspect(res.Stdout())
  381. assert.NilError(t, err)
  382. assert.Assert(t, is.Len(containerInspect.Ports, 1))
  383. endpoint := fmt.Sprintf("http://%s:%d", containerInspect.Ports[0].HostIP, containerInspect.Ports[0].HostPort)
  384. var route string
  385. switch cName {
  386. case serverContainer:
  387. route = "/words/noun"
  388. assert.Equal(t, containerInspect.Ports[0].HostPort, uint32(80))
  389. assert.Equal(t, containerInspect.Ports[0].ContainerPort, uint32(80))
  390. case wordsContainer:
  391. route = "/noun"
  392. assert.Equal(t, containerInspect.Ports[0].HostPort, uint32(8080))
  393. assert.Equal(t, containerInspect.Ports[0].ContainerPort, uint32(8080))
  394. }
  395. checkUp := func(t poll.LogT) poll.Result {
  396. r, _ := http.Get(endpoint + route)
  397. if r != nil && r.StatusCode == http.StatusOK {
  398. return poll.Success()
  399. }
  400. return poll.Continue("Waiting for container to serve request")
  401. }
  402. poll.WaitOn(t, checkUp, poll.WithDelay(1*time.Second), poll.WithTimeout(60*time.Second))
  403. res = c.RunDockerCmd("ps")
  404. p := containerInspect.Ports[0]
  405. res.Assert(t, icmd.Expected{
  406. Out: fmt.Sprintf("%s:%d->%d/tcp", p.HostIP, p.HostPort, p.ContainerPort),
  407. })
  408. }
  409. })
  410. t.Run("down", func(t *testing.T) {
  411. c.RunDockerCmd("compose", "down", "--project-name", composeProjectName)
  412. res := c.RunDockerCmd("ps")
  413. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  414. assert.Equal(t, len(out), 1)
  415. })
  416. }
  417. func TestRunEnvVars(t *testing.T) {
  418. c := NewParallelE2eCLI(t, binDir)
  419. _, _ = setupTestResourceGroup(t, c)
  420. t.Run("run", func(t *testing.T) {
  421. cmd := c.NewDockerCmd(
  422. "run", "-d",
  423. "-e", "MYSQL_ROOT_PASSWORD=rootpwd",
  424. "-e", "MYSQL_DATABASE=mytestdb",
  425. "-e", "MYSQL_USER",
  426. "-e", "MYSQL_PASSWORD=userpwd",
  427. "mysql:5.7",
  428. )
  429. cmd.Env = append(cmd.Env, "MYSQL_USER=user1")
  430. res := icmd.RunCmd(cmd)
  431. res.Assert(t, icmd.Success)
  432. out := strings.Split(strings.TrimSpace(res.Stdout()), "\n")
  433. container := strings.TrimSpace(out[len(out)-1])
  434. res = c.RunDockerCmd("inspect", container)
  435. containerInspect, err := ParseContainerInspect(res.Stdout())
  436. assert.NilError(t, err)
  437. assert.Equal(t, containerInspect.Image, "mysql:5.7")
  438. check := func(t poll.LogT) poll.Result {
  439. res := c.RunDockerOrExitError("logs", container)
  440. if strings.Contains(res.Stdout(), "Giving user user1 access to schema mytestdb") {
  441. return poll.Success()
  442. }
  443. return poll.Continue("waiting for DB container to be up")
  444. }
  445. poll.WaitOn(t, check, poll.WithDelay(5*time.Second), poll.WithTimeout(60*time.Second))
  446. })
  447. }
  448. func setupTestResourceGroup(t *testing.T, c *E2eCLI) (string, string) {
  449. startTime := strconv.Itoa(int(time.Now().Unix()))
  450. rg := "E2E-" + t.Name() + "-" + startTime
  451. azureLogin(t, c)
  452. sID := getSubscriptionID(t)
  453. err := createResourceGroup(sID, rg)
  454. assert.Check(t, is.Nil(err))
  455. t.Cleanup(func() {
  456. if err := deleteResourceGroup(rg); err != nil {
  457. t.Error(err)
  458. }
  459. })
  460. createAciContextAndUseIt(t, c, sID, rg)
  461. // Check nothing is running
  462. res := c.RunDockerCmd("ps")
  463. assert.Assert(t, is.Len(strings.Split(strings.TrimSpace(res.Stdout()), "\n"), 1))
  464. return sID, rg
  465. }
  466. func deleteResourceGroup(rgName string) error {
  467. ctx := context.TODO()
  468. helper := aci.NewACIResourceGroupHelper()
  469. models, err := helper.GetSubscriptionIDs(ctx)
  470. if err != nil {
  471. return err
  472. }
  473. if len(models) == 0 {
  474. return errors.New("unable to delete resource group: no models")
  475. }
  476. return helper.DeleteAsync(ctx, *models[0].SubscriptionID, rgName)
  477. }
  478. func azureLogin(t *testing.T, c *E2eCLI) {
  479. // in order to create new service principal and get these 3 values : `az ad sp create-for-rbac --name 'TestServicePrincipal' --sdk-auth`
  480. clientID := os.Getenv("AZURE_CLIENT_ID")
  481. clientSecret := os.Getenv("AZURE_CLIENT_SECRET")
  482. tenantID := os.Getenv("AZURE_TENANT_ID")
  483. assert.Check(t, clientID != "", "AZURE_CLIENT_ID must not be empty")
  484. assert.Check(t, clientSecret != "", "AZURE_CLIENT_SECRET must not be empty")
  485. assert.Check(t, tenantID != "", "AZURE_TENANT_ID must not be empty")
  486. c.RunDockerCmd("login", "azure", "--client-id", clientID, "--client-secret", clientSecret, "--tenant-id", tenantID)
  487. }
  488. func getSubscriptionID(t *testing.T) string {
  489. ctx := context.TODO()
  490. helper := aci.NewACIResourceGroupHelper()
  491. models, err := helper.GetSubscriptionIDs(ctx)
  492. assert.Check(t, is.Nil(err))
  493. assert.Check(t, len(models) == 1)
  494. return *models[0].SubscriptionID
  495. }
  496. func createResourceGroup(sID, rgName string) error {
  497. helper := aci.NewACIResourceGroupHelper()
  498. _, err := helper.CreateOrUpdate(context.TODO(), sID, rgName, resources.Group{Location: to.StringPtr(location)})
  499. return err
  500. }
  501. func createAciContextAndUseIt(t *testing.T, c *E2eCLI, sID, rgName string) {
  502. c.RunDockerCmd("context", "create", "aci", contextName, "--subscription-id", sID, "--resource-group", rgName, "--location", location)
  503. res := c.RunDockerCmd("context", "use", contextName)
  504. res.Assert(t, icmd.Expected{Out: contextName})
  505. res = c.RunDockerCmd("context", "ls")
  506. res.Assert(t, icmd.Expected{Out: contextName + " *"})
  507. }
  508. func createStorageAccount(t *testing.T, aciContext store.AciContext, name string) (azure_storage.Account, func() error) {
  509. account, err := storage.CreateStorageAccount(context.TODO(), aciContext, name)
  510. assert.Check(t, is.Nil(err))
  511. assert.Check(t, is.Equal(*(account.Name), name))
  512. return account, func() error { return deleteStorageAccount(aciContext, name) }
  513. }
  514. func deleteStorageAccount(aciContext store.AciContext, name string) error {
  515. _, err := storage.DeleteStorageAccount(context.TODO(), aciContext, name)
  516. return err
  517. }
  518. func getStorageKeys(t *testing.T, aciContext store.AciContext, saName string) []azure_storage.AccountKey {
  519. l, err := storage.ListKeys(context.TODO(), aciContext, saName)
  520. assert.NilError(t, err)
  521. assert.Assert(t, l.Keys != nil)
  522. return *l.Keys
  523. }
  524. func createFileShare(t *testing.T, key, share, storageAccount string) (*azfile.SharedKeyCredential, *url.URL) {
  525. // Create a ShareURL object that wraps a soon-to-be-created share's URL and a default pipeline.
  526. u, _ := url.Parse(fmt.Sprintf("https://%s.file.core.windows.net/%s", storageAccount, share))
  527. cred, err := azfile.NewSharedKeyCredential(storageAccount, key)
  528. assert.NilError(t, err)
  529. shareURL := azfile.NewShareURL(*u, azfile.NewPipeline(cred, azfile.PipelineOptions{}))
  530. _, err = shareURL.Create(context.TODO(), azfile.Metadata{}, 0)
  531. assert.NilError(t, err)
  532. return cred, u
  533. }
  534. func uploadFile(t *testing.T, cred azfile.SharedKeyCredential, baseURL, fileName, content string) {
  535. fURL, err := url.Parse(baseURL + "/" + fileName)
  536. assert.NilError(t, err)
  537. fileURL := azfile.NewFileURL(*fURL, azfile.NewPipeline(&cred, azfile.PipelineOptions{}))
  538. err = azfile.UploadBufferToAzureFile(context.TODO(), []byte(content), fileURL, azfile.UploadToAzureFileOptions{})
  539. assert.NilError(t, err)
  540. }
  541. func getContainerName(stdout string) string {
  542. out := strings.Split(strings.TrimSpace(stdout), "\n")
  543. return strings.TrimSpace(out[len(out)-1])
  544. }
  545. func waitForStatus(t *testing.T, c *E2eCLI, containerID string, statuses ...string) {
  546. checkStopped := func(logt poll.LogT) poll.Result {
  547. res := c.RunDockerCmd("inspect", containerID)
  548. containerInspect, err := ParseContainerInspect(res.Stdout())
  549. assert.NilError(t, err)
  550. for _, status := range statuses {
  551. if containerInspect.Status == status {
  552. return poll.Success()
  553. }
  554. }
  555. return poll.Continue("Status %s != %s (expected) for container %s", containerInspect.Status, statuses, containerID)
  556. }
  557. poll.WaitOn(t, checkStopped, poll.WithDelay(5*time.Second), poll.WithTimeout(90*time.Second))
  558. }