MicroBatchFrameworkとSeleniumでCLIツールを作った話(その2)

前回の記事の続きで、Selenium に関する話です。

SeleniumでWebサービスを自動操作する部分を実装します。

Webサービスは以下の流れで操作します。

WebサービスのIndexページにアクセスする
リンクをクリックし、画面遷移する
アップロードするExcelファイルを選択する
Excelファイルをアップロードする
Excelファイルのアップロードの完了を待つ
アップロード完了後、登録ボタンを押下する
登録ボタン押下後、バックグラウンドの処理の完了を待つ

前回の記事をご参照頂くと分かると思いますが、このツールはChromeDriverを含める形でExeを作成しています。ChromeDriverがアップデートされた際、新しいChromeDriverをExeに含めるためにツールをビルド・発行したくありません。

よってExeと同じ階層にChromeDriverが存在する場合は、そちらのChromeDriverを使用するという仕様でツールを実装します。
こうすることでツールのビルド・発行の手間を省きます。
※不要なChromeDriverがExeに含まれるのは怒られそうですが…。

ソースコード

コードは以下の通りです。
コメントに操作と対応する番号を記載しています。 ※URIとFilePathを関数の引数で受け取るのは微妙ですが、今回は目を瞑ります。。
※ヘッドレスモードにするかどうかはCommandLineの引数で受け取った方が良いですね。

ChromeBrowserAutoOperator.cs

using System;
using HogeBot.Interfaces;
using Microsoft.Extensions.Logging;

namespace HogeBot.Services
{
    public class ChromeBrowserAutoOperator : IAutoOperationable
    {
        private readonly ILogger<ChromeBrowserAutoOperator> _logger;

        public ChromeBrowserAutoOperator(ILogger<ChromeBrowserAutoOperator> logger)
        {
            this._logger = logger;
        }

        public void AutoOperate(string uri, string filePath)
        {
            try
            {
                // ヘッドレスモード(ブラウザ非表示状態で起動する。)
                var options = new ChromeOptions();
                options.AddArgument("--headless");

                // Exeと同じ階層にChromeDriverが存在する場合は、そちらのChromeDriverを使用する
                var exeDir = Path.GetDirectoryName(Process.GetCurrentProcess().MainModule.FileName);
                var dir = File.Exists(Path.Combine(exeDir, "chromeDriver.exe")) ? exeDir : Path.GetDirectoryName(Assembly.GetEntryAssembly().Location);
                using (var driver = new ChromeDriver(dir, options))
                {
                    // 1. WebサービスのIndexページにアクセスする
                    driver.Url = param.URI;

                    // 2. リンクをクリックし、画面遷移する
                    driver.FindElementById("link-id1").Click();

                    // 3. アップロードするExcelファイルを選択する
                    // ※Excelファイルのパスを<input type="file">にSendKeysで入力すると
                    //   ファイル選択ダイアログでファイルを選択したのと同じ状態となります。
                    var fileInfo = new FileInfo(param.ExcelFilePath);
                    driver.FindElementById("input-excel-file").SendKeys(fileInfo.FullName);

                    // 4. Excelファイルをアップロードする
                    driver.FindElementById("button-import-start").Click();

                    // 5. Excelファイルのアップロードの完了を待つ
                    // ※1分間Uploadの完了を待つ(Timeoutすると例外が吐かれる)
                    // 　Uploadの完了は指定したLabel内にメッセージが表示されることで判断する。
                    WebDriverWait wait = new WebDriverWait(driver, new TimeSpan(0, 1, 0));
                    var msgLabel = By.Id("label-upload-msg");
                    wait.Until(f => f.FindElement(msgLabel).Text == "アップロード完了しました。登録ボタンを押してください。");

                    // 6. アップロード完了後、登録ボタンを押下する
                    driver.FindElementById("button-registers").Click();

                    // 7. 登録ボタン押下後、バックグラウンドの処理の完了を待つ
                    wait.Until(f =>
                        f.FindElement(msgLabel).Text == "登録完了" || 
                        f.FindElement(msgLabel).Text == "登録失敗");
                }
                return true;
            }
            catch (Exception e)
            {
                logger.LogError(e, "RunAutomation Exception.");
                return false;
            }
        }
    }
}